o bq<@sdZddlZddlZddlmZddlmZmZmZm Z ddlm Z ddlm Z ddlm Z ddl mZmZdd lmZd d hZhd Zd dZddZd+ddZddZddZejeddZejeddZejed dZd,ddZddZd,dd Zd!Zd"d#Z d,d$d%Z!d,d&d'Z"e#e j$eeee#e j$eeeej#e j%e d(d)d*e#e j%ee!ee#e j$ee"edS)-z.Astroid hooks for the Python standard library.N)dedent)MANAGERUseInferenceDefault inference_tipInferenceError) arguments) exceptions)nodes)AstroidBuilder extract_node)util NamedTupletyping.NamedTuple> enum.Enum enum.IntEnum enum.IntFlagEnumIntEnumIntFlagcCsJ|tjurtzt|j|d}|tjurt|WSty$tw)Ncontext)r Uninferablernextinfer StopIterationr)nodervaluerb/var/www/secure340b-portal/env/lib/python3.10/site-packages/astroid/brain/brain_namedtuple_enum.py _infer_first+s   rcsdd fdd }|j|j}|rdd|Dni|ddd}|dd d}|r/|r/|j|fSt) Ncs<t|kr t|S|r|vrt|SdSdSN)lenrpositionZkey_nameargsrZfound_keywordsrr"_extract_namedtuple_arg_or_keyword9s  zE_find_func_form_arguments.._extract_namedtuple_arg_or_keywordcSsi|]}|j|jqSr)argr).0keywordrrr Csz-_find_func_form_arguments..rtypenamer" field_namesr )r%keywordsrr)rrr&r.namenamesrr$r_find_func_form_arguments8s   r1Fc sPzlt|\}}z |jdd}WnWtyj|s'fdd|jD}nAt|dr=t|jt r=fdd|jD}n't|drbt dd |jDrWfd d|jD}n fd d|jD}nt|shtYnwWn tt j fyyt w|p}d }t|d }|j|_|j||D]}t} || _|| _| g|j|<q|||fS)z=Specific inference function for namedtuple or Python 3 enum. , cg|]}t|jqSrrrr(constrrr [ z#infer_func_form..itemscs,g|]}t|dtjrt|djqSr) isinstancer Constrrr6rrrr8bs eltscss|] }t|tjVqdSr )r<r Tupler6rrr kz"infer_func_form..cs*g|]}t|tjrt|jdjqSr;)r<r r?rr>rr6rrrr8ls  cr4rr5r6rrrr8rr9r docstring)r1rreplacesplitAttributeErrorr>hasattrr<r:listallrrrr ClassDefparentbasesappendZ EmptyNodeattrnameZinstance_attrs) rZ base_typerenumr/r0 attributes class_nodeattrZ fake_noderrrinfer_func_formNsP          rRcCst|jt@S)zQPredicate for class inference tip :type node: ClassDef :rtype: bool )set basenamesTYPING_NAMEDTUPLE_BASENAMES)rrrr_has_namedtuple_basesrVcCs6|j}t|tjr|j|kSt|tjr|j|kSdSNF)funcr<r AttributerMNamer/)rr/rXrrr _looks_likes     r[ namedtupler/rc sRtjd|d}t|||d\}}}tj|}ttd }z t| |d| }Wn t y9d}Ynw|r@t |}ddd |D} d d fd d t|D} ttd ||| | d} | jdjd|jd<| jdjd|jd<| jdjd|jd<| jdjd|jd<|D] } | jdj| |j| <qt|gS)z4Specific inference function for namedtuple Call nodetuple)r/rJrz*import collections; collections.namedtuplerenameF, css|] }dj|dVqdS)z {arg}=None)r'Nformat)r(r'rrrr@rAz$infer_named_tuple..z[ {name} = property(lambda self: self[{index:d}], doc='Alias for field number {index:d}') c3s"|] \}}j||dVqdS))r/indexNra)r(rdr/Z field_defrrr@s  ao class %(name)s(tuple): __slots__ = () _fields = %(fields)r def _asdict(self): return self.__dict__ @classmethod def _make(cls, iterable, new=tuple.__new__, len=len): return new(cls, iterable) def _replace(self, %(replace_args)s): return self def __getnewargs__(self): return tuple(self) %(field_defs)s )r/fields field_defs replace_argsr_asdict_make_replace_fields)r rZrootrRrZCallSiteZ from_callrr rZinfer_argumentZ bool_valuer"_get_renamed_namedtuple_attributesjoin enumerater r string_buildbodylocalsiter) rrZtuple_base_namerPr/rOZ call_siterXr_rhrgfakerQrrerinfer_named_tuplesF   rvcCs|t|}t}t|D].\}}tdd|Dr.t|s.|r.|ds.|ds.||vr4d|||<||q t |S)Ncss |] }|p |dkVqdS)_N)isalnum)r(crrrr@sz5_get_renamed_namedtuple_attributes..rrwz_%d) rGrSrprHr) iskeywordisdigit startswithaddr^)r-r0seenir/rrrrns$   rncCs*td}t|||ddd}t|gS)z1 Specific inference function for enum Call node. 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)rrNr)r rRrtinstantiate_class)rrZ enum_metarPrrr infer_enums "ra 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) c Cs|jD]}|tvr q|jdkr|S|jD]\}}tdd|Dr'qg}|d}t|t j rJt|j dt j rF|j d }n|j }n t|t jrT|jg}d}t|t j r{t|jt jrvt|jjtrqt|jj}n |jj}n|j}g}|D]A}tdj|jd|j|d} d |vr| tj|jd 7} tt| |j} |j| _|D] } | g| j| j<q|| q||j|<q|S|S) z Specific inference for enums. rNcss|] }t|tj VqdSr )r<r Z AssignName)r(rrrrr@4sz#infer_enum_class..rNa class {name}({types}): @property def value(self): return {return_value} @property def name(self): return "{name}" r`)r/types return_valuerr]) rTENUM_BASE_NAMESrmr/rsr:anyZ statementr<r Assigntargetsr?Zitered AnnAssigntargetrr=strrepr as_stringrrbroINT_FLAG_ADDITION_METHODSr rrqrJ mymethodsrLr) rbasenamelocalvaluesrstmtZinferred_return_valueZ new_targetsrZclassdefrumethodrrrinfer_enum_class)sX 9         rcCsfdd|jD}tdj|jd|d}t|}tt||}|D] }|g|j |j<q$t |fS)z%Infer a subclass of typing.NamedTuplecSs g|] }t|tjr|jjqSr)r<r rrr/)r(Z annassignrrrr8qs  z1infer_typing_namedtuple_class..zU from collections import namedtuple namedtuple({typename!r}, {fields!r}) r2r+rf) rrrrbr/ror rrvrrsrt)rPrZannassigns_fieldscoderZgenerated_class_noderrrrinfer_typing_namedtuple_classns  rcCsz t|j}Wn tytw|dkrtt|jdkr$tt|jdt j t j fs2tg}|jdj D] }t|t j t j fsGtt|j dkrPt| |j dq:|jd}|rmdd|}nd}tdj||d }t||S) z$Infer a typing.NamedTuple(...) call.rr,rz({},)r2z''z namedtuple({typename}, {fields})r)rrXrrrqnamer!r%r<r Listr?r>rLrrbror rv)rrrXr0eltr+r-rrrinfer_typing_namedtuples4    rcCstdd|jDS)Ncss|] }|tvr|VqdSr )r)r(rrrrr@s  z..)rrT)clsrrrsr) predicaterWr )&__doc__ functoolsr)textwraprZastroidrrrrrrr Zastroid.builderr r r rUrrr1rRrVr[partialZ_looks_like_namedtupleZ_looks_like_enumZ_looks_like_typing_namedtuplervrnrrrrrZregister_transformCallrIrrrrsT       ?   : ( E %