%^gPVUddlmZddlmZddlmZddlmZmZm Z m Z m Z m Z m Z ddlmZddlmZddlmZddlmZmZmZmZdd lmZmZmZmZmZm Z m!Z!dd l"m#Z#m$Z$dd l%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2dd l3m4Z4e eeefZ5d e6d<Gdde Z7GddZ8eee e e,e e,ffZ9GddZ:ddZ;ddZr?rAs r8__repr__zFrame.__repr__9s:y4::,b1A1A0B"TE[E[D\\]^^r7NF)r<intr?r0returnNone)rIstr)r2r3r4__doc__rBrFr6r7r8r:r:&s 3_r7r:cReZdZUdZdZded<d$dZd%dZd&d'dZd(d)d Z d*d+d Z d*d,d Z d d d-dZ d$dZ d$dZd.dZd/dZd/dZd0dZd1dZd2dZd3dZed4dZ d( d5dZd6dZd7dZd8dZd$dZd$dZedd d ddd! d9d"Zed:d#Zy);ConditionalTypeBinderaKeep track of conditional types of variables. NB: Variables are tracked by literal expression, so it is possible to confuse the binder; for example, ``` class A: a: Union[int, str] = None x = A() lst = [x] reveal_type(x.a) # Union[int, str] x.a = 1 reveal_type(x.a) # int reveal_type(lst[0].a) # Union[int, str] lst[0].a = 'a' reveal_type(x.a) # int reveal_type(lst[0].a) # str ``` NzAssigns | Nonetype_assignmentscd|_t|jg|_g|_i|_i|_d|_t|_ g|_ g|_ y)NF) next_idr:_get_idframesoptions_on_return declarations dependencieslast_pop_changedset try_frames break_framescontinue_framesrEs r8rBzConditionalTypeBinder.__init__Ysa T\\^,- 575724!&$'E')*,r7cD|xjdz c_|jS)NrQ)rRrEs r8rSzConditionalTypeBinder._get_idys  ||r7c||}n3|jj|tj|t |D]}|j ||yN)rW setdefaultrYaddr_add_dependencies)rAkeyvalueelts r8rbz'ConditionalTypeBinder._add_dependencies}sM =E    ( (ce 4 8 8 ?3rEs r8r>z!ConditionalTypeBinder.unreachables&* B#r7c,d|jd_yr|)rTr@rEs r8r@z3ConditionalTypeBinder.suppress_unreachable_warningss8< B5r7cjt|}|Jd|j|}|y|jS)Nz/Internal error: binder tried to get non-literal)rrsr/)rArxrcfounds r8getzConditionalTypeBinder.gets;4 Q QQ # =zzr7c:td|jDS)Nc34K|]}|jywr_r>.0rhs r8 z7ConditionalTypeBinder.is_unreachable..s6+Q1==+anyrTrEs r8is_unreachablez$ConditionalTypeBinder.is_unreachables6$++666r7c:td|jDS)Nc34K|]}|jywr_)r@rs r8rzJConditionalTypeBinder.is_unreachable_warning_suppressed..sHKq122KrrrEs r8!is_unreachable_warning_suppressedz7ConditionalTypeBinder.is_unreachable_warning_suppressedsHDKKHHHr7cNt|}|Jd|j|y)z0Remove all references to a Node from the binder.Nz0Internal error: binder tried cleanse non-literal)r _cleanse_key)rArxrcs r8cleansezConditionalTypeBinder.cleanses*4 R RR #r7c^|jD]}||jvs|j|= y)z/Remove all references to a key from the binder.N)rTr=)rArcframes r8rz"ConditionalTypeBinder._cleanse_keys([[Eekk!KK$!r7c td|D}|Dcgc]}|jr|}}d}|Dchc]}|jD]}|}}}|D]}|j|}|Dcgc]}|jj || }}t d|DrQ|rtd|Drf|d} | J| j t|jj |} t| tr3tfd|ddDsttj| n|ddD]} | Jt|j|| j ttr ttt sWttshtj"j$D cgc]} tt| tr|  c} } | |j|k(s| |t'|dr|j)|d d }| |j*d _|Scc}wcc}}wcc}wcc} w) zUpdate the frame to reflect that each key will be updated as in one of the frames. Return whether any item changes. If a key is declared as AnyType, only update it if all the options are the same. c36K|]}|j ywr_rrs r8rz.s>v! -vsFc3$K|]}|du ywr_r6rxs r8rz.s7&619&6sc3BK|]}|duxr |j ywr_rtrs r8rz.s*%AQA 7a&7&7"77AQsrNc3VK|] }|duxrt|j"ywr_)rr/)rtr/s r8rz.s-J^QATM@l4&@@J^s&)rQ) source_anyTrtro)allr>r=rsrrr/r)rVrvrr$from_another_anyrr&collapse_variadic_unionr! make_unionitemsrrlrT)rArT all_reachablerhchangedrckeys current_valueresulting_values current_typedeclaration_typeotherr simplifiedr/s @r8update_from_optionsz)ConditionalTypeBinder.update_from_optionssM>v>> #9V1==!V9#7VqwwwV7C IIcNMIOPA C ? P7&677%AQ%"+A.L+ ++$$D.t/@/@/D/DS/IJ *G4JZ[\[]J^#9#=#=JZ-ab1E ,,,&t'8'8'=tUZZPD"$ 26t<!$ 3 48S&/%9%9(, b 1*_UVEWY`:aQ b& &):):3)??#-D)2*$L}Q?O,P #tT :kn+1j B#{:7 QZcs!III /#II !I c |dkDr|j| |jj}|jj}|r|j d|jd|j ||_|S)zcPop a frame and return it. See frame_context() for documentation of fall_through. rro) allow_jumprTpoprUinsertrrX)rAcan_skip fall_throughresultoptionss r8 pop_framezConditionalTypeBinder.pop_frame so !  OO\M *"((,,.  NN1dkk"o . $ 8 8 A r7c#Kd}|j |j}tt|_|j||_yw)aPush a new map to collect assigned types in multiassign from union. If this map is not None, actual binding is deferred until all items in the union are processed (a union of collected items is later bound manually by the caller). N)rOrlist)rAold_assignmentss r8accumulate_type_assignmentsz1ConditionalTypeBinder.accumulate_type_assignmentssE  ,"33O +D 1### /sAAc >t|}|j!|j|j||fyt|tt t fsyt|sy|j||yt||syt|}t|}t|j||}t|tr|s|j||nt|trt|trtd|j Drtt|j|t#t"rP|j Dcgc]}tt|t"r|n|!} }|j|t| nat|tr?t|trtd|j Ds|j||n|j|||j$D]} |j'| ycc}w)Nc3NK|]}tt|tywr_)rvr)rritems r8rz4ConditionalTypeBinder.assign_type..`s]L\DJt4h?L\#%c3NK|]}tt|tywr_)rvr)rrs r8rz4ConditionalTypeBinder.assign_type..ms\K[4Jt4g>K[r)rrOrgrvrrrrinvalidate_dependenciesrr)most_recent_enclosing_typerrzr&rrrrZr) rArxr/ declared_type restrict_any p_declaredp_typeenclosing_typer new_itemsrrs r8 assign_typez!ConditionalTypeBinder.assign_type.s 16  ,  ! !$ ' . .m/D E $J AB t}  $$T*   $ . $]3  &()H)Ht)TU ng .| HHT> * vw ':y1]JL\L\]] ? ?hj QRT\',,,D#?4#8(CM,  HHT9Y/ 0  ( z9 -\:K[K[\\ HHT= ) HHT4 A OOA !s$Hct|}|J|jj|tD]}|j |y)aInvalidate knowledge of types that include expr, but not expr itself. For example, when expr is foo.bar, invalidate foo.bar.baz. It is overly conservative: it invalidates globally, including in code paths unreachable from here. N)rrWrrYr)rArxrcdeps r8rz-ConditionalTypeBinder.invalidate_dependencieszsE4 $$((ce4C   c "5r7c Pt|}t|tr t|St |}|Jt|g|j Dcgc]E}||j vr5t||j |dr|j |jGc}z}|dScc}wrn) r)rvrrwrrTr=rr/)rArxr/rcrh encloserss r8rz0ConditionalTypeBinder.most_recent_enclosing_typest$ dG $"4( (4 $T*+[[/  agg~*T1773<?"C GGCL   /  } / sA B#cL|dkr|t|jz }t|j}|j|dzdD];}|j j |j |js5d|_=|j|j|y)NrrQT) rprUr:rSrTr=updater>rg)rArkrrhs r8rz ConditionalTypeBinder.allow_jumps 19 S//0 0Edlln%UQY[)A KK  qww '}}$(!* u%,,U3r7c`|j|jd|jyNro)rr[r>rEs r8 handle_breakz"ConditionalTypeBinder.handle_breaks% ))"-. r7c`|j|jd|jyr)rr\r>rEs r8handle_continuez%ConditionalTypeBinder.handle_continues% ,,R01 r7rQr)r break_framecontinue_framer? try_framec#Kt|jdkDsJ|r1|jjt|j|z |r1|jjt|j|z |r1|j j t|jdz |j|}|r|jd||j|||r|jj|r|jj|r2|j jt|jdz yyw)a4Return a context manager that pushes/pops frames on enter/exit. If can_skip is True, control flow is allowed to bypass the newly-created frame. If fall_through > 0, then it will allow control flow that falls off the end of the frame to escape to its ancestor `fall_through` levels higher. Otherwise control flow ends at the end of the frame. If break_frame > 0, then 'break' statements within this frame will jump out to the frame break_frame levels higher than the frame created by this call to frame_context. Similarly for continue_frame and 'continue' statements. If try_frame is true, then execution is allowed to jump at any point within the newly created frame (or its descendants) to its parent (i.e., to the frame that was on top before this call to frame_context). After the context manager exits, self.last_pop_changed indicates whether any types changed in the newly-topmost frame as a result of popping this frame. rQroN) rprTr[rgr\rZrarirrrremove)rArrrrr?r new_frames r8 frame_contextz#ConditionalTypeBinder.frame_contextsF4;;!###     $ $S%5 %C D   ' 'DKK(8>(I J  OO  DKK 01 4 5OO$56  OOB  x.     ! ! #   $ $ &  OO " "3t{{#3a#7 8 sEEc#Kt|jdk(sJ|j|jddt|jdk(sJyw)zqA variant of frame_context for use at the top level of a namespace (module, function, or class). rQTrN)rprTrirrEs r8top_frame_contextz'ConditionalTypeBinder.top_frame_contextsQ 4;;1$$$oo tQ4;;1$$$sAA)rIrJ)rIrHr_)rcrrdz Key | NonerIrJrG)r?r0rIr:)ro) rcrr/r#r1r0rkrHrIrJ)rcrrkrHrIzCurrentType | None)rxrryr#r1r0rIrJ)rxrrI Type | None)rIr0)rxrrIrJ)rcrrIrJ)rTz list[Frame]rIr0)rr0rrHrIr:)rIzIterator[Assigns]) rxrr/r#rrrr0rIrJ)rxr,rIrJ)rxr,r/r#rIr)rkrHrIrJ)rr0rrHrrHrrHr?r0rr0rIIterator[Frame])rIr) r2r3r4rLrOr5rBrSrbrirlrsrzr>r@rrrrrrrrrrrrrrrrrr6r7r8rNrN@sX,(,n+-@/KKO -+=7 I % EN& 0 0^cJJ&*J;FJVZJ JX #  4 "'7979 79  79  79 7979 7979r%%r7rNc,t|trt|jtr2|jj}tt |t s|Syt|jtrtt|jSyr_) rvrnoderr/r)r rr%r*)rxr/s r8rwrwsh$ dii %99>>Dod3[A   8 ,2499=> > r7cg}g}|jD]@}t|}t|tr|j |0|j |Bt |dkr|St |d}|d}|d}t|j}||S|j|}t|tsJt|j} t| ts|S| jjdk(sJ|j|dzd} t |jt | kr|S| r|jt |  d| k7r|S| r|jdt |  } n |j} | jd} t|ddD]%\} }|j| | g| dzzz| zk7s#|cS|j| | gt |jdz zz|gz| zk7r|St |jdk(r| j}nt| |gz| z|j}t!j"|g|zS) aSimplify a union involving variadic tuple if possible. This will collapse a type like e.g. tuple[X, Z] | tuple[X, Y, Z] | tuple[X, Y, Y, *tuple[Y, ...], Z] back to tuple[X, *tuple[Y, ...], Z] which is equivalent, but much simpler form of the same type. rQc,t|jSr_)rpr)rs r8z)collapse_variadic_union.. s CLr7)rcrroNzbuiltins.tuple)fallback)rr)rvr"rgrpsortedr(r'r/rfullnameargs enumerate copy_modifiedpartial_fallbackr&r)ry tuple_items other_itemsrp_tfirstlast unpack_indexunpackunpackedsuffixprefixargrritrs r8rrs<KK YYa  c9 %   s #   q !   ;1 *@AK NE r?D&tzz2L ZZ %F fj )) )v{{+H h ) == ! !%5 55 5 ZZ q(* +F 5;;#f+%  %++s6{ln-7  ^F |, -- C;q,-2 88vQ/&8 8J.  zzVses399~'9::fXENN  5;;1#113 v069DDYDYZ    { : ;;r7N)rxr,rIr)ryr&rIr#)= __future__r collectionsr contextlibrtypingrrrr r r r typing_extensionsr r+mypy.erasetyper mypy.joinr mypy.literalsrrrr mypy.nodesrrrrrrr mypy.subtypesrr mypy.typesrrrr r!r"r#r$r%r&r'r(r) mypy.typevarsr*r,r5r.r:AssignsrNrwrr6r7r8rs"#%RRR5<!==ZZZ21!&y*h'F!GJG* __. j$uT8D>-A'B"CC Dj%j%Z 9