+D#O " U d Z ddlmZ ddlZddlZddlZddlmZ ddlm Z ddl
Z
ddl
m Z m Z m
Z
mZmZ ddlmZmZmZ ddlmZ dd lmZmZmZmZmZ dd
l m Z ej d k rdd l m Z ndd l!m Z h d
Z"dZ#de$d< ddhZ%de$d< h dZ&de$d< d Z'd Z( d>d?d$Z)d% Z*d@d&Z+ ej, e+d'( Z- ej, e+d)( Z. ej, e+d*( Z/ dAdBd,Z0d- Z1dCd.Z2 dAdDd0Z3d1Z4dEd3Z5dAdFd4Z6dAdFd5Z7 dAdBd6Z8dGd8Z9dHd;Z: e ; ej< e
e0 e- e ; ej< e
e3 e. e ; ej= e5e:< e ; ej= e
e6 e* e ; ej> e
e7 d= e ; ej< e
e8 e/ dS )Iz.Astroid hooks for the Python standard library. ) annotationsN)Iterator)dedent) argumentsbases
inference_tipnodesutil)AstroidBuilder_extract_single_node extract_node)InferenceContext)AstroidTypeErrorAstroidValueErrorInferenceErrorMroErrorUseInferenceDefault)AstroidManager) )Final> enum.IntEnum enum.IntFlagEnumIntEnumIntFlag enum.Enumr z
Final[str]
ENUM_QNAMEtyping.NamedTuple typing_extensions.NamedTupler TYPING_NAMEDTUPLE_QUALIFIED>
NamedTupler r TYPING_NAMEDTUPLE_BASENAMESc t | t j rt t | | }n# t $ r
}t |d }~ww xY wt |t j rt |S )Ncontext)
isinstancer
UninferableBaser nextinfer
StopIterationr )noder& valueexcs u/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/astroid/brain/brain_namedtuple_enum.py _infer_firstr0 6 s $,-- "!!&TZZZ0011 & & & #%&%-.. $!### Ls #A
A AA c dfd }| j | j }|r d |D ni |dd } |dd }|r |r |j |fS t ) Nc t | k rt | S |r|v rt | S d S d S N)lenr0 )positionkey_nameargsr& found_keywordss r/ "_extract_namedtuple_arg_or_keywordzE_find_func_form_arguments.._extract_namedtuple_arg_or_keywordC sd t99x X88 8 CN22 x 8'BB B C C22 c ( i | ]}|j |j S )argr- ).0keywords r/
z-_find_func_form_arguments..N s <<<gm<< constr& s r/
z#infer_func_form..p s3 " " "?D UG44:" " "r: itemsc g | ]=}t |d t j "t |d j >S r )r' r Constr0 r- rU s r/ rW z#infer_func_form..y sP " " " !%eAh <<"$U1Xw77=" " "r: eltsc 3 J K | ] }t |t j V d S r3 )r' r Tuple)r> rV s r/ z"infer_func_form.. s. RRe:eU[99RRRRRRr: c g | ]<}t |t j t |j d j =S rZ )r' r r^ r0 r\ r- rU s r/ rW z#infer_func_form.. sN & & & % )% = =& (A @ @ F& & &r: c : g | ]}t | j S r< rT rU s r/ rW z#infer_func_form.. s3 & & &CHL 8 8 >& & &r: Nc , g | ]}t | S r< )strr> attrs r/ rW z#infer_func_form.. s 777Dc$ii777r: c g | ]}d |v| S )rR r< rd s r/ rW z#infer_func_form.. s AAA4D $ r: Uninferable)r body
decorators)rG r- replacesplitAttributeError_get_namedtuple_fieldsr
r\ hasattrr' rX listallr r r ClassDefparentpostinit EmptyNodeattrnameinstance_attrs)
r, rI r& rL rE rF
attributesr. fields fields_node
class_nodere fake_nodes
` r/ infer_func_formr| Y s ,+/g>> e( 2$)K$7$7S$A$A$G$G$I$IJJ & 2 & 2 & 2 $
2
/55 ".v"6"6K" " " "HSHX" " "J 5'** 2z%+t/L/L 2" " " "%*[" " "JJ
UF++ 2 RRuzRRRRR & & & &).& & &
& & & &LQJ& & &
)c1! 2(c1M& 2N
N + + + +!s*+ 877J777
BA:AAAJ =D %%J J
6 6O%% % ! +4+
!$'' tZ ''sA D; -A D;
D7C D2-D; 2D77D; ;E EEc : t | j t z S )zRPredicate for class inference tip.
:type node: ClassDef
:rtype: bool
)set basenamesr# r, s r/ _has_namedtuple_baser s t~ !< < r= s r/ r_ z$infer_named_tuple.. s( A As]]] A A A A A Ar: z[ {name} = property(lambda self: self[{index:d}], doc='Alias for field number {index:d}')
c 3 L K | ] \ }} || V dS ))rE indexN)format)r> r rE field_defs r/ r_ z$infer_named_tuple.. sN E4 d%00 r: 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
r _asdict_make_replace_fields)r r rootr| r CallSite from_callr
r) r* r+ r infer_argument
bool_value _check_namedtuple_attributesr r rc r join enumerater r string_buildrh localsiter)r, r& tuple_base_namerz rE rw call_siter er r. replace_args
field_defsfakere r s @r/ infer_named_tupler s( ,1:7499;;+W+W+W*XO#2 ow$ $ $ Jj ",,T7,CCI D E ED/DJJLL!! / / / $'''Q./i..tXwGGHHSSUU M * F1$
FKK
E E E!-#c((":;;D F F F!.3s88";<<#EF 99 A Aj A A AAAL 2 $Z00 J .** + + 8 8 % D" $(9Q<#6y#AJi !%1 !4W!=Jg $(IaL$7
$CJj!#'9Q<#6y#AJi < <"&)A,"5d";
$ sN 2!B
B4 B//B486C/ /DD D
E7% E
E7 E22E7c t | }t }t | D ]\ }}t d |D rIt j | s5|r3|d s| d s||v rd|z ||< | | t | S )Nc 3 J K | ] }| p|d k V dS )_N)isalnum)r> cs r/ r_ z5_get_renamed_namedtuple_attributes.. s3 ::AIIKK+18::::::r: r r z_%d)
ro r~ r rp r? iskeywordisdigit
startswithaddr )rC rF seenirE s r/ "_get_renamed_namedtuple_attributesr s E 55D [))
4::T::::: ! && ! ! Aw !
s## ! t|| qyE!H <<r: c t | }|rt | }| f|z D ]u}t |t st d | st
dd|z t j | rt
d| vt }|D ]V}|
d r|st
d| ||v rt
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: r z-Field names cannot start with an underscore: z"Encountered duplicate field name: ) r r r' rc r isidentifierr r? r r~ r r )rA rw r rE r s r/ r r sZ z""J
D7
CC
j(
$$$ Q"#OPP P "" #:=UT=U=UU
T " " #K4KK
55D ??3 #HHH
4<<#$Q$Q$QRR R r: Iterator[bases.Instance]c R | j | }n!# t t f$ r
}t |d}~ww xY wt d |D st t
d }t | |g|d d }t | g S )z/Specific inference function for enum Call node.Nc 3 K | ];}t |t j o | t k V items r/ r_ z infer_enum..C sT 4 ((GTZZ\\Z-G r: 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)r& rL r )
r r* r r+ r anyr r| r instantiate_class)r, r& inferredr. enum_metarz s r/
infer_enumr 9 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 | D D ]}| j dk r ni }t }| j D ]\ }}t
d |D r |d d }t |t j
rMt |j d t j r |j d
}n,|j }n$t |t j r |j g}nd}|j ~t |j t j rFt |j j t$ rt' |j j }n&|j j }n|j }g } |D ];}
t |
t j r | |
j t/ d |
j d | j |
} d |v r#| t6 |
j z
} t9 t; d
| |
j } |
j | _ | D ]}
|
g| j |
j < | ! | " | ||< =| | j |< t j# | g| j d< t j# | $ fd| D g| j d< d|vrNt/ d }t9 t; | d }|g| j d< | S )z Specific inference for enums.c 3 . K | ]}|j D ]}|V d S r3 )r )r> clsbs r/ r_ z#infer_enum_class.. s1 DD3cmDDQDDDDDDDr: rL c 3 L K | ] }t |t j V d S r3 )r' r
AssignName)r> r- s r/ r_ z#infer_enum_class.. s2 OOuz%)9:::OOOOOOr: r T)futureNa
class {name}({types}):
@property
def value(self):
return {return_value}
@property
def name(self):
return "{name}"
r )rE types return_valuer r F)apply_transformsrr _value2member_map_c x g | ]6\ }}t j | t j |j f7S )r )r r[ r rE )r> kvmemberss r/ rW z$infer_enum_class.. sR
Aq Qw///AF71S1S1ST
r: __members__rE zY
@property
def name(self):
return ''
)%mror rE r~ r rX r statementr' r Assigntargetsr^ itered AnnAssigntargetr- r[ rc repr as_stringStarredr r r r r INT_FLAG_ADDITION_METHODSr r r rr mymethodsappendr Dictrs )r, basenamedunder_members target_nameslocalvaluesstmtr inferred_return_value new_targetsr classdefr methodcodename_dynamicclassattrr s @r/ infer_enum_classr s DD$((**DDD g g 99;; v % % Euu ![ . . 0 0 = - = -ME6OOOOOOO
!9&&d&33D$ --
dl1ou{;; + "l1o 4 4 6 6GG "lGGD%/22
;-$( !z% dj%+66 C!$*"2C88 A04TZ5E0F0F--04
0@--,0J,@,@,B,B) K! "
- "
- fem44 ---! F #[ "ii 7 7%: ((
9 @ @fk @ R RRH%"$$u ,x((6 %m "nn . . 8 8F06xDK ,,""4#9#9#;#;<<<(,u%%!,DK .3Zt-D-D-D,E ()*D)))
* 0 0 2 2
'.Y M" % % D %3>3C3C$D$D$Q$QRV$W$W% ! $9"9DK
Kr: c d | j D }t d | j d | }t | } t
t || }n$# t $ r}t || |d}~ww xY w|
D ]}|g|j |j < | j D ]m}t |t j r'|j D ] } | j }
| j |
|j |
< Ct |t j r|g|j |j < nt# |f S )z&Infer a subclass of typing.NamedTuple.c Z g | ](}t |t j |j j )S r< )r' r r r rE )r> annassigns r/ rW z1infer_typing_namedtuple_class.. s@ i 1 1 r: zU
from collections import namedtuple
namedtuple({typename!r}, {fields!r})
rQ )rA rx )r, r& N)rh r r rE r r
r) r r+ r r r r' r r r rq r ) rz r& annassigns_fieldsr r, generated_class_noder r body_noder re s r/ infer_typing_namedtuple_classr s #
fjochh7H.I.IfJJ D@ #$5dG$D$D E E @ @ @ $888a?@ &&(( < <4:8 #FK00 _ F F i . . F#+
L
L {4>4Ed4K$+D11
L 5>
2
2 F;D+ ' 7 %' ( ((s A< <
B BB c J t d }| | 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
)r
r* )r, r& klasss r/ infer_typing_namedtuple_functionr s,
E ;;w r: c t | j }n!# t t f$ r
}t
|d}~ww xY w| t vrt
t | j dk rt
t | j d t j t j
f st
t | | S )z$Infer a typing.NamedTuple(...) call.N rB )r) r r* r r+ r r r! r4 r7 r' r Listr^ r )r, r& r r. s r/ infer_typing_namedtupler s +DIOO%%&& M * + + +!s*+ zz||666!!
49~~!! dilUZ$= > > "!! T7 + ++s &) AAArc c g }d} t | j d }n,# t t f$ r
}t
|d}~wt $ r Y nw xY w|s`| j D ]X}|j dk rK t |j }n!# t t f$ r
}t
|d}~ww xY w nYt |t j st
|j
D ]}t |t j r(| | Dt |t j t j f st
t' |j
dk rt
| |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.
NrB rC r r (rQ z,) )r) r7 r* r r+ r
IndexErrorrD r= r- r' r
BaseContainerr\ r[ r r r r^ r4 r )r, rF containerr. keyword_nodeeltrC s r/ rm rm 5 s
EI
1 ++--.. M * + + +!s*
M L =007 $\%7%=%=%?%? @ @II&
6 7 7 7-367 1 i!4 5 5 "!!~ . . c5; ' ' LL ) ) ) #
EK 899 &% % sx==A % %
SXa[**,,---- -#((5//--- s3 ,3 A A
A A 6&B B;.B66B;r astroid.ClassDefc | t d | D S # t $ r Y dS w xY w)z,Return whether cls is a subclass of an Enum.c 3 K | ]9}|j t v o&t | d d dk V :dS )rE NrL )rE ENUM_BASE_NAMESgetattrr )r> r s r/ r_ z$_is_enum_subclass..b s`
J/ )
>
fd33v=
r: F)r r r )r s r/ _is_enum_subclassr _ sc
uus *-
;;) predicatec f | j dk o&t | dd dk S )Nr" rE typing)rE r
r r s r/ r z s1 l* 7 VT**h6 r: r )
r, rH rI rJ r& rK rL rM rN rO )rN rM r3 )r, rH r& rK rN r )F)r, rH r& rK rN r )r, r rN r )r& rK )r, rH rN rc )r r rN rM )?__doc__
__future__r functoolsr? syscollections.abcr textwrapr astroidr r r r r
astroid.builderr r r
astroid.contextr astroid.exceptionsr r r r r astroid.managerr version_infor r typing_extensionsr r __annotations__r! r# r0 rG r| r r partial_looks_like_namedtuple_looks_like_enum _looks_like_typing_namedtupler r r r r r r r r rm r register_transformCallrq FunctionDefr< r: r/ r% s
5 4 4 " " " " " "
$ $ $ $ $ $ @ @ @ @ @ @ @ @ @ @ @ @ @ @ N N N N N N N N N N , , , , , , + * * * * *v '''''' %
$ $ $ $"& & & & 4 (, V( V( V( V( V(r= = = +*;\JJJ $9$[v>>> 1 1+L Q Q Q :>; ; ; ; ; | " D :>32 32 32 32 32l $j j j jZ ) ) ) ) )B :>, , , , ,.' ' ' 'T # # J
/002H # # J
j))+; # # N$0A $ # # NMM"?@@BV # # M2337 7 # # J
5668U r: