eV^dZddlmZddlmZddlmZmZmZddl m Z m Z m Z ddl mZddlmZmZddlmZmZmZdd lmZdd lmZdd lmZdd lmZmZmZed eede j@feede jBffZ"e#dZ$dZ%e#dZ&dZ'e$f d(dZ(d)dZ) d* d+dZ*d,dZ+ d-dZ, d.dZ-d/dZ. d0dZ/ d1 d2dZ0 d1 d3dZ1e$f d4dZ2d5dZ3 d6 d7d Z4d8d!Z5d9d"Z6d9d#Z7d9d$Z8e#d%Z9 d1 d:d&Z:d;d'Z;y ) Astroid hook for the dataclasses library. Support built-in dataclasses, pydantic.dataclasses, and marshmallow_dataclass-annotated dataclasses. References: - https://docs.python.org/3/library/dataclasses.html - https://pydantic-docs.helpmanual.io/usage/dataclasses/ - https://lovasoa.github.io/marshmallow_dataclass/ ) annotations)Iterator)LiteralTupleUnion)basescontextnodes)parse) PY39_PLUS PY310_PLUS)AstroidSyntaxErrorInferenceErrorUseInferenceDefault) inference_tip)AstroidManager)InferenceResult) UninferableUninferableBase safe_inferNdefaultdefault_factory) dataclassfield) dataclassesmarshmallow_dataclasszpydantic.dataclasses_HAS_DEFAULT_FACTORYct|tjr |jsyt fd|jjDS)zDReturn True if a decorated node has a `dataclass` decorator applied.Fc36K|]}t|ywN)_looks_like_dataclass_decorator).0decorator_attributedecorator_namess A/usr/lib/python3/dist-packages/astroid/brain/brain_dataclasses.py z.is_decorated_with_dataclass..2s!  ((;_Ms) isinstancer ClassDef decoratorsany)noder$s `r%is_decorated_with_dataclassr,+s@ dENN +4?? #'??#8#8 c d|_t|D]m}|jj}t j |j |j|}tj|}|g|j|<ot|syd}tr|jjrt|jjD][}t|tjsd}n=|j D],}|j"dk(s|j$j'}.]t)|t+t|d|} t-|d}||_d\|_|_|g|j0d<|j3} t4| j0vrHt-t4d j6d } | | _| j8d g| j0t4<yy#t:$rYywxYw) z6Rewrite a dataclass to be easily understood by pylint.T)lineno col_offsetparentNFkw_only)init__init__NNz = object()r) is_dataclass_get_dataclass_attributestargetnamer Unknownr/r0rvisit_transformsinstance_attrs_check_generate_dataclass_initr r)r'Callkeywordsargvalue bool_value_generate_dataclass_initlistr r1localsrootDEFAULT_FACTORYbodytargetsr) r+ assign_noder9rhs_nodekw_only_decorated decoratorkeywordinit_str init_noderF new_assigns r%dataclass_transformrR8sD06 / !!&&==%%"--  "#44X>%-JD! / *$ /doo++.. CIi4$)!$-- C;;)+(/ (@(@(B% C  C(  &t$ 78H C(OJ/  1;. ).#,+ Jyy{ $++ -/!2+>?DDQGJ $J ,6,>,>q,A+BDKK ( .    sG"" G.-G.c#<K|jD]}t|tjr$t|jtj sBt |jrXt|jrn|st|jr|yw)zqYield the AnnAssign nodes of dataclass attributes for the node. If init is True, also include InitVars. N) rHr'r AnnAssignr8 AssignName _is_class_var annotation_is_keyword_only_sentinel _is_init_var)r+r3rJs r%r7r7jsyy +u7z    0 0@   // 0  $[%;%; <  [%;%;< #sBBcd|jvryd}|jjD]+}t|tjst |s*|}-|yt d|jD S)zReturn True if we should generate an __init__ method for node. This is True when: - node doesn't define its own __init__ method - the dataclass decorator was called *without* the keyword argument init=False r4FNTc3tK|]0}|jdk(xr|jj 2ywr3Nr@rArBr"rNs r%r&z1_check_generate_dataclass_init..s@   v + ((* * +68)rEr)r r'r>r!r*r?)r+foundr#s r%r=r=sT[[  E#44(-uzz:  *+> ?'E ( }~~ r-cZi}i}t|jD]y}|js |jdd}|j j \}}|jD] \}}|||< |jD] \} }||| < {||fS#t$rYwxYw)z@M$(   +/;;z+B1+EI&NN>>@'$NN, *LFD&*N6 " *#==? (KE4#'M%  ('(* = ((!   sB B*)B*cd\}}|jD]/\}}||z }|dr |d|dzz }|dr |d|dzz }|dz }1|jD]/\}}||z }|dr |d|dzz }|dr |d|dzz }|dz }1||fS)zKParse positional and keyword arguments into strings for an __init__ method.)rrr:  = , )rg)rhrirlr2pos_argrnkw_args r%_parse_arguments_into_stringsrys Hg'--/ G 7 tAw &H 7 Q 'HD &++- 6 7 td1g~ %G 7 utAw &G4  W r-c t|jD]}|js||jvs|j|D]}t |j t js(|j js?t |j jt jsnt|j jst|j j}|s|dccSy)z4Get the default value of a previously defined field.rtN) rbrcr6rEr'r1r rTrAr> _looks_like_dataclass_field_call_get_field_default)r+r9rjassignrs r%_get_previous_field_defaultr~s$ *   4;; ++d+ *v}}eoo> ++"6==#6#6 C89L9LM01D1DEG&qz) * * r-c hg}g}g}t|\}}|D]}|jj|j|j} } } d} |j | D]@} t | tjs| js+d| jvs>| } nt | tjxr t| d}|rAtd| jDr%|j| d|j| dt!| r.d}t | tj"r | j$} nd} d}n d}d| d | }d \}}| | j'}| rj|rWt)| }|r|\}}|d k(r|j'}n|d k(rt*}d| d |j'd | dt*d| }nd| j'}nS| r3 t-t/| j1dj'}nt7|| }|r|j'}| }||d|z }||d |z }|rm| jDcgc]}|j8dk(s|}}|rB|djj;r|j=|n|j=||r| |vr||f|| <nJ|j=|n8| |vr||f|| <n,| |vr| g|}|j| n|j=||r|j=|t?||\}}d|vrdnd}||djA|zz }|jCds|dz }|s|r|dz }||djA|z }|rdjA|nd}d|d|S#t2t4f$rYwxYwcc}w)z8Return an init method for a dataclass given the targets.Nzbuiltins.propertyF) check_scopec3tK|]0}|jdk(xr|jj 2ywr\r]r^s r%r&z+_generate_dataclass_init..s9 v%Hgmm.F.F.H*HHr_Trrzself.rur5rrz if z is z else rsr2rselfzself, rvz*, z passz def __init__(z) -> None: )"rpr8r9rWrArEr'r FunctionDefr)decoratornamesr>r{r*r?poprY Subscriptslice as_stringr|rGstrnextinfer_call_resultr StopIterationr~r@rBappendryjoinendswith)r+assignsrLparamskw_only_params assignmentsprev_pos_only_storeprev_kw_only_storer}r9rWrA property_nodeadditional_assignis_fieldinit_varassignment_strann_str default_strresult default_type default_nodeprevious_default param_strkr2 prev_pos_only prev_kw_only params_stringassignments_strings r%rCrCs;F "NK.OPT.U++r/"(--"4"4f6G6G%j37 !%T!2  /1B1BC$//"&7&F&F&HH 1  eUZZ0 5U u6  $~~ $''d3"&&tT2  #H*eoo6'-- " NH$TF#dV4N)  ! **,G +E217.L,#y0&2&<&<&> %)::&5 #D6\-C-C-E,FG""&tO+?IIK  ;4F .88:    2gY 'I  " 3{m, ,I "'..GQAEEY4FqGGG1:##..0")))4MM), )),3[+A"4(%%i0**-4k,B#D)++"&&t, i(   ~ .er/h#@/#M<!M1BxM]TYYv%666M  ! !$ ' ~  ~dii&?%@AAM7B{3 =/)9:L9M NNC#M2  (Hs1NN/N/N,+N,c#K|j}t|tjs ty|j |j }}||j|Ed{|t||Ed{yty7%7w)zInference tip for an Unknown node that was dynamically generated to represent a dataclass attribute. In the case that a default value is provided, that is inferred first. Then, an Instance of the annotated class is yielded. Nr )ctx) r1r'r rTrrWrAinfer_infer_instance_from_annotation)r+rr}rWrAs r%infer_dataclass_attributersz[[F feoo .))6<<J ;;s;+++2:3GGG ,Gs$AB !B"B 8B 9B  B c#Kt|jtjtjfst t |}|s ty|\}}|dk(r|j|Ed{yt|jjdj}|j|_|j|Ed{y7a7w)z(Inference tip for dataclass field calls.rrNr) r'r1r rTAssignrr|rrr rrHrA)r+rrrrnew_calls r%infer_dataclass_field_callrs dkkEOOU\\#B C!!  %F  & g 9 $}}S}1 1 1W..0166q9??H"kkHO~~c~2 2 2 2 3s%A0C2C3ACCCCct|tjr |j} t |j }t|trQt|tjr|j|vSt|tjr|j|vSyt|tjxr0|j|vxr |jjt vS#t tf$r t}YwxYw)zReturn True if node looks like a dataclass decorator. Uses inference to lookup the value of the node, and if that fails, matches against specific names. F)r'r r>funcrrrrrrNamer9 AttributeattrnamerrFDATACLASS_MODULES)r+r$inferreds r%r!r!s$ #yy %(O, dEJJ '99/ / dEOO ,==O3 3 8U../ 6 MM_ , 6 MMO $5 5 M *sC..DDc|j}|sy|j}t|tjxr't|tj xr t |S)z^Return True if node was dynamically generated as the child of an AnnAssign statement. F)r1scoper'r rTr(r,)r+r1rs r%_looks_like_dataclass_attributersQ[[F  LLNE65??+ / uenn - / ' .r-c|rl|j}|j}t|tjr1|j %t|tj r t|sy t|jj}t|tjsy|jtk(xr |j!jt"vS#ttf$rYywxYw)zReturn True if node is calling dataclasses field or Field from an AnnAssign statement directly in the body of a ClassDef. If check_scope is False, skips checking the statement and body. F) statementrr'r rTrAr(r,rrrrrrr9 FIELD_NAMErFr)r+rstmtrrs r%r{r{s~~  tU__ - &5%..1+E2 )* h 1 1 2 ==J & T8==?+?+?CT+TT M *s0#C##C54C5cd\}}|jD]:}|jdk(r |j}|jdk(s/|j}<||d|fS|f|dtj|j |j |j|j|j}|j|ggd|fSy)ahReturn a the default value of a field call, and the corresponding keyword argument name. field(default=...) results in the ... node field(default_factory=...) results in a Call node with func ... and no arguments If neither or both arguments are present, return ("", None) instead, indicating that there is not a valid default value. r5rrN)r/r0r1 end_linenoend_col_offset)rrer?) r?r@rAr r>r/r0r1rrpostinit) field_callrrrNrs r%r|r|s *G_&&, ;;) #mmG [[- -%mmO , 6'!!?6::$$!,,$$!,,%44   R"E (** r-ctr* t|j}t |dddk(St |tjxrt |jtjxr|jjdk(xs?t |jtjxr|jjdk(S#ttf$rYywxYw)z@Return True if node is a ClassVar, with or without subscripting.Fr9rrClassVar)r rrrrgetattrr'r rrArr9rrr+rs r%rVrVs DJJL)Hx, :: dEOO , 4::uzz* * JJOOz ) . djj%// 2 . JJ  : -  .  sC CCctsyt|}t|tjxr|j dk(S)z,Return True if node is the KW_ONLY sentinel.Fzdataclasses._KW_ONLY_TYPE)r rr'rInstanceqnamers r%rXrX0s: $H8U^^, < NN  ; ;r-c t|j}t |dddk(S#ttf$rYywxYw)z@Return True if node is an InitVar, with or without subscripting.Fr9rrInitVar)rrrrrrs r%rYrY;sF % 8VR (I 55 M *s +==)Dict FrozenSetListSetrc#Kd} t|j|}t |t js ty|jjdvr.|jtvr|jyty|jy#ttf$r tYwxYww)zInfer an instance corresponding to the type annotation represented by node. Currently has limited support for the typing module. Nr>rrtyping_collections_abc) rrrrrr'r r(rFr9_INFERABLE_TYPING_TYPESinstantiate_class)r+rklasss r%rrQs ETZZZ,- eU^^ ,     ::0 0))+ + %%'' M *s(B>B!BB>!B;8B>:B;;B>c&|jtjtt|jtj t tdt|jtjt tdty)NT)raise_on_overwrite) register_transformr r(rRr,r>rrr{r:rr)managers r%registerrlsg  +-H  0TJ(   /DI'r-)r+nodes.ClassDefr$frozenset[str]returnbool)r+rrNone)F)r+rr3rrzIterator[nodes.AnnAssign])r+rrr)r+rrzYtuple[dict[str, tuple[str | None, str | None]], dict[str, tuple[str | None, str | None]]])rh(dict[str, tuple[str | None, str | None]]rirrztuple[str, str])r+rr9rrznodes.NodeNG | None)r+rrzlist[nodes.AnnAssign]rLrrrr )r+ nodes.Unknownrcontext.InferenceContext | NonerIterator[InferenceResult])r+ nodes.Callrrrr)r+ nodes.NodeNGr$rrr)r+rrr)T)r+rrrrr)rrr_FieldDefaultReturn)r+rrr)r+rrrrz*Iterator[UninferableBase | bases.Instance])rrrr)<__doc__ __future__rcollections.abcrrrrrastroidrr r astroid.builderr astroid.constr r astroid.exceptionsrrrastroid.inference_tiprastroid.managerrastroid.typingr astroid.utilrrrNodeNGr>r frozensetDATACLASSES_DECORATORSrrrGr,rRr7r=rpryr~rCrrr!rr{r|rVrXrYrrrr-r%rsB #$(())!/VV/**AA ') ell *+ '# $ejj 013 #>2  D)=S   +9   /Cf(-  $6< )  ) )F<;0&PO PO#8POMQPOPOhAE =0>B3 3:33(;Q )7 < "+/U U#'U U<D(6$@D( (<(/(6r-