eh]UdZddlmZddlZddlZddlmZddlmZddl m Z ddl Z ddl m Z m Z mZmZmZddlmZmZmZdd lmZdd lmZmZmZmZdd lmZd Zd ed<ddhZ ded<hdZ!ded<dZ"dZ# d* d+dZ$dZ%d,dZ&ejNe&dZ(ejNe&dZ)ejNe&dZ* d- d.dZ+dZ,d/d Z- d- d0d!Z.d"Z/d1d#Z0d-d2d$Z1d-d2d%Z2 d- d.d&Z3d3d'Z4d4d(Z5d5d)Z6y)6z.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 |j |}t|tjr t|S#t $r }t|d}~wwxYw)Ncontext) isinstancer UninferableBasernextinfer StopIterationr)nodervalueexcs E/usr/lib/python3/dist-packages/astroid/brain/brain_namedtuple_enum.py _infer_firstr('sl$,,-!!&TZZZ01%--.!## L &#%&sA## A8,A33A8c dfd }|j|j}|r%|Dcic]}|j|jc}ni|dd}|dd}|r|r|j|fSt cc}w)Nclt|kDrt|S|r|vrt|SyyN)lenr()positionkey_nameargsrfound_keywordss r'"_extract_namedtuple_arg_or_keywordzE_find_func_form_arguments.._extract_namedtuple_arg_or_keyword4sE t9x X8 8 N2x 8'B B38rtypename)r-r. field_namesr+)r/keywordsargr%r) r$rr1r6keywordnamenamesr/r0s ` @@r'_find_func_form_argumentsr;3sC 99D}}H@H8<gmm #c  t||\}} |jjddj}|s|D cgc] } t'| }} |D cgc] } d| vs|  }} |xsd}tj(||j*|j,|j.|j0tj2 } |j4| _| j7|gd |D]4} tj8} | | _| | _| g| j<| <6| ||fS#t$r}|sPg}t |}|rt |} | jD cgc]} t| |jncc} w}} nAt|drnt|jtrT|jD cgc]8} t| dtjrt| d|j:ncc} w}} nt|drtd|jDr[|jD cgc]?} t| tj r#t| jd|jAncc} w}} n;|jD cgc]} t| |jncc} w}} nt||st|Yd}~d}~wwxYw#tt"f$r }t$|d}~wwxYwcc} wcc} w) zz"infer_func_form..ssRe:eU[[9Rs$&N Uninferable)lineno col_offset end_linenoend_col_offsetparent)rbody decorators)r;r%replacesplitAttributeError_get_namedtuple_fieldsrr@r(hasattrrr?listr ConstallrBrrstrClassDefrGrHrIrJUnknownrKpostinit EmptyNodeattrnameinstance_attrs)r$ base_typerenumr9r: attributesr&fields fields_noderDattr class_node fake_nodes r'infer_func_formreJs,+/g> e( 2$)KK$7$7S$A$G$G$IJV ,67Dc$i7 7$.A4D$AJA  =D {{????**}} J J  6OO% % ! +4+ !!$' 6 tZ ''[& 2 /5".v"6KHSHXHX"?D UG4::""J"5'*z%++t/L&+[["!%eAh <%U1Xw7==""J" UF+RuzzRR*/& %)%=)A@FF&& &MRJJ&CHL8>>&& &)c1!(c1M& 2N N ++!s*+ 8Bs}K*D(K1 K6$K6( K2+K E;:>K 8=G65?K 4AI98K J-,K K KKK."K))K.c:t|jtzS)zRPredicate for class inference tip. :type node: ClassDef :rtype: bool )set basenamesrr$s r'_has_namedtuple_baserjs t~~ !< <.sAsuE]Asz[ {name} = property(lambda self: self[{index:d}], doc='Alias for field number {index:d}') c3JK|]\}}j||yw))r9indexN)format)rCr|r9 field_defs r'rEz$infer_named_tuple..s- E4 d%0s #z 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 rorootrerCallSite from_callrr!r"r#rinfer_argumentr bool_value_check_namedtuple_attributesrrrVrjoin enumerater r string_buildrLlocalsiter)r$rtuple_base_namercr9r_ call_sitermervr& replace_args field_defsfakerbr~s @r'infer_named_tuplers  99;   +O$3 ow$ Jj"",,T7,CI D ED/DJJL!  $ $T8W5R@P@R S *,  F1$ FK 99AjAAL 2$Z0J .* + 8 8  f n %~&    D"$(99Q<#6#6y#AJi !%1!4!4W!=Jg$(IIaL$7$7 $CJj!#'99Q<#6#6y#AJi <"&))A,"5"5d"; $<  c /$'Q./ M * E!-#c(":;D F!.3s8";<#EFsN+H6H3< I  H0 H++H03II JI,, J8JJc<t|}t}t|D]o\}}td|Dr?t j |s*|r(|dj s|jds||vrd|z||<|j|qt|S)Nc3JK|]}|jxs|dk(yw)_N)isalnum)rCcs r'rEz5_get_renamed_namedtuple_attributes..s":AIIK+18+:s!#rrz_%d) rSrgrrUr8 iskeywordisdigit startswithaddru)r5r:seenir9s r'"_get_renamed_namedtuple_attributesr s  E 5D[) 4:T::  &Aw s#t|qyE!H   <r2ct|}|r t|}|g|D]a}t|ts t d|j st dd|ztj|sUt d|t}|D]F}|jdr|st d|||vrt d||j|H|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: ) rurrrVr isidentifierrr8rrgrr)r3r_rvr9rs r'rrsz"J 7 C 'J' $$"#OP P  "#:}TH=UU    T "#B4(K   5D ??3 #?xH  4<#&H$QR R  r2c |jj|}t d|Dstt d}t||g|dd}t|jgS#ttf$r }t|d}~wwxYw)z/Specific inference function for enum Call node.Nc3K|]7}t|tjxr|jtk(9ywr+)rr rWqnamer)rCitems r'rEzinfer_enum..Fs6  4(GTZZ\Z-GGs=?a 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)rr^r) rmr"rr#ranyr rerinstantiate_class)r$rinferredr& enum_metarcs r' infer_enumr<s +99??7+  "!$ !ID! {G$OPQRJ --/0 11Y M *+!s*+sA--B<BBa 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) cf d|jDD]}|jjdk(r|Si}t}|jj D]\}}t d|Ds|dk(r|dj}t|tjrRt|jdtjr|jdj}n6|j}n)t|tjr|jg}nd}|j t|j tj"r[t|j j t$r t'|j j }n1|j j }n|j j)}g} |D]C} t| tj*r|j-| jt/dj1| jdj3|j4| } d |vr#| t6j1| j z } t9t;d j=| | j} | j>| _|jAD]} | g| j| j<| jC| jE|j ?| ||<F| |j|<tjF||jH|jJ|jL|jNg|jd<tjF||jH|jJ|jL|jN}|jQ|j Dcgc]i\}}tj"||tjR|j||jH|jJ|jL|jNfkc}}|g|jd<d|vr@t/d}t9t;j=|d}|g|jd<|S|Scc}}w)zSpecific inference for enums.c3BK|]}|jD]}|ywr+)rh)rCclsbs r'rEz#infer_enum_class..s D3cmmDQDQDsr^c3RK|]}t|tj !ywr+)rr AssignName)rCr%s r'rEz#infer_enum_class..s P 5%*:*:;;Ps%'_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}" rw)r9types return_valueIntFlagrrF)apply_transforms)rKrGrHrIrJ_value2member_map_)rK __members__r9zY @property def name(self): return '' )*mrorr9rgrr?r statementrr AssigntargetsrBitered AnnAssigntargetr%rTrVrepr as_stringStarredrrr}rrhINT_FLAG_ADDITION_METHODSr rrrK mymethodsappendrDictrGrHrIrJrYro)r$basenamedunder_members target_nameslocalvaluesstmtrinferred_return_value new_targetsrclassdefrmethodmemberskvcodename_dynamicclassattrs r'infer_enum_classrsD$((*DJ 99;  v % P KOu ![[..0H -ME6PPPJ&!9&&(D$ -dll1ou{{;"ll1o446G"llGD%//2;;-$( !zz%djj%++6!$**"2"2C804TZZ5E5E0F-04 0@0@-,0JJ,@,@,B)K!* -femm4  -!F#[["ii7%:*(  9 @ @fkk @ RRH%"$u,x(6%mm "nn.8F06xDKK ,8""4#9#9#;<::%(,u%U* -V"-DKK QH -X JJ{{????#22  -  ()**;;..   +002 AqKK'2JJ& xx#$<<#$<<'('7'7    '.Y M"  %D%3>3C$D$Q$QRV$W% !$9"9DKK  KWJV KQ sA.R- c.|jDcgc]2}t|tjr|jj 4}}t dj|j dj|}t|} tt||}|jD]}|g|j|j <|jD]} t| tj r:| j"D]*} | j } |j| |j| <,Wt| tj$sr| g|j| j <t'|fScc}w#t$r}t|||d}~wwxYw)z&Infer a subclass of typing.NamedTuple.zU from collections import namedtuple namedtuple({typename!r}, {fields!r}) r=)r3r`)r$rN)rLrr rrr9rr}rrr!rr#rrrrrrWr) rcr annassignannassigns_fieldsrr$generated_class_noderr body_noderrbs r'infer_typing_namedtuple_classrs $  i 1      fjoochh7H.IfJ   D@#$5dG$DE&&(<4:8##FKK0< __F i .#++ L{{4>4E4Ed4K$++D1 L 5>> 2;D+ ' '  7 F %' ((7 @$8a?@s7E3 E88 FFFc:td}|j|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|jj}|j tvrt t|jdk7rt t|jdtjtjfst t||S#ttf$r }t |d}~wwxYw)z$Infer a typing.NamedTuple(...) call.Nr4)r!rmr"rr#rrrr,r/rr ListrBr)r$rrmr&s r'infer_typing_namedtuplerDs +DIIOO%& zz|66!! 499~!! diilUZZ$= >!! T7 ++ M *+!s*+s#B((C7B>>Cczg}d} t|jdj}|sF|jD]7}|jdk(s t|jj}nt|tjst |jD]}t|tjr |j|j!=t|tj"tj$fst t'|jdk7rt |j|jdj!|rddj)|d}|Sd }|S#ttf$r }t |d}~wt $rYowxYw#ttf$r }t |d}~wwxYw) 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. Nr4r5rr(r=z,))r!r/r"rr#r IndexErrorr6r7r%rr BaseContainerr@rTrrrrBr,r)r$r: containerr& keyword_nodeeltr5s r'rQrQ[s EI 1++-.  MM L=07 $\%7%7%=%=%? @I   i!4!4 5!!~~. c5;; ' LL ) # EKK89% % sxx=A % % SXXa[**,-. #((5/*"-   ; M *+!s*    ' 67-367s5&E6#F6FF  FFF:.F55F:c$|jdS)z,Return whether cls is a subclass of an Enum.r) is_subtype_of)rs r'_is_enum_subclassrs   [ ))r2cF|jtjttt |jtjtt t|jtjtt|jtjttt|jtjttd|jtjttt y)N) predicatec`|jdk(xrt|jdddk(S)Nrr9typing)r9getattrrris r'zregister..s.TYY,.; DIIK .( :r2)register_transformr Callr r_looks_like_namedtupler_looks_like_enumrWrrrrj FunctionDefrr_looks_like_typing_namedtuple)managers r'registerrs  M"346L uzz=+DFVW  (4E  34   67 ;   -.%r2rl) r$ nodes.Callr]zlist[nodes.NodeNG]rInferenceContext | Noner^boolreturnz%tuple[nodes.ClassDef, str, list[str]])rrr+)r$rrrrzIterator[nodes.ClassDef])F)r$rrrrzIterator[bases.Instance])r$nodes.ClassDefrr)rr)r$rrrV)rzastroid.ClassDefrr)rrrNone)7__doc__ __future__r functoolsr8collections.abcrtextwraprrrastroidrrr r r astroid.builderr r rastroid.contextrastroid.exceptionsrrrrastroid.managerrr__annotations__rrr(r;rerjrppartialrrrrrrrrrrrrrQrrryr2r'rs 5"$@@NN, +$ J$"&U&U  4(, ]( ](!](%](  ]( + ](@=+**;\J$9$$[v> 1 1 1+L Q:>F F6FFR"D:>32 3263232l$M`)B  :>, ,6,,.'T* r2