X-Ph\ddlmZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZmZmZmZmZmZmZmZmZmZmZmZdd lmZdd lmZdd l m!Z!dd l"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+ddl,m-Z-d;dZ.d;dZ/dd$Z2d?d)Z3Gd*d+eZ4d@d1Z5Gd2d3eZ6dAd5Z7Gd6d7Z8dBd:Z9dS)C) annotationsN) defaultdict) cmp_to_key)Callable)State) format_type)PYTHON_EXTENSIONS) LDEF Decorator ExpressionFuncBase MemberExprMypyFileNodeOverloadedFuncDefRefExpr SymbolNodeTypeInfoVar)FineGrainedBuildManager)ExtendedTraverserVisitor)tuple_fallback) FunctionLikeInstance LiteralType ProperType TupleType TypedDictType TypeVarType UnionTypeget_proper_type)fill_typevars_with_anyorlineintcolumnreturnboolcD|j|kp|j|ko |j|kSN)r$r&r#r$r&s P/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/mypy/inspections.pynode_starts_afterr-*s% 6D= @AFdN@qx&/@@ch|j*|j#|j|ks|j|kr |j|krdSdS)NTF)end_line end_columnr+s r,node_ends_beforer2.sB z!,": :   d 2 2q|f7L7L4 5r.exprr strcJ|jd|jdzd|jd|jS)z1Format expression span as in mypy error messages.:)r$r&r0r1)r3s r, expr_spanr87s2i M M$+/ M MDM M MDO M MMr.typrlist[Instance]ct|tr|gSt|trt|gSt|tr|jgSt|t r|jgSt|tr|jgSt|tre|j r=g}|j D]1}| tt|2|Stt|j St|tr=g}|jD]1}| tt|2|SgS)zBReturns the Instance fallback for this type if one exists or None.) isinstancerrrrfallbackrrrvaluesextendget_instance_fallbackr! upper_boundr items)r9rests r,r@r@<sc#x  u C # #s##$$ C ' ' ~ C & & ~ C % %  ~ C % %  : CZ F F 01C1CDDEEEEJ$_S_%E%EFFF C # # B BA JJ,_Q-?-?@@ A A A A Ir.nameinforVar | FuncBase | Nonecp||}|rgt|tr|jS|jrBt|t sJ|jd}t|tsJ|jS|S||}|r|jnd}t|tr|SdS)z7Find the node defining member 'name' in given TypeInfo.rN) get_methodr<r var is_propertyrrBgetnoder)rErFmethoddecrMvs r, find_noderQWs__T " "F  fi ( ( :    f&788 8 8 8,q/Cc9-- - - -7N xx~~ 'DII4 a   H 4r.fullnamemodulesdict[str, State] State | Nonec|}||vr||S d|vrdS|dd\}}||}||S:)znFind module by a node fullname. This logic mimics the one we use in fixup, so should be good enough. T.Nr7maxsplit)rsplitrL)rRrSheadtailmods r,find_module_by_fullnamer^msj D wt} d??4[[q[11 dkk$ ?J r.c"eZdZdZdd Zdd ZdS) SearchVisitorzGVisitor looking for an expression whose span matches given one exactly.r$r%r&r0r1r'NonecL||_||_||_||_d|_dSr*)r$r&r0r1result)selfr$r&r0r1s r,__init__zSearchVisitor.__init__s)    $)- r.r#rr(c@t||j|jrdSt||j|jrdS|j|jkrL|j|jkr<|j|jkr,|j|jkrt |tr||_|jduS)NF) r-r$r&r2r0r1r<r rcrdr#s r,visitzSearchVisitor.visits Q 4; 7 7 5 At}do > > 5 Fdi   dm++DK'' //!Z((  {d""r.N) r$r%r&r%r0r%r1r%r'rar#rr'r(__name__ __module__ __qualname____doc__rerhr.r,r`r`sBQQ.... # # # # # #r.r`treerr0r1Expression | Nonec||krtd||kr||krtdt||||}|||jS)z=Find an expression matching given span, or None if not found.z$"end_line" must not be before "line"z#"end_column" must be after "column") ValueErrorr`acceptrc)rpr$r&r0r1visitors r,find_by_locationrvsm$?@@@4J&00>???D&(J??GKK >r.c"eZdZdZd dZdd Zd S)SearchAllVisitorzGVisitor looking for all expressions whose spans enclose given position.r$r%r&r'rac0||_||_g|_dSr*)r$r&rc)rdr$r&s r,rezSearchAllVisitor.__init__s  (* r.r#rr(ct||j|jrdSt||j|jrdSt |t r|j|dS)NFT)r-r$r&r2r<r rcappendrgs r,rhzSearchAllVisitor.visitsi Q 4; 7 7 5 Aty$+ 6 6 5 a $ $ " K  q ! ! !tr.N)r$r%r&r%r'rarirjror.r,rxrxsBQQ++++ r.rxlist[Expression]ct||}||tt|jS)zFFind all expressions enclosing given position starting from innermost.)rxrtlistreversedrc)rpr$r&rus r,find_all_by_locationrs;tV,,GKK (( ) ))r.ceZdZdZdddddddddIdZdJdZdKdZdLdZdMdZdNd#Z dKd$Z dOd)Z dPd,Z dQd/Z dKd0ZdRd1ZdRd2ZdSd4ZdTd>ZdUd?ZdVdBZdWdDZdXdEZdXdFZdXdGZdHS)YInspectionEnginez:Engine for locating and statically inspecting expressions.rF) verbositylimit include_span include_kindinclude_object_attrs union_attrs force_reload fg_managerrrr%rrr(rrrrr'rac||_||_||_||_||_||_||_||_d|_dSr*) rrrrrrrrmodule) rdrrrrrrrrs r,rezInspectionEngine.__init__sL%" (($8!&($( r.statercV|jjjj}d|jjj_ |j|jJ|j|j|jfgg||jjj_dS#||jjj_wxYw)z4Reload given module while temporary exporting types.TN)rmanageroptions export_types flush_cachepathupdateid)rdrolds r, reload_modulezInspectionEngine.reload_moduleso%-:7;'4 ? O ' ' ) ) ):))) O " "UXuz$:#;R @ @ @;>DO # + 8 8 83DO # + 8 > > > >s A BB( expressionr tuple[str, bool]c|jjj|}|||dfSt ||jjj|j}|||dfS)zFormat type for an expression using current options. If type is known, second item returned is True. If type is not known, an error message is returned instead, and second item returned is False. NF)rT) rr all_typesrL missing_typerrr add_prefixes)rdr expr_typetype_strs r,rzInspectionEngine.expr_types O+599*EE  $$Z00%7 7 t.6$.     :66<t| t|} i}| D]}| |s | |||<|S)z3Collect attributes from all union/typevar variants. attr_dictrr'set[str]c\t}|D]}|t||z}|Sr*)set)rattrsbases r, item_attrsz2InspectionEngine.collect_attrs..item_attrss7EEE! . .Yt_---Lr.xryr%c2||jvrdS||jvrdSdS)Nr7r)mro)rrs r, cmp_typesz1InspectionEngine.collect_attrs..cmp_typess'AEzzqAEzzr1r.Nrrr7)key)rrr'r)rrrrr'r%) typerrsortedrr{rr~rkeysr)rdrrr all_attrsinstancerrr intersectionitemcombined_attrsrE sorted_basesrcs r, collect_attrszInspectionEngine.collect_attrss          ! $ $HE-#C, #2#h 1 1$TZ00d   U # # # #"z)A,// abbM - -D JJt,, ,LL%T** : :D : :>)) J::D':4<+?+?&t,33D999: :n1133I9N9NOOO   0 0D!$' )$/F4LL r. attrs_strs list[str] attrs_dictc |D]\}|jdkr|jn|j}d||D}|d|dd|d]dS)Nr7cg|]}d|d S"ro).0attrs r, z4InspectionEngine._fill_from_dict..6s >>>T[[[[>>>r.rz": [, ])rrErRr{join)rdrrrcls_namers r,_fill_from_dictz InspectionEngine._fill_from_dict1s E ED$(NQ$6$6tyyDMH>>Z-=>>>E   C(CC %0@0@CCC D D D D E Er.c |jjj|}|||dfSt |}t |}|s|g}||}t|trft|j trL|j }t|j}d|vr|dd|jdd|Di}ni}t|t"rX|rDt'|}|t |} ni} g} |r9|D]6} | | dd|| d 7|| | || ||d d| d |d fS) avFormat attributes that are valid for a given expression. If expression type is not an Instance, try using fallback. Attributes are returned as a JSON (ordered by MRO) that maps base class name to list of attributes. Attributes may appear in multiple bases if overridden (we simply follow usual mypy logic for creating new Vars etc). NF __builtins__z""cg|]}d|d Srro)rrEs r,rz/InspectionEngine.expr_attrs..Ts 0O0O0OT0O0O0Or.z: [rr{}T)rrrrLrr!r@rrr<rrMrrrremoverRr is_type_objr" type_objectr{rrr) rdrrrrrMrmod_dicttemplate class_dict base_attrsr]s r, expr_attrszInspectionEngine.expr_attrs9s)O+599*EE  $$Z00%7 7#I.. ))44  -))++,I'' 22  j' * * z*/8/T/T ?D4:&&E&& ^,,,.T]...0O0O0O0O0OPHHH i . . 93H3H3J3J -i.C.C.E.EFFH++,A(,K,KLLJJJ  J J J!!S"H"HTYYx}-E-E"H"H"HIIII Z444 Z444  !?dii &;&;!?!?!?LLdRRr.rrMFuncBase | SymbolNoder4cJ|jd|jd|jdzd|jS)Nr6r7)rr$r&rE)rdrrMs r, format_nodezInspectionEngine.format_nodeis2+II IIDK!OIIdiIIIr.rlist[FuncBase | SymbolNode]c|j}|6t|tr|j|jjj|j}|gSt|}t|}g}|D]3}t|j |j }|r||4|st|try|rett#|}|D]3}t|j |j }|r||4ngSngS|g}|S)zCollect nodes that can be referred to by an expression. Note: it can be more than one for example in case of a union attribute. )rMr<rkindrrrrLr3r!r@rQrErr{rrr"r)rdrrM base_typernodesrs r, collect_nodeszInspectionEngine.collect_nodeslsg .8_ <*j11 jo6M O3=AA*/RR $I,I66 1)<<  )++H$Z_hmDDD+ T*** "!)\:: "y?T?T?V?V "$9293H3H3J3JKK%% )233H#,Z_hm#L#LD#3 % T 2 2 23 "  " FE r.r/tuple[dict[FuncBase | SymbolNode, State], bool]c4i}d}|D]}t|j|jj}|s |jt kr|jr|j}nC|||<|jr|jjs|j r,||j p |jjz}| |||fS)zGather modules where given nodes where defined. Also check if they need to be refreshed (cached nodes may have lines/columns missing). F) r^rRrrrr rrpis_cache_skeletonrr)rdrrrS reload_neededrMrs r,modules_for_nodesz"InspectionEngine.modules_for_nodess  + +D,T]DO r)rr{rrkrr8r)rdrcrprefixesprefixs r,rzInspectionEngine.add_prefixess   < OOtJ//8: ; ; ;   3 OOIj11 2 2 2  XXh''&0FFFr.rprr$r&r0r1rN(Callable[[Expression], tuple[str, bool]]dict[str, object]c t|||dz ||}n(#t$r}dt|icYd}~Sd}~wwxYw||d|d|d|} d| dddS||\} } | d| rdnddS) zzGet type of an expression matching a span. Type or error is returned as a standard daemon response dict. r7errorNr6zCan't find expression at span routerrstatusr)rvrsr4) rdrpr$r&r0r1rNrrspaninspection_strsuccesss r, run_inspection_by_exact_locationz1InspectionEngine.run_inspection_by_exact_locations ')$fqj(JWWJJ ' ' 'SXX& & & & & & & '  ==V==h====DBDBB2YZ[[ ["(&"4"4%b9>>ct|||dz }|s|d|}d|dddSg}d}|D]+} || \} } | sd}| r|| ,|jr|d|j}d|d|dS) zGet types of all expressions enclosing a position. Types and/or errors are returned as a standard daemon response dict. r7r6z'Can't find any expressions at position rrrN )rr{rr) rdrpr$r&rN expressionspositioninspection_strsrrrrs r,run_inspection_by_positionz+InspectionEngine.run_inspection_by_positions+4vzBB  ))))HKKK  % 7 7J&,fZ&8&8 #NG  7&&~666 : <-l l;Oyy11"OOOr.file&tuple[State | None, dict[str, object]]c,tfdtDsdddifStjt fd|jjDd}||_ || dddd nifS) zFind module by path, or return a suitable error message. Note we don't use exceptions to simplify handling 1 vs 2 statuses. c3BK|]}|VdSr*)endswith)rextr s r, z/InspectionEngine.find_module..%s/CC#4==%%CCCCCCr.Nrz Source file is not a Python filec32K|]}|jk |VdSr*)abspath)rsabs_paths r,rz/InspectionEngine.find_module..+s0YYA19PXCXCXaCXCXCXCXYYr.zUnknown module: rr7r) anyr osrrnextrrr>r)rdr rrs ` @r, find_modulezInspectionEngine.find_module s CCCC1BCCCCC G'#EFF F7??4((YYYY!6!=!=!?!?YYY[_`` JO--t--bA F F F]_  r.locationc  t|\}}n(#t$r}dt|icYd}~Sd}~wwxYw||\}}||sJ|S|jr|jjs|jr|||jJt|dkr&|\}} } } | |j|| | | |St|dksJ|\}} | |j|| |S)z{Top-level logic to inspect expression(s) at a location. This can be reused by various simple inspections. rN) parse_locationrsr4rrprrrlenrr) rdrrNr posrrerr_dictr$r&r0r1s r,run_inspectionzInspectionEngine.run_inspection2sG '&x00ID## ' ' 'SXX& & & & & & & '**400x =OOOOz &UZ9 &T=N &   u % % %z%%% s88q==14 .D&(J88 D&(J 3xx1}}}} f..uz4PPPs :5::c8|||jS)z)Get types of expression(s) at a location.)r rrdrs r,get_typezInspectionEngine.get_typeSs""8T^<<>>>r.rrtuple[str, list[int]]c~|ddkrtd|dd}|^}}|ddkr|d|DfS|dd}|^}}|ddkr|d||zDfStd)Nr6rz7Format should be file:line:column[:end_line:end_column]rXc,g|]}t|Sror%rps r,rz"parse_location..ms,,,!s1vv,,,r.c,g|]}t|Sror-r.s r,rz"parse_location..qs999!s1vv999r.)countrsrZ)rpartsstartrest start_rests r,rrfs~~cQRSSS OOC!O , ,ELED {{3!,,t,,,,, LLqL ) )EEJ {{3!99zD'899999 N O OOr.)r#rr$r%r&r%r'r()r3r r'r4)r9rr'r:)rEr4rFrr'rG)rRr4rSrTr'rU) rprr$r%r&r%r0r%r1r%r'rq)rprr$r%r&r%r'r|)rr4r'r*): __future__rr collectionsr functoolsrtypingr mypy.buildr mypy.messagesrmypy.modulefinderr mypy.nodesr r r r rrrrrrrrmypy.server.updatermypy.traverserr mypy.typeopsr mypy.typesrrrrrrrr r! mypy.typevarsr"r-r2r8r@rQr^r`rvrxrrrror.r,rCs%"""""" ###### %%%%%%//////                            766666333333''''''                      100000AAAANNNN 6,&#####,###4    /$****ccccccccL P P P P P Pr.