7g{PUdZddlmZddlZddlZddlZddlmZm Z m Z m Z ddl m Z mZmZddlmZmZmZmZddlmZddlmZmZdd lmZmZmZmZmZdd l m!Z!dd l"m#Z#m$Z$m%Z%e red ejLejNZ(d Z)ejTejVejXejZdZ.de/d<dd&Z> d?d'Z?ejx d@d(Z@ejx dAd)ZA d*ZBej dB dCd+ZD dB dDd,ZE dB dEd-ZF dFd.ZG dB dGd/ZHej dB dHd0ZI dB dId1ZJd2ZKej dB dJd3ZLd4ZMej dB dKd5ZNej dL dMd6ZOejx dB dNd7ZPejx dL dOd8ZQejx dL dPd9ZRejx dL dQd:ZSejx dL dRd;ZTy)SzcThis module contains a set of functions to handle python protocols for nodes where it makes sense. ) annotationsN)Callable GeneratorIteratorSequence) TYPE_CHECKINGAnyTypeVar)bases decoratorsnodesutil)Context)InferenceContext copy_context)AstroidIndexErrorAstroidTypeErrorAttributeInferenceErrorInferenceError NoDefault) node_classes)ConstFactoryResultInferenceResultSuccessfulInferenceResult_TupleListNodeTzcontextlib.contextmanager)+-~notzdict[str, Callable[[Any], Any]]_UNARY_OPERATORScd|tur|}nt|}||}tj|S)z{Perform unary operation on `obj`, unless it is `NotImplemented`. Can raise TypeError if operation is unsupported. )NotImplementedr r const_factory)objopvaluefuncs 1lib/python3.12/site-packages/astroid/protocols.py_infer_unary_opr)/s5  n#S    u %%c@tt|j|SN)r)tupleeltsselfr%s r(tuple_infer_unary_opr1<s 5+R 00r*c.t|j|Sr,)r)r.r/s r(list_infer_unary_opr3@s 499b ))r*c@tt|j|Sr,)r)setr.r/s r(set_infer_unary_opr6Ds 3tyy>2 ..r*c.t|j|Sr,)r)r&r/s r(const_infer_unary_opr8Hs 4::r **r*c@tt|j|Sr,)r)dictitemsr/s r(dict_infer_unary_opr<Ls 4 +R 00r*c ||zSr,abs r(rBSa!er*c ||z Sr,r>r?s r(rBrBTrCr*c ||z Sr,r>r?s r(rBrBUrCr*c ||zSr,r>r?s r(rBrBVqAvr*c ||zSr,r>r?s r(rBrBWrCr*c ||zSr,r>r?s r(rBrBXsq!tr*c ||zSr,r>r?s r(rBrBYrCr*c ||zSr,r>r?s r(rBrBZrCr*c ||zSr,r>r?s r(rBrB[rCr*c ||z Sr,r>r?s r(rBrB\rCr*c ||zSr,r>r?s r(rBrB]rGr*c ||z Sr,r>r?s r(rBrB^rGr*) rr/z//***%&|^z<>@=c#Ktjt}t|tjr|dk(rct|jt t frCt|jt t fr#|jdkDs|jdkDr|y t|} tj||j|jyyyt|jtr|dk(rtjy|y#t$r|YGt$rtjYdwxYw#t$r|YywxYww)NrRgj@rS)r Constr" isinstancer&intfloat BIN_OP_IMPLr# TypeError Exceptionr Uninferablestr)r0opnodeoperatorothercontext_not_implementedimpls r(const_infer_binary_oprjeskk.1O%%  4::U|45;;e 5c!U[[3%6! !  "x(D '))$tzz5;;*GHH '& DJJ $S &%% '&&& ' "! ! "sTBE E)2D:E E"E$E>EEE EEEEc|j|}|dkDrtjg|_|Sfd|jD}t ||z|_|S)NparentgחAc3K|]F}t|tjs*tj|xstjHywr,)r[rUninferableBase safe_inferra).0eltrfs r( z'_multiply_seq_by_int..sBC#t334 W%9)9)99sA A) __class__rrar.list)r0rcr&rfnode filtered_eltss ` r(_multiply_seq_by_intrxs\ >>> (D s{%%&  99M ]#e+DI Kr*c#K|D]~}t|tjrtj4|j |D]7}t|tjs|"tj9ywr,)r[rror Unknowninfer)r.rfrrinferreds r(_filter_uninferable_nodesr}sb c4// 0--/ !IIg.!(D,@,@A"N--/) / BBc #Kddlm}d|_tjt }t ||jrh|dk(rc|j|}ttjt|j|t|j||_ |yt |tjr?|dk(r:t |jts|yt|||j|yt |t j"rk|dk(rf|j%|} | st&j(yt | jts t+dt||| j|y|yw)zInfer a binary operation on a tuple or list. The instance on which the binary operation is performed is a tuple or list. This refers to the left-hand side of the operation, so: 'tuple() + 1' or '[] + A()' r)helpersNrrlrQzPlease open a bug report.)astroidr boundnoder rZr"r[rtru itertoolschainr}r.r&r\rxr Instanceclass_instance_as_indexrraAssertionError) r0rcrdrerfmethodrrhrvas_indexs r(tl_infer_binary_oprs* Gkk.1O%(X_~~V~, OO)$))W=)%**g>    E5;; 'HO%++s+! ! "4gFF E5>> *x32259"" "HNNC0 !<= =&tVX^^WM MsFF c&|j||Sr,)infer_call_result)r0rcrdrerfrs r(instance_class_infer_binary_oprs  # #D' 22r*c#K|dd}|jd}|D]}t|tjrt |ds+ |j } t||tjtjfr|g}|D]r}tj|} |j||}|s|2t|tjr t|j!|||Ed{ty#t $rYwxYw#t$rYwxYw#tttf$rYwxYw7A#t"$rY.wxYww)z__init__)rrrrarvarargkwargrmtyperscoper[r ClassDefrr rrinstantiate_class callcontextcalleerCallSite extra_contextinfer_argumentr#r.r default_valuer{r) r0rrfrargargsfunctypecls is_metaclassr call_siterrs r(_arguments_infer_argnamer\s " >>>> W>CSXXdkk4::=V-VC>D W{{H DNN1%vt 4 <  &kk  &&(!#u~~6R388{;R    G,=,=u~~!N##,,C 8}4I  x '') ) 7&&$$++fj)__Ffj) 664 (DKK,<,< <!**7+>+>@U@UVI // T7K K K  t{{$$R(  ((J6++$$**,C0023FK  tzz##B'  w'%%d+11':::a X6 L( ; sa4L9)L L$D=L9"A*L9 L B:L9.L6L7L L9LL63L95L66L9cddlm} |j}|rH|jr<|jj }t |dr'|j}t |drn t|||S|rvt|dd|jjk(rN|j}t|}d|_|j||}|j|j||St|||S#t$rd}YwxYw)Nrrrrrf)rrrrrrrrrrframerrrrm) r0rvrfrr node_namerrrs r(arguments_assigned_stmtsrs "II 7&&$$++fj)__Ffj)(iAA -1B1BB)) w'"!!+w!?""4;; 7CC #D)W ==%  s C00 C>=C>c#K|s|jyt|jj|||Ed{||||dS7 w)Nr)r&_resolve_assignment_partsr{rs r(assign_assigned_stmtsrs] jj( !; "   s;A A  A c#dKt||||D]}|tj|ywr,)rrra)r0rvrfrr|s r(assign_annassigned_stmtsrs5 *$g{K  "" "N Ls.0c#K|dd}|jd}|D]}d}t|tjr |j|\}}n4t |dr(tj|} |j||}|sy|s|rt|tjry t|j|||Ed{y#t $rYywxYw#ttf$rYywxYw7-#t$rYywxYww)z3Recursive function to resolve multiple assignments.Nrr)rr[r Dictr;rrrZrrrrrorr{r)rrrfrrrrgrs r(rrsa.K OOA E dEJJ ' "jj/ !T9 %U+J << G< N $"6"6 7  4NN7+[';   %&78   "  s8D C "D 0C!'D *C9 C7 C9D  CD CD !C40D 3C44D 7C99 DD DD c#Kddlm}tj|jD]1}t |t jr|j|}|3||||dSw)Nr)objectsr) rrr unpack_inferrr[r rExceptionInstance)r0rvrfrrrs r(excepthandler_assigned_stmtsrs_  --dii8 h /00:H 9 "  sA!A#c#K t|j|}t |t j r|j}|js td||jjD]R}t|j|d}t |tjs:|jtk(sRn t| t|jyt |t jr^ t|jd|}t |t j"s t||j%||Ed{yt|#t$r}t||d}~wwxYw#t$r}t||d}~wwxYw#tt tf$r} t|| d} ~ wwxYw7vw)Nr)rvz,No decorators found on inferred generator %s __enter__)nextr{ StopIterationrr[r rrmr r FunctionDefqname_CONTEXTLIB_MGRinfer_yield_typesrigetattrr BoundMethodr) r0mgrrfr|er'decorator_node decoratorenterrs r(_infer_context_managerrs. ' 23(EOO, >T #oo33N^11'1BDII)U%6%67??$7 4!d+ + 3x1134 4 Henn - 9**;*HIE%!2!23 e, ,**4999#&&A .#&A-.( 3 d+ 2 3  7G 9 h/S 8 9 :sGE6BG.GGF1G F2);G$G%G6 F? F  FG F/ F**F//G2G GGGc #K tfd|jD}|t|||Ed{nKt|||D];}|}|D].}t |dst d||| |j |}0|=|||dS#t$rYywxYw7f#t$r}t d||||d}~wt$r}t d||||d}~wwxYww) aInfer names and other nodes from a *with* statement. This enables only inference for name binding in a *with* statement. For instance, in the following code, inferring `func` will return the `ContextManager` class, not whatever ``__enter__`` returns. We are doing this intentionally, because we consider that the context manager result is whatever __enter__ returns and what it is binded using the ``as`` keyword. class ContextManager(object): def __enter__(self): return 42 with ContextManager() as f: pass # ContextManager().infer() will return ContextManager # f.infer() will return 42. Arguments: self: nodes.With node: The target of the assignment, `as (a, b)` in `with foo as (a, b)`. context: Inference context used for caching already inferred objects assign_path: A list of indices, where each index specifies what item to fetch from the inference results. c34K|]\}}|k(s |ywr,r>)rqrvarsrvs r(rsz&with_assigned_stmts..dsE*;C 3*s Nr.z0Wrong type ({targets!r}) for {node!r} assignment)rvtargetsrrfzCTried to infer a nonexistent target with index {index} in {node!r}.z1Tried to unpack a non-iterable value in {node!r}.r) rr;rrrrr.rr_) r0rvrfrrresultr$rrs ` r(with_assigned_stmtsrBs0BE$**EE)$W===,T3@FC$sF+(J! $$/ ' ((5/C%8I?AB"  K  > "('! $$/ ' !(K! $$/ '   saCB CB8C/B>C BCBC C'B88 CCCCc#K|j|k(r%|jj|Ed{ytd|||7w)z:Infer names and other nodes from an assignment expression.rNz$Cannot infer NamedExpr node {node!r}r)targetr&r{rrs r(named_expr_assigned_stmtsrsM {{d::##G#444 2#    5s/AAAc# K dfd |j}t|tjtjfst d||||| t }t|tjr |j}|jd}t|tjstjytd|jtjDdkDrt d||||  t|j!|}t|tj$s t'|d stjy t)j*|j-}t1|j2D]\} } t| tjs|sn|j55t)j*t7|j2| d} | D]} t| tjs|s|j92tj:t<j>||j@|jB } | jEtG| | t|tjr t|jHj!| }t|tj$s t'|d stjy |j-}|jJ}t|tjLs t d| g}||||s t d| |d\}}||dz k(}tO||rdn||z }|}|D]}d}t1|D]C\} }t'|d sn2| dztQ|ur|}n|d} |j-}||}|}Etj:t<j>||j@|jB }|jE|xsg |ytjyy#t t"f$rtjYywxYw#t.$rtjYywxYw#t t"f$rtjYywxYw#t.$rtjYywxYw#tR$rY t.$rtjYywxYww)aD Arguments: self: nodes.Starred node: a node related to the current underlying Node. context: Inference context used for caching already inferred objects assign_path: A list of indices, where each index specifies what item to fetch from the inference results. c|j}t|D]\}}t|tjrK|j j |j j k(r|j|t|fyt|tjs|j|t|jf|||yr,) r enumerater[r Starredr&rappendlenr)starredrlookupsrrelement$_determine_starred_iteration_lookupss r(rzDstarred_assigned_stmts.._determine_starred_iteration_lookupss'/NE77EMM2MM&&'--*<*<<s6{34'5;;/s7>>+;'<=>4WgwO0r*zDStatement {stmt!r} enclosing {node!r} must be an Assign or For node.)rvrrrfNrc3 K|]}dyw)Nr>)rqrgs r(rsz)starred_assigned_stmts..s<;Qq;s rz=Too many starred arguments in the assignment targets {lhs!r}.)rvrrrfr)ctxrmlineno col_offset)r.rz8Could not make sense of this, the target must be a tuplez5Could not make sense of this, needs at least a lookup)r nodes.Starredrz nodes.Tuplerzlist[tuple[int, int]]returnNone)* statementr[r AssignForrrr&r BaseContainerrrasumnodes_of_classrrr{rror collectionsdequerr_rr.popleftreversedrrrStorerrpostinitrurrrslicerr)r0rvrfrrr&lhsrhsr.r left_nodelhs_elts right_nodepackedinferred_iterablerrrlast_element_indexlast_element_lengthis_starred_last lookup_slice last_lookupr found_elementlookup cur_lookupitered_inner_elementunpackedrs @r(starred_assigned_stmtsr*s$PP(3P>SP P >> D dU\\5995 6 R   "$$ % ll1o#u223"" "  <#,,U]];< g xG "" "  &--/F &%++. J  *,,T67C GQX  3:"+//,1Dq1HI #D*=@R*R # G!M!*7!3 vw119G ,.9J"(J ,+2>>+;(2:>G%,M%"4(zzMM{{?? H   =#6B  7N QTi#[ . "" "   "" "  N . "" "   "" "  h" ***sD TQ(9T"#Q6D#T)%R9TR>B1T SA0T!Q30T2Q33T6RTRT!R;8T:R;;T>STST T (T,T T T  Tc# KywzbReturn empty generator (return -> raises StopIteration) so inferred value is Uninferable. Nr>rs r(match_mapping_assigned_stmtsr-j  c# Kywr,r>rs r(match_star_assigned_stmtsr1xr.r/c#Kt|jtjr_t|jjtjr0|j #|jjj yyyyw)z}Infer MatchAs as the Match subject if it's the only MatchCase pattern else raise StopIteration to yield Uninferable. N)r[rmr MatchCaseMatchpatternsubjectrs r(match_as_assigned_stmtsr7sa 4;;0 t{{))5;; 7 LL kk  ((( ! 8 1r~c#:Ktjdyw)zzHack. Return any Node so inference doesn't fail when evaluating __class_getitem__. Revert if it's causing issues. N)r rZrs r(generic_type_assigned_stmtsr9s ++d s)r$r r%rbr r)r0z nodes.Constrcnodes.AugAssign | nodes.BinOprdrbrerrfrrgrr z4Generator[ConstFactoryResult | util.UninferableBase]) r0rrcr:r&r\rfrr r)r.zSequence[InferenceResult]rfrr z#Iterator[SuccessfulInferenceResult])r0rrcr:rdrbrerrfrrrr z?Generator[_TupleListNodeT | nodes.Const | util.UninferableBase])r0znodes.ClassDefrcr:rdrbrerrfrrrr Generator[InferenceResult])NNN) r0znodes.For | nodes.Comprehensionrv&node_classes.AssignedStmtsPossibleNoderfInferenceContext | Nonerlist[int] | Noner r ) r0znodes.Tuple | nodes.Listrvr<rfr=rr>r r ) r0z#nodes.AssignName | nodes.AssignAttrrvr<rfr=rr>r r )rz str | Nonerfrr r;) r0znodes.Argumentsrvr<rfr=rr>r r ) r0zBnodes.AugAssign | nodes.Assign | nodes.AnnAssign | nodes.TypeAliasrvr<rfr=rr>r r ) r0znodes.AnnAssignrvr<rfr=rr>r r ) r0znodes.ExceptHandlerrvr<rfr=rr>r r ) r0z nodes.Withrvr<rfr=rr>r r )NN) r0znodes.NamedExprrvr<rfr=rr>r r ) r0r rvr<rfr=rr>r r ) r0znodes.MatchMappingrvnodes.AssignNamerfr=rr r Generator[nodes.NodeNG]) r0znodes.MatchStarrvr?rfr=rr r r@) r0z nodes.MatchAsrvr?rfr=rr r r@) r0z4nodes.TypeVar | nodes.TypeVarTuple | nodes.ParamSpecrvr?rfr=rr r r@)U__doc__ __future__rrrrd operator_modcollections.abcrrrrtypingrr r rr r r r astroid.constrastroid.contextrrastroid.exceptionsrrrrr astroid.nodesrastroid.typingrrrrrrrposneginvertnot_r __annotations__r)r1r3r6r8r<matmulr^rur;_KEY_IMPLyes_if_nothing_inferredrjrxr}rrrraise_if_nothing_inferredrrrrrrrrrrrrr*r-r1r7r9r>r*r(rUs #CC..22!:' /ejjIO-             51 &1*/+1                 ))+,KD%#Ks - ##! ! )!!  !  ! ! !:!$!H  )    & * # *.> *( * ##, , ),,  ,  , & ,E,$,^ ##3 3 )33  3  3 & 3 3$3 &R %%48'+$(  ) 0%"   &<48'+$(  " 0%"   448'+$( B -B 0B%B" B  B==%5==D48'+$( > > 0>%>" >  >> %%48'+$(  L 0%"   &.48'+$(    0 % "   %P %%48'+$(   0%"   &*#'L %%48'+$( L L 0L%L" L  L&L^ %%(,$(   0 % "    & $ ##48'+$( A A 0A%A" A  A$AH ##(,      %      $   ##(,      %      $   ##(, ) ) )%) )  )$)" ##(,  >   %    $ r*