5 fh]UdZddlmZddlZddlZddlmZddlmZddl m Z ddl Z ddl m Z m Z mZmZmZddlmZmZmZdd lmZdd lmZmZmZmZdd lmZd Zd ed<ddhZ ded<hdZ!ded<dZ"dZ# d>d?d"Z$d#Z%d@d$Z&ej'e&d%&Z(ej'e&d'&Z)ej'e&d(&Z* dAdBd*Z+d+Z,dCd,Z- dAdDd.Z.d/Z/dEd1Z0dAdFd2Z1dAdFd3Z2 dAdBd4Z3dGd6Z4dHd9Z5dId=Z6dS)Jz.Astroid hooks for the Python standard library.) annotationsN)Iterator)dedent)Final) argumentsbases inference_tipnodesutil)AstroidBuilder_extract_single_node extract_node)InferenceContext)AstroidTypeErrorAstroidValueErrorInferenceErrorUseInferenceDefault)AstroidManager enum.Enumz Final[str] ENUM_QNAMEtyping.NamedTupletyping_extensions.NamedTuplerTYPING_NAMEDTUPLE_QUALIFIED> NamedTuplerrTYPING_NAMEDTUPLE_BASENAMESct|tjrt t ||}n#t $r }t|d}~wwxYwt|tjrt|S)Ncontext) isinstancer UninferableBasernextinfer StopIterationr)nodervalueexcs Y/var/www/equiseq/venv/lib/python3.11/site-packages/astroid/brain/brain_namedtuple_enum.py _infer_firstr('s$,--"!!&TZZZ0011 &&&#%&%-..$!### Ls#A AAAc dfd }|j|j}|r d|Dni|dd}|dd}|r |r |j|fSt) Nct|krt|S|r|vrt|SdSdSN)lenr()positionkey_nameargsrfound_keywordss r'"_extract_namedtuple_arg_or_keywordzE_find_func_form_arguments.._extract_namedtuple_arg_or_keyword4sd t99x  X88 8  CN22x 8'BB B C C22c(i|]}|j|jS)argr%).0keywords r' z-_find_func_form_arguments..?s<<<gm<<z#infer_func_form..as3"""?D UG44:"""r2itemscg|]=}t|dtj"t|dj>Sr)rr Constr(r%rMs r'rOz#infer_func_form..jsP"""!%eAh <<"$U1Xw77="""r2eltsc3JK|]}t|tjVdSr+)rr Tuple)r6rNs r' z"infer_func_form..ss.RRe:eU[99RRRRRRr2cg|]<}t|tjt|jdj=SrR)rr rVr(rTr%rMs r'rOz#infer_func_form..tsN&&& %)%==&(A@@F&&&r2c:g|]}t|jSr4rLrMs r'rOz#infer_func_form..zs3&&&CHL88>&&&r2Nc,g|]}t|Sr4)strr6attrs r'rOz#infer_func_form..s777Dc$ii777r2cg|]}d|v| S)rJr4r\s r'rOz#infer_func_form..sAAA4D$r2 Uninferable)lineno col_offset end_linenoend_col_offsetparent)rbody decorators)r?r%replacesplitAttributeError_get_namedtuple_fieldsrrThasattrrrPlistallrrr ClassDefr`rarbrcUnknownrdpostinit EmptyNodeattrnameinstance_attrs) r$rArrDr=r> attributesr&fields fields_node class_noder] fake_nodes ` r'infer_func_formryJs,+/g>> e( 2$)K$7$7S$A$A$G$G$I$IJJ& 2& 2& 2$ 2 /55".v"6"6K""""HSHX"""J5'**2z%+t/L/L2""""%*["""JJ UF++2RRuzRRRRR &&&&).&&& &&&&LQJ&&& )c1!2(c1M& 2N N ++++!s*+ 877J777 BA:AAAJ  =D {??*} J J  66O%% % ! +4+ !$'' tZ ''sAD;-AD; D7CD2-D;2D77D;;E EEc:t|jtzS)zRPredicate for class inference tip. :type node: ClassDef :rtype: bool )set basenamesrr$s r'_has_namedtuple_baser~s t~  !< <.s(AAs]]]AAAAAAr2z[ {name} = property(lambda self: self[{index:d}], doc='Alias for field number {index:d}') c3LK|]\}}||VdS))r=indexN)format)r6rr= field_defs r'rWz$infer_named_tuple..sN E4 d%00r2z class z*(tuple): __slots__ = () _fields = z def _asdict(self): return self.__dict__ @classmethod def _make(cls, iterable, new=tuple.__new__, len=len): return new(cls, iterable) def _replace(self, zP): return self def __getnewargs__(self): return tuple(self) z _asdict_make_replace_fields)r rrootryrCallSite from_callrr!r"r#rinfer_argumentr bool_value_check_namedtuple_attributesrrr[rjoin enumerater r string_buildrelocalsiter)r$rtuple_base_namerwr=rt call_sitererr& replace_args field_defsfaker]rs @r'infer_named_tuplersS  99;;     +O$3 ow$$$ Jj",,T7,CCI D E ED/DJJLL!! ///$'''Q./  $ $T8W5R@P@R@R S S  *,,  M *F1$ FKK EEE!-#c((":;;D FFF!.3s88";<<#EF99AAjAAAAAL 2$Z00J .** + + 8 8      %        D"$(9Q<#6y#AJi !%1!4W!=Jg$(IaL$7 $CJj!#'9Q<#6y#AJi <<"&)A,"5d"; $   sO6!B B8"B33B8.s3::AIIKK+18::::::r2rrz_%d) rlr{rrmr7 iskeywordisdigit startswithaddr)r;r>seenir=s r'"_get_renamed_namedtuple_attributesr s   E 55D[))  4::T::::: ! && ! !Aw   ! s##  ! t||qyE!H  <<r2c t|}|rt|}|g|RD]u}t|tst d|st dd|ztj|rt d|vt}|D]V}| dr|st d|||vrt d|| |W|S)Nz*Type names and field names must be stringsz(Type names and field names must be validz identifiers: z0Type names and field names cannot be a keyword: rz-Field names cannot start with an underscore: z"Encountered duplicate field name: ) rrrr[r isidentifierrr7rr{rr)r9rtrr=rs r'rrs^z""J D7 CC 'J''  $$$ Q"#OPP P  "" #:=UT=U=UU   T " " #K4KK   55D ??3    #HHH  4<<#$Q$Q$QRR R  r2Iterator[bases.Instance]cR |j|}n!#ttf$r }t|d}~wwxYwt d|Dstt d}t||g|dd}t| gS)z/Specific inference function for enum Call node.Nc3K|];}t|tjo|tkV.FsT  4((GTZZ\\Z-Gr2a class EnumMeta(object): 'docstring' def __call__(self, node): class EnumAttribute(object): name = '' value = 0 return EnumAttribute() def __iter__(self): class EnumAttribute(object): name = '' value = 0 return [EnumAttribute()] def __reversed__(self): class EnumAttribute(object): name = '' value = 0 return (EnumAttribute, ) def __next__(self): return next(iter(self)) def __getitem__(self, attr): class Value(object): @property def name(self): return '' @property def value(self): return attr return Value() __members__ = [''] T)rrDr) rr"rr#ranyr ryrinstantiate_class)r$rinferredr& enum_metarws r' infer_enumr<s +9??7++ M *+++!s*+   ""!$ !!ID! {G$OOOPQRJ --//0 1 11s;6;a def __or__(self, other): return {name}(self.value | other.value) def __and__(self, other): return {name}(self.value & other.value) def __xor__(self, other): return {name}(self.value ^ other.value) def __add__(self, other): return {name}(self.value + other.value) def __div__(self, other): return {name}(self.value / other.value) def __invert__(self): return {name}(~self.value) def __mul__(self, other): return {name}(self.value * other.value) nodes.ClassDefc d|DD]}|jdkrni}t}|jD]\}}t d|Ds|dkr&|d}t|tj rMt|j dtj r |j d }n,|j }n$t|tjr |jg}nd}|j~t|jtjrFt|jjt$rt'|jj}n&|jj}n|j}g} |D]D} t| tjr|| jt/d| jd|j| } d |vr#| t6| j z } t9t;d | | j} | j| _| D]} | g| j| j<| !| "|j?| ||<F| |j|<tj#||j$|j%|j&|j'g|jd<tj#||j$|j%|j&|j'(fd|Dg|jd<d|vrNt/d}t9t;|d}|g|jd<|S)zSpecific inference for enums.c3.K|]}|jD]}|VdSr+)r|)r6clsbs r'rWz#infer_enum_class..s1DD3cmDDQDDDDDDDr2rDc3LK|]}t|tj V dSr+)rr AssignName)r6r%s r'rWz#infer_enum_class..s2PP 5%*:;;;PPPPPPr2_ignore_rNa class {name}({types}): @property def value(self): return {return_value} @property def _value_(self): return {return_value} @property def name(self): return "{name}" @property def _name_(self): return "{name}" r)r=types return_valueIntFlagrF)apply_transformsrdr`rarbrc_value2member_map_c g|]N\}}tj|tj|j|j|j|j|jfOS))rdr)r rSrr=r`rarbrc)r6kvmemberss r'rOz$infer_enum_class..st   AqK'222J& x#$<#$<'('7     r2 __members__r=zY @property def name(self): return '' ))mrorr=r{rrPr statementrr AssigntargetsrVitered AnnAssigntargetr%rSr[repr as_stringStarredrrrrr|INT_FLAG_ADDITION_METHODSr rrrd mymethodsappendrDictr`rarbrcrp)r$basenamedunder_members target_nameslocalvaluesstmtrinferred_return_value new_targetsrclassdefrmethodcodename_dynamicclassattrrs @r'infer_enum_classrs5DD$((**DDDJJ 99;; v % % Euu ![..00H -H -ME6PPPPPPP J&&!9&&((D$ -- dl1ou{;;+"l1o4466GG"lGGD%/22 ;-$( !z%dj%+66C!$*"2C88A04TZ5E0F0F--04 0@--,0J,@,@,B,B)K!* -* -fem44  ---!F#["ii77%:*((  9 @ @fk @ R RRH%"$$u,x((6%m "nn..88F06xDK ,,""4#9#9#;#;<<<:%(,u%%!,DK   J{??#2    -  ()*;.         +0022      '.Y M"  % %D%3>3C3C$D$D$Q$QRV$W$W% !$9"9DK  Kr2cd|jD}td|jd|}t |} t t||}n$#t$r}t|||d}~wwxYw| D]}|g|j |j<|jD]m}t|tjr'|jD]} | j} |j | |j | <Ct|tjr|g|j |j<nt#|fS)z&Infer a subclass of typing.NamedTuple.cZg|](}t|tj|jj)Sr4)rr rrr=)r6 annassigns r'rOz1infer_typing_namedtuple_class..s@  i 1 1r2zU from collections import namedtuple namedtuple({typename!r}, {fields!r}) rI)r9ru)r$rN)rerrr=rrr!rr#rrrrr rrrnr) rwrannassigns_fieldsrr$generated_class_noderr body_noderr]s r'infer_typing_namedtuple_classrs#      fjochh7H.I.IfJJ    D@#$5dG$D$DEE @@@$888a?@&&((<<4:8#FK00_FF i . . F#+ L L{4>4Ed4K$+D11 L 5> 2 2 F;D+ '  7 %' ( ((sA<< BBBcJtd}||S)z Starting with python3.9, NamedTuple is a function of the typing module. The class NamedTuple is build dynamically through a call to `type` during initialization of the `_NamedTuple` variable. zD from typing import _NamedTuple _NamedTuple )rr")r$rklasss r' infer_typing_namedtuple_functionr5s,    E ;;w  r2c t|j}n!#ttf$r }t |d}~wwxYw|tvrt t|j dkrt t|j dtj tj fst t||S)z$Infer a typing.NamedTuple(...) call.Nr:)r!rr"rr#rrrr,r/rr ListrVr)r$rrr&s r'infer_typing_namedtuplerDs +DIOO%%&& M *+++!s*+ zz||666!! 49~~!! dilUZ$= > >"!! T7 + ++s&)AAAr[cg}d} t|jd}n,#ttf$r }t |d}~wt $rYnwxYw|s`|jD]X}|jdkrK t|j }n!#ttf$r }t |d}~wwxYwnYt|tj st |j D]}t|tjr(||Dt|tjtjfst t'|j dkrt ||j d|rdd|d}nd }|S) zGet and return fields of a NamedTuple in code-as-a-string. Because the fields are represented in their code form we can extract a node from them later on. Nr:r;rr(rIz,))r!r/r"rr#r IndexErrorr<r5r%rr BaseContainerrTrSrrrrVr,r)r$r> containerr& keyword_nodeeltr;s r'rjrj[s EI 1++--.. M *+++!s*       M  L=007 $\%7%=%=%?%? @ @II& 6777-367 1 i!4 5 5"!!~.. c5; ' '  LL ) ) ) # EK899 &% % sx==A  % % SXa[**,,---- -#((5//---  s3,3AA  AA6&BB;.B66B;rastroid.ClassDefc,|dS)z,Return whether cls is a subclass of an Enum.r) is_subtype_of)rs r'_is_enum_subclassrs   [ ) ))r2managerrNonec|tjttt |tjtt t|tjtt|tjttt|tj ttd|tjttt dS)N) predicatecf|jdko&t|dddkS)Nrr=typing)r=getattrrr}s r'zregister..s1TY,.; DIIKK . .( :r2)register_transformr Callr r_looks_like_namedtupler_looks_like_enumrnrrrr~ FunctionDefrr_looks_like_typing_namedtuple)r s r'registerrs  M"3446L uz=+D+DFVWWW  (4E  344   677 ; ;   -..%r2r) r$r@rArBrrCrDrErFrG)rFrEr+)r$r@rrCrFr)F)r$r@rrCrFr)r$rrFr)rrC)r$r@rFr[)rrrFrE)r rrFr )7__doc__ __future__r functoolsr7collections.abcrtextwraprrrastroidrrr r r astroid.builderr r rastroid.contextrastroid.exceptionsrrrrastroid.managerrr__annotations__rrr(r?ryr~rpartialrrrrrrrrrrrrrjrrr4r2r'r$sK 544""""""$$$$$$@@@@@@@@@@@@@@NNNNNNNNNN,,,,,, +*****$ $$$$"&&&&      4(, ](](](](](@===+*;\JJJ$9$[v>>> 1 1+L Q Q Q:>FFFFFR"D:>3232323232l$MMMM`)))))B      :>,,,,,.''''T**** r2