~ B d Z ddlmZ ddlZddlmZ ddlmZmZ ddl m
Z
m Z m Z m
Z
ddlmZ ddlmZmZ dd lmZmZmZ dd
lmZ dd lmZ dd lm Z dd
l m Z m Z ej dk rddlm!Z! nddl"m!Z! edee!d e
j# f ee!d e
j$ f f Z% e&d Z'dZ( e&d Z)dZ*e'fdJd Z+dKd Z, dLdMd"Z-dNd#Z.dOd%Z/dPd*Z0dQd.Z1dRd2Z2 dSdTd7Z3 dSdUd9Z4e'fdVd;Z5dWd<Z6 dXdYd?Z7dZdBZ8d[dCZ9d[dDZ:d[dEZ; e&dF Z< dSd\dHZ= e > e
j? e,e+ e > e
j$ ee4d=I e7 e > e
j@ ee3d=I e6 dS )]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/
) annotationsN)Iterator)TupleUnion)basescontexthelpersnodes)parse) PY39_PLUS
PY310_PLUS)AstroidSyntaxErrorInferenceErrorUseInferenceDefault)
inference_tip)AstroidManager)InferenceResult) UninferableUninferableBase) )Literaldefaultdefault_factory) dataclassfield) dataclassesmarshmallow_dataclasszpydantic.dataclasses_HAS_DEFAULT_FACTORYnodenodes.ClassDefdecorator_namesfrozenset[str]returnboolc t | t j r| j sdS t fd| j j D S )zDReturn True if a decorated node has a `dataclass` decorator applied.Fc 3 8 K | ]}t | V d S N) _looks_like_dataclass_decorator).0decorator_attributer" s q/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/astroid/brain/brain_dataclasses.py z.is_decorated_with_dataclass..8 sB ((;_MM )
isinstancer
ClassDef
decoratorsany)r r" s `r, is_decorated_with_dataclassr3 1 sd dEN + + 4? u #'?#8 r. Nonec d| _ t | D ][}|j j }t j |j |j | }t | }|g| j
|< \t | sdS d}t rg| j
j r[| j
j D ]N}t |t j sd} n/|j D ]&}|j dk r|j }'Ot) | t+ t | d | } t- | d }| |_ d\ |_ |_ |g| j d< | } t4 | j vrFt- t4 d j d
}
| |
_ |
j d
g| j t4 < dS dS # t: $ r Y dS w xY w) z6Rewrite a dataclass to be easily understood by pylint.Tlineno
col_offsetparentNFkw_only)init__init__NNz = object()r ) is_dataclass_get_dataclass_attributestargetnamer
Unknownr7 r8 r visit_transformsinstance_attrs _check_generate_dataclass_initr
r1 r/ Callkeywordsargvalue
bool_value_generate_dataclass_initlistr r9 localsrootDEFAULT_FACTORYbodytargetsr ) r assign_noderA rhs_nodekw_only_decorated decoratorkeywordinit_str init_noderN
new_assigns r, dataclass_transformrZ > s D066 / / !&= %"-
"##44X>>%-JD!! )$ / / Cdo+ C . C CI i44
$)!$-
C
C;)++(/
(@(@(B(B%
C (
&t$
7
7
788 H
C(OOJ/ 1;. ).#,+ J yy{{ $+ - - / > > >??DQGJ $J ,6,>q,A+BDK ( ( ( . -
s "G
GGFr; Iterator[nodes.AnnAssign]c # K | j D ]}t |t j r t |j t j s. sX
v +
((** * r. )rM r1 r
r/ rF r) r2 rG )r foundr+ s r, rE rE s T[ u E # 4 ( (-uz:: *+> ? ? ('E }t ~ r. Ytuple[dict[str, tuple[str | None, str | None]], dict[str, tuple[str | None, str | None]]]c h i }i }t | D ]}|j s
|j d d }n# t $ r Y +w xY w|j \ }}| D ]
\ }}|||< | D ]
\ } }||| < ||fS )z rM KeyErrorargs_get_arguments_dataitems)
r pos_only_store
kw_only_storebase base_initpos_onlyr: posargdatakwargs
r, !_find_arguments_from_base_classesry s @BN>@M $$ ( ( +/;z+B1+EII H &N > > @ @' $NN , , * *LFD &*N6 " ""==?? ( (KE4#'M% ( = ((s A
AArq (dict[str, tuple[str | None, str | None]]rr tuple[str, str]c T d\ }}| D ];\ }}||z
}|d r|d|d z z
}|d r|d|d z z
}|dz
}<| D ];\ }}||z
}|d r|d|d z z
}|d r|d|d z z
}|dz
}<||fS )zKParse positional and keyword arguments into strings for an __init__ method.) r} r : = , )rp )rq rr ru r: pos_argrw kw_args r, _parse_arguments_into_stringsr s
Hg'--//
G 7 ' tAw &H 7 ( Q 'HD%++-- 6 7 & td1g~ %G 7 ' utAw &G4 W r. rA strnodes.NodeNG | Nonec t | D ]}|j s
||j v r|j | D ]}t |j t j rp|j j rdt |j j t j r@t |j j r't |j j }|r |d c c S dS )z4Get the default value of a previously defined field.r N) rk rl r> rM r/ r9 r
r] rI rF _looks_like_dataclass_field_call_get_field_default)r rA rs assignr s r, _get_previous_field_defaultr s $$
*
* 4; +d+
*
* v}eo>>*
+* #6=#6
CC* 9 9LMM * 1 1D E EG * &qz))))) 4r. assignslist[nodes.AnnAssign]rT c g }g }g }t | \ }}|D ]5}|j j |j |j } }
} d} | j | D ]?}
t
|
t j s |
j s%d|
v r|
} n@t
| t j ot | d }|rKt d | j D r-| | d | | d t! |
r)d}t
|
t j r|
j }
nd}
d}n
d}d| d | }d
\ }}|
|
}| r~|rgt) | }|rU|\ }}|d k r| }n|d k r/t* }d| d | d
| dt* d|
}n| }n| rY t- t/ | }n=# t2 t4 f$ r Y n*w xY wt7 | | }|r| }| }||d| z
}||d | z
}|r_d | j D }|rL|d j r| | n| | |r"| |v r||f|| < nW| | nA| |v r||f|| < n5| |v r | g|z }| | n| | |s| | 7t= || \ }}d|v rdnd}||d | z z
}| d s|dz
}|s|r|dz
}|| d | z
}|rd | nd} d| d| S )z8Return an init method for a dataclass given the targets.Nzbuiltins.propertyF) check_scopec 3 ` K | ])}|j d k o|j V *dS re rf rg s r, r- z+_generate_dataclass_init.. sS v%Hgm.F.F.H.H*H r. Tr} zself.r r= r r z if z is z else r~ c ( g | ]}|j d k
|S )r: )rH )r* ks r,
z,_generate_dataclass_init..X s$ GGGQAEY4F4Fq4F4F4Fr. r selfzself, r z*, z
passz
def __init__(z) -> None:
)!ry r@ rA r` rI rM r/ r
FunctionDefr1 decoratornamesrF r r2 rG poprb Subscriptslice as_stringr rO r nextinfer_call_resultr
StopIterationr rJ appendr joinendswith) r r rT paramskw_only_params assignmentsprev_pos_only_storeprev_kw_only_storer rA r` rI
property_nodeadditional_assignis_fieldinit_varassignment_strann_str default_strresult default_type default_nodeprevious_default param_strr:
prev_pos_only prev_kw_only
params_stringassignments_strings r, rK rK sj F "N K.OPT.U.U++ p/ p/"(-"4f6G %j 37
!%T!2 /1BCC
$/
"&7&F&F&H&HHH 1
I eUZ00
5U u6
6
6
$~
$''d333"&&tT222
# #
5H*eo66
" ' -
"
NN H 4T 4 4d 4 4N ) ! **,,G ;
0+E22 17.L,#y00&2&<&<&>&> %):::&5 JD J J\-C-C-E-E J J"& J J,; J JCG J J '
$oo / /
;
!$}'F'F'H'H"I"I"S"S"U"U V V "M2
;4 F F
; . 8 8 : : 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 NNs A H H43H4
nodes.Unknownctx context.InferenceContext | NoneIterator[InferenceResult]c # K | j }t |t j s t V dS |j |j }}| | | E d{V |t || E d{V dS t V dS )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 ) r9 r/ r
r] r r` rI infer _infer_instance_from_annotation)r r r r` rI s r, infer_dataclass_attributer s [F feo . . )6<J ;;s;+++++++++ 2:3GGGGGGGGGGGGr.
nodes.Callc # K t | j t j t j f st
t
| }|s t V dS |\ }}|dk r | | E d{V dS t |
j d j }| j |_ | | E d{V dS )z(Inference tip for dataclass field calls.r r Nr )
r/ r9 r
r] Assignr r r r r r rP rI )r r r r r new_calls r, infer_dataclass_field_callr s dkEOU\#B C C "!!
%
%F 3 & g 9 $ $ }}S}11 1 1 1 1 1 1 1 1 1 W . . 0 0116q9?H "kHO ~~c~22 2 2 2 2 2 2 2 2 2r. nodes.NodeNGc t | t j r| j } t | }n # t t f$ r
t }Y nw xY wt |t rHt | t j
r | j |v S t | t j r | j
|v S dS t |t j o(|j |v o | j t v S )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
rF funcr r r r r r NamerA Attributeattrnamer rN DATACLASS_MODULES)r r" inferreds r, r) r) s $
## y
%% M * (O,, dEJ ' ' 09/ / dEO , , 4=O3 3u 8U .// 6 M_ , 6 MMOO $5 5s !A A A c | j }|sdS | }t |t j o(t |t j ot
| S )z^Return True if node was dynamically generated as the child of an AnnAssign
statement.
F)r9 scoper/ r
r] r0 r3 )r r9 r s r, _looks_like_dataclass_attributer s] [F u LLNNE65?++ / uen - - / ' . .r. Tr c
|rv| d }| }t |t j r0|j )t |t j rt | sdS t | j
}n# t t f$ r Y dS w xY wt |t j
sdS |j t k o | j t" v S )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.
T)futureNF) statementr r/ r
r] rI r0 r3 r r r r r r rA
FIELD_NAMErN r )r r stmtr r s r, r r s ~~T~**
tU_ - -
&5%.11 '+E22 ' 5 ))** M * uu h 1 2 2 u =J & T8==??+?CT+TTs :&B! !B65B6
field_call_FieldDefaultReturnc d\ }}| j D ]'}|j dk r|j }|j dk r|j }(||d|fS |B|@t j | j | j | j }| | d|fS dS )ah Return 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= r r Nr6 )r ) rG rH rI r
rF r7 r8 r9 postinit)r r r rV r s r, r r s * G_ & , , ;) # # mGG
[ -
-
- %mO 6'!!?6: $!, $
/// (** 4r. c t rP t | }n# t t f$ r Y dS w xY wt |dd dk S t
| t j o]t
| j t j
r| j j dk p.t
| j t j o| j j
dk S )z@Return True if node is a ClassVar, with or without subscripting.FrA r} ClassVar)r r r r r getattrr/ r
r rI r rA r r r r s r, r_ r_ s ; DJJLL))HH
. 55 x,,
:: dEO , , 4:uz** * JOz ) . dj%/ 2 2 . J : - s !+ A A c t sdS t j | }t |t j o| dk S )z,Return True if node is the KW_ONLY sentinel.Fzdataclasses._KW_ONLY_TYPE)r
r
safe_inferr/ r Instanceqnamer s r, ra ra 2 sK u!$''H8U^,, < NN ; ;r. c t | }n# t t f$ r Y dS w xY wt |dd dk S )z@Return True if node is an InitVar, with or without subscripting.FrA r} InitVar)r r r r r r s r, rb rb = s^
%% M * uu 8VR ( (I 55s !$ 99)Dict FrozenSetListSetr *Iterator[UninferableBase | bases.Instance]c # K d} t | | }n # t t f$ r t V Y nw xY wt |t j s t V dS | j dv r1|j t v r| V dS t V dS | V dS )zInfer an instance corresponding to the type annotation represented by node.
Currently has limited support for the typing module.
Nr > r} typing_collections_abc) r r r r r r/ r
r0 rN rA _INFERABLE_TYPING_TYPESinstantiate_class)r r klasss r, r r S s
ETZZZ,,-- M * eU^ , , (
:0 0 0))++ + + + + + %%'''''''s #* AA)raise_on_overwrite)r r! r" r# r$ r% )r r! r$ r4 )F)r r! r; r% r$ r[ )r r! r$ r% )r r! r$ ri )rq rz rr rz r$ r{ )r r! rA r r$ r )r r! r r rT r% r$ r r( )r r r r r$ r )r r r r r$ r )r r r" r# r$ r% )r r r$ r% )T)r r r r% r$ r% )r r r$ r )r r r$ r% )r r r r r$ r )A__doc__
__future__r syscollections.abcr r r r astroidr r r r
astroid.builderr
astroid.constr r
astroid.exceptionsr r r astroid.inference_tipr astroid.managerr astroid.typingr astroid.utilr r version_infor typing_extensionsNodeNGrF r frozensetDATACLASSES_DECORATORSr r rO r3 rZ r? rE ry r r rK r r r) r r r r_ ra rb r r register_transformr0 rB r. r, r sH
# " " " " "
$ $ $ $ $ $ 2 2 2 2 2 2 2 2 2 2 2 2 ! ! ! ! ! ! / / / / / / / / V V V V V V V V V V / / / / / / * * * * * * * * * * * * 5 5 5 5 5 5 5 5v )))))) ')
el
*+ '#
$ej
013 #>22
ID ) =S
/C /C /C /Cf (- 6 < ) ) ) )F 0 &NO NO NO NOd AE 0 >B3 3 3 3 3( ;Q <
" +/U U U U U< @ ( 6 6 6 6 $) @D( ( ( ( (6 # # N')D # # JM,FFF$ # # MM+EEE# r.