5 fVdZddlmZddlmZddlmZmZmZddl m Z m Z m Z ddl mZddlmZmZddlmZmZmZdd lmZdd lmZdd lmZdd lmZmZmZed eede j feede j!ffZ"e#dZ$dZ%e#dZ&dZ'e$fdJdZ(dKdZ) dLdMd Z*dNd!Z+dOd#Z,dPd(Z-dQd,Z.dRd0Z/ dSdTd5Z0 dSdUd7Z1e$fdVd9Z2dWd:Z3 dXdYd=Z4dZd@Z5d[dAZ6d[dBZ7d[dCZ8e#dDZ9 dSd\dFZ:d]dIZ;d S)^a> 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_FACTORYnodenodes.ClassDefdecorator_namesfrozenset[str]returnboolct|tjr|jsdSt fd|jjDS)zDReturn True if a decorated node has a `dataclass` decorator applied.Fc38K|]}t|VdSN)_looks_like_dataclass_decorator).0decorator_attributer s U/var/www/equiseq/venv/lib/python3.11/site-packages/astroid/brain/brain_dataclasses.py z.is_decorated_with_dataclass..2sB  ((;_MM) isinstancer ClassDef decoratorsany)rr s `r*is_decorated_with_dataclassr1+sd dEN + +4?u #'?#8  r,Nonec d|_t|D][}|jj}t j|j|j|}t |}|g|j |<\t|sdSd}trg|j jr[|j jD]N}t|tjsd}n/|jD]&}|jdkr|j}'Ot)|t+t|d|} t-|d}||_d\|_|_|g|jd<|} t4| jvrFt-t4d jd } | | _| jd g| jt4<dSdS#t:$rYdSwxYw) z6Rewrite a dataclass to be easily understood by pylint.T)lineno col_offsetparentNFkw_only)init__init__NNz = object()r) is_dataclass_get_dataclass_attributestargetnamer Unknownr4r5rvisit_transformsinstance_attrs_check_generate_dataclass_initr r/r-Callkeywordsargvalue bool_value_generate_dataclass_initlistr r6localsrootDEFAULT_FACTORYbodytargetsr) r assign_noder>rhs_nodekw_only_decorated decoratorkeywordinit_str init_noderK new_assigns r*dataclass_transformrW8sD066 / / !&=%"-    "##44X>>%-JD!! )$ / /Cdo+C. C CIi44 $)!$- C C;)++(/ (@(@(B(B% C(  &t$ 7 7 788H C(OOJ/  1;. ).#,+ Jyy{{ $+ - -/>>>??DQGJ $J ,6,>q,A+BDK ( ( ( . -      s"G GGFr8Iterator[nodes.AnnAssign]c#K|jD]}t|tjrt|jtjs.sX   v + ((** *r,)rJr/r r-rCr'r0rD)rfoundr)s r*rBrBsT[  u E#4((-uz::   *+> ? ? ('E }t~ r,Ytuple[dict[str, tuple[str | None, str | None]], dict[str, tuple[str | None, str | None]]]chi}i}t|D]}|js |jdd}n#t$rY+wxYw|j\}}|D] \}}|||< |D] \} }||| < ||fS)z@M$$((    +/;z+B1+EII    H &N>>@@'$NN,, * *LFD&*N6 " ""==?? ( (KE4#'M%  ( = ((sA AArn(dict[str, tuple[str | None, str | None]]rotuple[str, str]cTd\}}|D];\}}||z }|dr|d|dzz }|dr|d|dzz }|dz }<|D];\}}||z }|dr|d|dzz }|dr|d|dzz }|dz }<||fS)zKParse positional and keyword arguments into strings for an __init__ method.)rzr:  = , )rm)rnrorrr7pos_argrtkw_args r*_parse_arguments_into_stringsrs Hg'--// G 7 ' tAw &H 7 ( Q 'HD%++-- 6 7 & td1g~ %G 7 ' utAw &G4 W r,r>strnodes.NodeNG | Nonect|D]}|js ||jvr|j|D]}t |jt jrp|jjrdt |jjt j r@t|jjr't|jj}|r |dccSdS)z4Get the default value of a previously defined field.r|N) rhrir;rJr-r6r rZrFrC _looks_like_dataclass_field_call_get_field_default)rr>rpassignrs r*_get_previous_field_defaultrs$$ * *    4;  +d+ * *v}eo>>* +*#6=#6 CC*99LMM * 11DEEG*&qz))))) 4r,assignslist[nodes.AnnAssign]rQc g}g}g}t|\}}|D]5}|jj|j|j} } } d} |j| D]?} t | tjs| j s%d| vr| } n@t | tj ot| d}|rKtd| jDr-|| d|| dt!| r)d}t | tjr| j} nd} d}n d}d| d | }d \}}| | }| r~|rgt)| }|rU|\}}|d kr|}n|d kr/t*}d| d |d | dt*d| }n| }n| rZ t-t/| d}n=#t2t4f$rYn*wxYwt7|| }|r|}| }||d|z }||d |z }|r_d| jD}|rL|djr||n|||r"| |vr||f|| <nV||n@| |vr||f|| <n4| |vr| g|}|| n|||s||7t=||\}}d|vrdnd}||d|zz }| ds|dz }|s|r|dz }||d|z }|rd|nd}d|d|S)z8Return an init method for a dataclass given the targets.Nzbuiltins.propertyF) check_scopec3`K|])}|jdko|j V*dSrbrcrds r*r+z+_generate_dataclass_init..sS v%Hgm.F.F.H.H*Hr,Trzzself.r}r:rrz if z is z else r{c(g|]}|jdk |S)r7)rE)r(ks r* z,_generate_dataclass_init..Ts$GGGQAEY4F4Fq4F4F4Fr,rselfzself, r~z*, z passz def __init__(z) -> None: )!rvr=r>r]rFrJr-r FunctionDefr/decoratornamesrCrr0rDpopr_ Subscriptslice as_stringrrLrnextinfer_call_resultr StopIterationrrGappendrjoinendswith)rrrQparamskw_only_params assignmentsprev_pos_only_storeprev_kw_only_storerr>r]rF property_nodeadditional_assignis_fieldinit_varassignment_strann_str default_strresult default_type default_nodeprevious_default param_strr7 prev_pos_only prev_kw_only params_stringassignments_strings r*rHrHssF "NK.OPT.U.U++r/r/"(-"4f6G%j37 !%T!2   /1BCC $/ "&7&F&F&H&HHH 1 IeUZ00 5U u6 6 6   $~  $''d333"&&tT222  # # 5H*eo66 "'- " NNH4T44d44N)  ! **,,G  ; 0+E22 17.L,#y00&2&<&<&>&> %):::&5 JDJJ\-C-C-E-EJJ"&JJ,;JJCGJJ' $oo//  ; !88>>??IIKK #M2     ;4FF  ;.88::    g 'I  " ,{,, ,I  GG%.GGGG 1:#..00-")))4444MM),,,  )))),3[+A"4((%%i0000***-4k,B#D))+++"&&t,,,, i((( /   ~ . . ."?/##M<!M11BBxM]TYYv%6%666M  ! !$ ' ' ~  Adii&?&?AAAM7BN{333 N= N N:L N NNsAH!!H54H5 nodes.Unknownctxcontext.InferenceContext | NoneIterator[InferenceResult]c#K|j}t|tjs tVdS|j|j}}|||Ed{V|t||Ed{VdStVdS)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 )r) r6r-r rZrr]rFinfer_infer_instance_from_annotation)rrrr]rFs r*infer_dataclass_attributers[F feo . .)6<J ;;s;+++++++++2:3GGGGGGGGGGGGr, nodes.Callc#Kt|jtjtjfst t |}|s tVdS|\}}|dkr||Ed{VdSt| j dj }|j|_||Ed{VdS)z(Inference tip for dataclass field calls.rrNr) r-r6r rZAssignrrrrr rrMrF)rrrrrnew_calls r*infer_dataclass_field_callrs dkEOU\#B C C"!!  % %F  3 & g 9 $ $}}S}11 1 1 1 1 1 1 1 1 1W..00116q9?H"kHO~~c~22 2 2 2 2 2 2 2 2 2r, nodes.NodeNGct|tjr|j} t |}n#t tf$r t}YnwxYwt|trHt|tj r |j |vSt|tj r |j |vSdSt|tjo(|j |vo|j t vS)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 rCfuncrrrrrrNamer> AttributeattrnamerrKDATACLASS_MODULES)rr inferreds r*r'r's$ ##y %% M *(O,, dEJ ' ' 09/ / dEO , , 4=O3 3u 8U.// 6 M_ , 6 MMOO $5 5s!AA A c|j}|sdS|}t|tjo(t|tjot |S)z^Return True if node was dynamically generated as the child of an AnnAssign statement. F)r6scoper-r rZr.r1)rr6rs r*_looks_like_dataclass_attributers][F u LLNNE65?++ / uen - - / ' . .r,Trc|rt|}|}t|tjr0|j)t|tjrt|sdS t|j }n#ttf$rYdSwxYwt|tj sdS|jtko|jt"vS)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. NF) statementrr-r rZrFr.r1rrrrrrr> FIELD_NAMErKr)rrstmtrrs r*rrs ~~  tU_ - -  &5%.11'+E22'5 ))** M *uu h 1 2 2u =J & T8==??+?CT+TTs8&BB43B4 field_call_FieldDefaultReturnc"d\}}|jD]'}|jdkr|j}|jdkr|j}(||d|fS|P|Ntj|j|j|j|j|j }| |ggd|fSdS)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. r:rrN)r4r5r6 end_linenoend_col_offset)rrkrD) rDrErFr rCr4r5r6rrpostinit)rrrrSrs r*rrs *G_&,, ;) # #mGG [- - -%mO6'!!?6:$!,$!,%4     R"EEE (** 4r,ctrP t|}n#ttf$rYdSwxYwt |dddkSt |tjo]t |j tj r|j j dkp.t |j tj o|j j dkS)z@Return True if node is a ClassVar, with or without subscripting.Fr>rzClassVar)r rrrrgetattrr-r rrFrr>rrrrs r*r\r\s; DJJLL))HH .   55 x,, :: dEO , , 4:uz** * JOz ) . dj%/ 2 2 . J : - s!+AActsdSt|}t|tjo|dkS)z,Return True if node is the KW_ONLY sentinel.Fzdataclasses._KW_ONLY_TYPE)r rr-rInstanceqnamers r*r^r^0sH u$H8U^,, < NN   ; ;r,c t|}n#ttf$rYdSwxYwt |dddkS)z@Return True if node is an InitVar, with or without subscripting.Fr>rzInitVar)rrrrrrs r*r_r_;s^ %% M *uu 8VR ( (I 55s !$99)Dict FrozenSetListSetr*Iterator[UninferableBase | bases.Instance]c#Kd} t||}n #ttf$r tVYnwxYwt |t js tVdS|j dvr1|j tvr| VdStVdS| VdS)zInfer an instance corresponding to the type annotation represented by node. Currently has limited support for the typing module. Nr>rztyping_collections_abc) rrrrrr-r r.rKr>_INFERABLE_TYPING_TYPESinstantiate_class)rrklasss r*rrQs ETZZZ,,-- M * eU^ , , (     :0 0 0))++ + + + + +     %%'''''''s#*AAmanagerrcD|tjtt|tjt tdt|tj t tdtdS)NT)raise_on_overwrite) register_transformr r.rWr1rCrrrr?rr)rs r*registerrls  +-H  0TJJJ(   /DIII'r,)rrr r!r"r#)rrr"r2)F)rrr8r#r"rX)rrr"r#)rrr"rf)rnrwrorwr"rx)rrr>rr"r)rrrrrQr#r"rr&)rrrrr"r)rrrrr"r)rrr r!r"r#)rrr"r#)T)rrrr#r"r#)rrr"r)rrr"r#)rrrrr"r)rrr"r2)<__doc__ __future__rcollections.abcrrrrrastroidrr r astroid.builderr astroid.constr r astroid.exceptionsrrrastroid.inference_tiprastroid.managerrastroid.typingr astroid.utilrrrNodeNGrCr frozensetDATACLASSES_DECORATORSrrrLr1rWr<rBrvrrrHrrr'rrrr\r^r_rrrr,r*rs} #"""""$$$$$$(((((((((())))))))))!!!!!!////////VVVVVVVVVV//////************AAAAAAAAAA ') el *+ '# $ej 013 #>22  ID)=S     /C/C/C/Cf(-6< ) ) ) )F0&POPOPOPOhAE0>B33333(;Q<    "+/UUUUU<D(6666$)@D(((((6r,