o i@sdZddlZddlZddlZddlZddlZddlmZddl m Z m Z ddl m Z ddlmZmZmZddlmZmZdd lmZmZdd lmZmZdd lmZdd lmZmZm Z m!Z!m"Z"m#Z#m$Z$d dl%m&Z&m'Z'd dl(m)Z)e rddl*m+Z+ddl,m-Z-ddZ.d.ddZ/ddZ0Gddde'Z1Gddde'Z2Gddde1Z3Gddde1Z4Gd d!d!e1Z5Gd"d#d#e5Z6Gd$d%d%e5Z7Gd&d'd'e'Z8Gd(d)d)e8Z9Gd*d+d+e8Z:Gd,d-d-e8Z;dS)/a Dictionary-related variable tracking classes for PyTorch Dynamo. This module implements variable tracking for different types of dictionary-like objects: - Regular Python dictionaries (dict) - Ordered dictionaries (collections.OrderedDict) - Default dictionaries (collections.defaultdict) - Dictionary views (keys and values) - Sets and frozensets (implemented internally using dictionaries) These classes are responsible for tracking dictionary operations during graph compilation, maintaining proper guards for dictionary mutations and key existence checks. They handle dictionary creation, modification, key/value access, and view operations while ensuring correct behavior in the compiled code through appropriate guard installation. The implementation uses a special _HashableTracker wrapper to handle dictionary keys while preserving proper aliasing semantics. Sets are implemented as dictionaries with None values for efficiency and code reuse. NHashable)Optional TYPE_CHECKING)is_fake)graph_break_hints polyfills variables)create_call_functioncreate_instruction)raise_observed_exceptionunimplemented_v2) GuardBuilder install_guard)is_from_local_source)cmp_name_to_op_mapping dict_items dict_keys dict_valuesistyperaise_args_mismatchspecialize_symnode)ValueMutationNewVariableTrackerConstantVariable) PyCodegenInstructionTranslatorcCst|jddS)N__instancecheck__F)type__dict__get)objr&S/home/ubuntu/vllm_env/lib/python3.10/site-packages/torch/_dynamo/variables/dicts.pywas_instancecheck_override=r(cCs>|durddlm}|}tt|tdt|gddS)Nrrzunhashable type: args)torch._dynamo.symbolic_convertr current_txr TypeErrorrr")argtxr r&r&r'raise_unhashableAs   r1cCst|tjr|s|rdSt|tjr!|jj dduSt|tj r1t dd|j DSt|tj rCt dd|jDSt|tjrdt|jsdt|jdtjurdt|jtrdt|jtSt|tjtjtjtjtjtjtjtjtj j!tj"tj#tj$tj%tj&tj'tj(tj)fS)NT example_valuecs|]}t|VqdSN is_hashable.0er&r&r' ]zis_hashable..csr3r4r5r7r&r&r'r:_r;__hash__)* isinstancer LazyVariableTracker is_realizedr6TensorVariableas_proxynodemetar$ TupleVariableallitemsFrozenDataClassVariablefieldsvaluesUserDefinedObjectVariabler(valueinspectgetattr_staticintr< py_HashableBuiltinVariableSymNodeVariabler EnumVariableFrozensetVariableUserDefinedClassVariableUserFunctionVariableSkipFunctionVariablemisc NumpyVariableNNModuleVariableUnspecializedNNModuleVariableMethodWrapperVariableTorchInGraphFunctionVariableTypingVariableFunctoolsPartialVariableWeakRefVariable TorchHigherOrderOperatorVariable)xr&r&r'r6KsT       r6csLeZdZejZdhejZGdddZe fde eefddffdd Z d d Z d d Z d dZ ddZddZddZdefddZdefddZdefddZddZddZd?d!d"Zd#d$d%efd&d'Zd#d$d%efd(d)Zd%efd*d+Zd%efd,d-Zd.d/Zd0d1Z 2 3 4 5  6d@fd7d8 Zd9d:Z d;d<Z!fd=d>Z"Z#S)AConstDictVariableuser_clsc@sLeZdZdZdddZeddZdd Zed d Z d dde fd dZ dS)z"ConstDictVariable._HashableTrackeraB Auxiliary opaque internal class that wraps a VariableTracker and makes it hashable This should not be seen or touched by anything outside of ConstDictVariable and its children Note that it's also fine to put VTs into dictionaries and sets, but doing so does not take into account aliasing returnNcCs"t|}t|s t|||_dSr4)rr6r1vt)selfrer&r&r'__init__s z+ConstDictVariable._HashableTracker.__init__csNt|jtjr|js|jr|jSt|jtjr(|jj j d}|St|jtj rAt j tfdd|jjD}|St|jtjrL|jjSt|jtjrW|jjSt|jtjrc|jSt|jtjrtt j |jjjSt|jtjrt j fdd|jjD}tj|j|St|jtjr|jjS|j}|S)Nr2c3s|]}|jVqdSr4underlying_valuer7rr&r'r:zFConstDictVariable._HashableTracker.underlying_value..csi|] \}}||jqSr&rhr8kvrr&r' szGConstDictVariable._HashableTracker.underlying_value..)r=rer r>r?r6original_valuer@rArBrCrDrb_HashableTrackertuplerFrYrKrZrU get_functionr_ referent_vtrirGrH HashWrapper python_typerJas_python_constant)rfra fields_valuesr&rr'risF       z3ConstDictVariable._HashableTracker.underlying_valuecC t|jSr4)hashrirfr&r&r'r<s z+ConstDictVariable._HashableTracker.__hash__csht|t|kr dSt|tr(tjt|t|ko'tfddt||DSt|r0||uS||kS)NFc3s |] \}}||VqdSr4)_eq_impl)r8urmrr&r'r:s z>ConstDictVariable._HashableTracker._eq_impl..) r"r=rqrbrplenrEzipr)abr&rr'r{s z+ConstDictVariable._HashableTracker._eq_implothercCsNtj}t||st|sJt|t||r ||j|jS||j|Sr4)rbrpr=r is_literalr"r{ri)rfrrr&r&r'__eq__s z)ConstDictVariable._HashableTracker.__eq__rdN) __name__ __module__ __qualname____doc__rgpropertyrir< staticmethodr{boolrr&r&r&r'rps  ' rprFrdNc sd|vr |dd|vr|dtjd i|tjtfdd|Ds-Jfdd||}|fdd|D|_t|j  |_ | |_ ||_ dS) Noriginal_itemsshould_reconstruct_allc3s,|]\}}t|tfot|tVqdSr4)r=rr8rarmrr&r'r:s  z-ConstDictVariable.__init__..cst|r|S|Sr4)r=)keyrr&r' make_hashablesz1ConstDictVariable.__init__..make_hashablecsi|] \}}||qSr&r&r)rr&r'rnz.ConstDictVariable.__init__..r&)popsuperrgrbrprErF_get_dict_cls_from_user_clsrsourcercopyrrc)rfrFrckwargsdict_cls __class__)rrr'rgs       zConstDictVariable.__init__csVttjtjf|vr|}n tfdd|jD}|vs"J||tjur)t}|S)Nc3s|] }|vr|VqdSr4r&)r8baseaccepted_dict_typesr&r'r:sz@ConstDictVariable._get_dict_cls_from_user_cls..)dict collections OrderedDict defaultdictnext__mro__)rfrcrr&rr'rs  z-ConstDictVariable._get_dict_cls_from_user_clscCdd|jDS)NcS i|] \}}|j|qSr&rerArkr&r&r'rn z.ConstDictVariable.as_proxy..rFrzr&r&r'rAzConstDictVariable.as_proxycCs"dddd|jDdS)N{, css,|]\}}|jd|VqdS)z: Nre debug_reprrkr&r&r'r:s z/ConstDictVariable.debug_repr..})joinrFrzr&r&r'rs zConstDictVariable.debug_reprcCr)NcSrr&rervrkr&r&r'rn z8ConstDictVariable.as_python_constant..rrzr&r&r'rvsz$ConstDictVariable.as_python_constantcCs|dd|jDS)NcSsi|] \}}|j|qSr&rrkr&r&r'rn'sz=ConstDictVariable.keys_as_python_constant..)install_dict_keys_match_guardrFrzr&r&r'keys_as_python_constant%z)ConstDictVariable.keys_as_python_constantcC|jSr4)rcrzr&r&r'ru)zConstDictVariable.python_typecCsBt|tsJtj}t|o |||jvo t|j||tj Sr4)r=rrbrpr6rFr DeletedVariable)rfrerr&r&r' __contains__,s zConstDictVariable.__contains__cCstdd|jDS)Ncss|] }t|tj VqdSr4)r=r rr8rar&r&r'r:6s z(ConstDictVariable.len..)sumrFrIrzr&r&r'r}5szConstDictVariable.lencs"jptfddjDS)Nc3s*|]\}}j|j|VqdSr4) is_new_itemrr$re)r8rrKrzr&r'r:;s  z2ConstDictVariable.has_new_items..)ranyrFrzr&rzr' has_new_items:szConstDictVariable.has_new_itemscCs<|r|r|rt|t|kSt|t|kSr4)r?idrealize)rfrKrr&r&r'r@szConstDictVariable.is_new_itemcCshd}|jD]!\}}|j|j}|||s|jr(||j|||d7}q|td|ddS)Nrr BUILD_MAPr/)rFrr$rerr append_outputr )rfcodegennum_argsrrKitemr&r&r'reconstruct_kvs_into_new_dictFs z/ConstDictVariable.reconstruct_kvs_into_new_dictrrcsJ|jtjurfdd|tdddS|dS)NctdgS)Nr extend_outputcreate_load_python_modulercreate_load_attrr&rr&r'V z/ConstDictVariable.reconstruct..rF)rcrr add_push_nullrrr rfrr&rr' reconstructRs   zConstDictVariable.reconstructr0r r/cCs(t|}||jvrtt||j|Sr4)rbrprFr KeyError)rfr0r/rr&r&r''getitem_const_raise_exception_if_absentbs    z9ConstDictVariable.getitem_const_raise_exception_if_absentcCsLt|}||jvr!d|jd}tdd|j|dgtjd|j|S)NzDictionary key z not found during tracingzkey not found in dictzKey z>Check if the key exists in the dictionary before accessing it.gb_typecontext explanationhints)rbrprFrKrr USER_ERROR)rfr0r/rmsgr&r&r' getitem_constjs    zConstDictVariable.getitem_constcCs"t|}||jvr dS|j|Sr4)rbrprF)rfr/rr&r&r'maybe_getitem_constys   z%ConstDictVariable.maybe_getitem_constcCsX||vsJt|}t|j|}t|j|}t|tj r*| dSdSr4) rbrprqrFkeysindexrr=r r>r)rfr/rroriginal_key_vtr&r&r'realize_key_vts    z ConstDictVariable.realize_key_vtcCs|jr t|tjdSdSr4)rr make_guardrDICT_KEYS_MATCHrzr&r&r'rsz/ConstDictVariable.install_dict_keys_match_guardcCsddlm}|js dS|jj|rdS|d|v}|djdur>t|d|r>t|t j t |j |dj | ddS|djrT|rN||ddS|dSdS)Nrrr)rinvert)rroutput side_effects is_modifiedr=rr functoolspartialr"CONTAINS_GUARDrKrr)rfr0r+rcontainsr&r&r'install_dict_contains_guards*    z-ConstDictVariable.install_dict_contains_guardr+list[VariableTracker]rdict[str, VariableTracker]rcsV ddlm}mtj|ot|d}|dkr9ttj|g|Ri|}|j j ||j |j dS|dkrPt|dkrHt|||||dS|dkrp|sX|r]t||||jrl|j j|jt|S|dkrt|r}t||||jr|j j|jt|S|dkr|s|rt||||jr|j j|j|s|rtt|t|S|d kr||s|rt|||j|j tdd S|d kr|s|rt|||t|j S|d kr)|r)|st|d||st|d ksJ|j j ||d|j |d<dS|dkrS|rS|rS|d|_ |j j ||j !|ddS|dkrt|dvrdt|||smt|d|d|vr|"||t|dkrdS|dS|#||dS|dkr|rt|dvrt|||st|d|d|vr|"||t|dkrtt$||dSd|_ |j j ||j %|dS|dkrx|rxt&|j'trt&|j't(j)st|rt|||j sd}tt$||gd|j't(j)ur_t|dks)d|vr_t|dkr>t*|dr>|dj+} n|,d} rPt*| rP| j+} nt|||j j-| d\} } n|j -\} } d|_ |j j |t.| j/| gS|dkr|s|rt||d|_ |j j ||j 0dS|dkr |r |t|dk} t|dk} | s| r|j j || rt*|dtr|d|d}n |1|t|d}|j |j | rfdd| D}|j |dSt23||||S|dkr5t|st|||s$t|d|"|||d|v}|S|dkr|rt|dvrKt|||sTt|d||r]Jt|d ksfJ|4|d}|durt|St|dkrd}n|d}|j j |||j |d<|S|dkr||j j ||d|vrtt$|d} t|d krt*|dr|dj+} |rd|vrt*|dr|,dj+} |d}|j j5|| ddS|dkrt6|trt|dkr t||t7t8j9:|||dgiS|d kr*|3|d||j+ S|d!krt|dks8J|d}t6|ttj;fs\d"|<j=d#|<j=d$}tt||gd|j'|j'h}t>d%d&|Drot(j)nt}||j|j td|d'}|d|j |dj |S|d(kr|3|d|||St23||||S))Nr)rPrrrg __getitem__rFrrIr)rF mutation_typer__len__ __setitem__r __delitem__Tr$)rrrpopitemzpopitem(): dictionary is emptyr*last)rclearupdatecs i|] \}}||qSr&)createrkrrr&r'rn]rz1ConstDictVariable.call_method..r setdefault move_to_endr__ne____or__z$unsupported operand type(s) for |: 'z'and ''css|] }t|tjVqdSr4) issubclassrr)r8tr&r&r'r:sz0ConstDictVariable.call_method..)rFrrrc__ior__)?rrPrrbrpr6r r call_dictrrmutationrFrrr}rrrrguard_on_key_orderaddDictItemsVariableDictKeysVariabler r.DictValuesVariableclonerr is_mutabler1rrrrrrrrcrrr=rKr$rrDrercall_custom_dictr call_methodrrrrUr dict___eq__ call_functionUserDefinedDictVariablerurr)rfr0namer+rrP arg_hashable temp_dict_vtrrrmrlhas_arg has_kwargsdict_vtrrKrarrtsrc new_dict_vtrrr'r s                                                                       zConstDictVariable.call_methodcCs|dd|jDS)NcSg|]}|jqSr&rerr&r&r' z9ConstDictVariable.unpack_var_sequence..)rrFrrfr0r&r&r'unpack_var_sequencerz%ConstDictVariable.unpack_var_sequencecCs\|jtur||jjvrtdStdSd|jd}tdd|j|dgtjddS) NTFz hasattr on z is not supportedzunsupported hasattr operationzClass z+Consider using a regular dictionary insteadr)rcrr#rrrr SUPPORTABLE)rfr0rrr&r&r'call_obj_hasattrs      z"ConstDictVariable.call_obj_hasattrc s|tjdi|SNr&)rrr )rfrrr&r'r szConstDictVariable.clonerrr+rrrrdr)$rrrr DICT_CONTAINSrr_nonvar_fieldsrprrgrrArrvrrurrrNr}rrrrrrrrrrr rrr  __classcell__r&r&rr'rbs^X #     '-rbcsbeZdZdeddffdd ZddZdd Zdd d Zdedde e dfddfddZ Z S)MappingProxyVariabledv_dictrdNc s*tjdi|t|tsJ||_dSr )rrgr=rbr'rfr'rrr&r'rgs zMappingProxyVariable.__init__cCstjSr4)typesMappingProxyTyperzr&r&r'rurz MappingProxyVariable.python_typecCs |j|Sr4)r'rrr&r&r'rs z(MappingProxyVariable.unpack_var_sequencerrcsd|jrd|jd}tdd|j|dgtjdfdd|jtd d dS) Nz*Preexisting MappingProxyVariable (source: zS) cannot be reconstructed because the connection to the original dict will be lost.z%mapping proxy cannot be reconstructedSource: zCUse a mapping proxy constructed in the same `torch.compile` region.rcr)Nr*)rrr)rr&rr&r'rrz2MappingProxyVariable.reconstruct..rF)rrrrrr'rr )rfrrr&rr'rs"    z MappingProxyVariable.reconstructr+rrcCsF|jr|jjrd}tdd|jd|ddgd|j||||S)NaA dict has been modified while we have an existing mappingproxy object. A mapping proxy object, as the name suggest, proxies a mapping object (usually a dict). If the original dict object mutates, it is reflected in the proxy object as well. For an existing proxy object, we do not know the original dict it points to. Therefore, for correctness we graph break when there is dict mutation and we are trying to access a proxy object.z-mapping proxy affected by dictionary mutationr+z, Dict mutation detectedzNAvoid modifying dictionaries that might be referenced by mapping proxy objectszROr avoid using the mapping proxy objects after modifying its underlying dictionaryr)rrrhas_existing_dict_mutationrr'r )rfr0rr+rrr&r&r'r "s   z MappingProxyVariable.call_methodr!) rrrrbrgrurrlistrstrr r%r&r&rr'r&s  r&c@seZdZddZddZdS)NNModuleHooksDictVariablecCdSr4r&rzr&r&r'rBz7NNModuleHooksDictVariable.install_dict_keys_match_guardcCr0r4r&rfr0r+r&r&r'rEr1z5NNModuleHooksDictVariable.install_dict_contains_guardN)rrrrrr&r&r&r'r/@s r/csfeZdZddfdd ZfddZfddZed d Z   dfdd ZddZ Z S)DefaultDictVariableNrdc s.tj||fi||tjusJ||_dSr4)rrgrrdefault_factory)rfrFrcr4rrr&r'rgJs zDefaultDictVariable.__init__cs$|jtttfvr |js dStS)NF)r4r-rqrrFris_python_constantrzrr&r'r5Os z&DefaultDictVariable.is_python_constantcsd|jdtdS)Nz defaultdict(r))r4rrrzrr&r'rVszDefaultDictVariable.debug_reprcCs,t|tjr|jttttfvSt|tjj Sr4) r=r rPfnr-rqrset functionsBaseUserFunctionVariablerr&r&r'is_supported_arg[s z$DefaultDictVariable.is_supported_argr+rrrrcs|dkr=t|dks J|d|vr|||dS|jdur&t|d|j|gi}t|d|d|f||St||||S)Nrrrr)r}rr4rrrr )rfr0rr+r default_varrr&r'r bs  zDefaultDictVariable.call_methodcs:fdd|j|tdddS)Ncr)Nrrr&rr&r'r}rz1DefaultDictVariable.reconstruct..rF)rr4rrr rr&rr'rzs    zDefaultDictVariable.reconstructr4rr") rrrrgr5rrr;r rr%r&r&rr'r3Is   r3cseZdZdZejZdeeddffdd Z ddZ e d d Z e d d Zd dZddZddZd'ddZddZdeedeeefddffdd Zddd efd!d"Zd#d$Zfd%d&ZZS)( SetVariablez.We model a sets as dictionary with None valuesrFrdNc s(t|t}tj|fi|dSr4)rfromkeysr=_default_valuerrgrfrFrrr&r'rgszSetVariable.__init__cC,|jsdSdddd|jDdS)Nzset()r,cs|]}|jVqdSr4rr8rlr&r&r'r:rjz)SetVariable.debug_repr..rrFrrrzr&r&r'r"zSetVariable.debug_reprcCst|jSr4)r8rFrrzr&r&r' set_itemsszSetVariable.set_itemscCs tdSr4)rrr&r&r&r'r?s zSetVariable._default_valuecCdd|jDS)NcSh|]}|jqSr&rrDr&r&r' z'SetVariable.as_proxy..rGrzr&r&r'rAzSetVariable.as_proxycCtSr4)r8rzr&r&r'rur1zSetVariable.python_typecCrH)NcSrIr&rrDr&r&r'rJrKz1SetVariable.as_python_constant..rLrzr&r&r'rvrMzSetVariable.as_python_constantrrcCs2|dd|jD|tdt|jddS)NcSrr&rrr&r&r'rrz+SetVariable.reconstruct.. BUILD_SETr)foreachrGrr r}rr&r&r'rszSetVariable.reconstructc Csz|dd|g|Didd|D}Wn"ty:}ztt||tttj|jdWYd}~nd}~wwt ||S)NcSsg|]}|qSr&rvrr&r&r'rsz0SetVariable._fast_set_method..cSsi|] \}}||qSr&rQrkr&r&r'rnrz0SetVariable._fast_set_method..r*) rF Exceptionr r"r-maprrr+rbuild)rfr0r7r+rresexcr&r&r'_fast_set_methods  zSetVariable._fast_set_methodr+rrc sddlm}|dvr$|||r$|tur$|}||t||||S|dkrNttj|g||R}|j j ||j |j |j tdS|dkrm|rVJt|dkrat||d}|dtf}n|d kr|ruJ|ryJz|jj}Wn ty} ztt|tttj| jd WYd} ~ nd} ~ wwt|||f||S|d krt|dkrt|||rJtt j!"|||dgiS|d kr|rJtt j#"||g|iS|d kr|rJtt j$"||g|iS|dkr|rJtt j%"||g|iS|dkr,|rJtt j&"||g|iS|dkrD|r6Jtt j'"||g|iS|dkrht|dkrUt|||rZJtt j("||g|iS|dkrt|dkryt|||r~Jtt j)"||g|iS|dkr|*r|rJtt j+"||g|iS|dkr|rJt|dksJ|d|vrtt||d t|d ||S|dkr|rJt|dksJ|d|vrt|d ||StjddS|dvr:t|dkr t||t,j-t,j.d} |d/} t0| ts+tt"|| gi} t| 1|"||| giS|dvryd dddd1|} t2|dttj3fsqtd|d|4d|d4d} tt5|| gd ||| ||S|dvrt2|dttj3fstd|d|4d|d4d} tt5|| gd d dddd1|} ||| |||S|dkrt2|dttj3fstdS||d||}tt|jdkS|t6vrt2|dttj3fstt7Stt6||j|djSt||||S) Nr)check_constant_args) isdisjointunion intersection differencesymmetric_differencergrrrrrr*rYr[intersection_updaterZr\difference_updater]symmetric_difference_updaterremovediscard)rK)issubset issuperset)__and__r__xor____sub__z unsupported operand type(s) for z: 'z' and 'r)__iand__r__ixor____isub__rF)8utilsrXrur8rWgetattrr rPcall_setrrrrFrrrrr}rr=r?rGrrerr r-rSr+rr rUr set_isdisjointrset_intersectionset_intersection_update set_unionset_differenceset_difference_updateset_symmetric_differenceset_symmetric_difference_updater  set_updateoperatorlegerrr$r=UserDefinedSetVariablepython_type_namer.rNotImplemented)rfr0rr+rrXpy_type temp_set_vtresultr9oprmrrrr&r'r sH                                    " "    zSetVariable.call_methodr0r r/cCstd)NzIllegal to getitem on a set) RuntimeError)rfr0r/r&r&r'rhszSetVariable.getitem_constcCr0r4r&rzr&r&r'rkz)SetVariable.install_dict_keys_match_guardcst||dSr4)rrr2rr&r'ror)z'SetVariable.install_dict_contains_guardr!)rrrrr SET_CONTAINSrr-rrgrrrGrr?rArurvrrWrr.r rrrr%r&r&rr'r=s:     (r=cs~eZdZdeeddffdd ZddZedd Zd d Z d d Z dddZ deede e efddffdd ZZS)rSrFrdNc tj|fi|dSr4rrgr@rr&r'rgtzFrozensetVariable.__init__cCrA)Nz frozenset()rrBcsrCr4rrDr&r&r'r:rjz/FrozensetVariable.debug_repr..rrErzr&r&r'r{rFzFrozensetVariable.debug_reprcCs |jSr4)rFrrzr&r&r'rG zFrozensetVariable.set_itemscCrNr4) frozensetrzr&r&r'rur1zFrozensetVariable.python_typecCstdd|jDS)NcSrIr&rrDr&r&r'rJrKz7FrozensetVariable.as_python_constant..)rrGrzr&r&r'rvrz$FrozensetVariable.as_python_constantrrcs<dd|jDfddtdddS)NcSrr&rrr&r&r'rrz1FrozensetVariable.reconstruct..csdgS)Nr)rcreate_load_globalr&rr&r'rsz/FrozensetVariable.reconstruct..rF)rPrGrrr rr&rr'rs  zFrozensetVariable.reconstructr+rrcs`|dvr td|d|dkrtdS|dvr't||||}t|jSt||||S)NrrrrarbrIllegal call_method z on a frozensetrg)rr\r[r])rrrrr rSrF)rfr0rr+rrrr&r'r s  zFrozensetVariable.call_methodr!)rrrr-rrgrrrGrurvrrr.r r%r&r&rr'rSss&    rScseZdZdeeddffdd ZddZdd Zd d Ze d d Z ddZ ddZ deede eefddffdd ZZS)DictKeySetVariablerFrdNc rr4rr@rr&r'rgrzDictKeySetVariable.__init__cCrA)Nz dict_keys([])z dict_keys([rBcsrCr4rrDr&r&r'r:rjz0DictKeySetVariable.debug_repr..z])rErzr&r&r'rszDictKeySetVariable.debug_reprcCr0r4r&rzr&r&r'rrz0DictKeySetVariable.install_dict_keys_match_guardcCr0r4r&r2r&r&r'rrz.DictKeySetVariable.install_dict_contains_guardcCrr4rrzr&r&r'rGszDictKeySetVariable.set_itemscCrNr4rrzr&r&r'rur1zDictKeySetVariable.python_typecCstdd|jDdS)NcSrIr&rrDr&r&r'rJrKz8DictKeySetVariable.as_python_constant..)rr>rGrrzr&r&r'rvs z%DictKeySetVariable.as_python_constantr+rrcs*|dvr td|dt||||S)Nrrz on a dict_keys)rrr rfr0rr+rrr&r'r szDictKeySetVariable.call_method)rrrr-rrgrrrrrGrurvrr.r r%r&r&rr'rs(   rcseZdZUdZdZeeed<deddffdd Z e dd Z e d d Z d d Z dddZddZdeddeedfddffdd ZZS)DictViewVariablezu Models _PyDictViewObject This is an "abstract" class. Subclasses will override kv and the items method Nkvr'rdc s8tjdi||jdvsJt|tsJ||_dS)N)rrIrFr&)rrgrr=rbr'r(rr&r'rgs zDictViewVariable.__init__cCst|jj|jSr4)rlr'rFrrzr&r&r' view_itemsszDictViewVariable.view_itemscCstr4)NotImplementedErrorrzr&r&r' view_items_vtszDictViewVariable.view_items_vtcCrr4)rrr&r&r'rrz$DictViewVariable.unpack_var_sequencerrcCs$||j||j|ddS)Nr)r' load_methodrr rr&r&r'rs  zDictViewVariable.reconstructcCs"||jvr tdStdS)NTF)rur#rr)rfr0rr&r&r'rs  z!DictViewVariable.call_obj_hasattrr+rrcs,|dkr |j||||St||||S)Nr)r'r rrrr&r'r  szDictViewVariable.call_methodr!)rrrrrrr.__annotations__rbrgrrrrrrr-rr r%r&r&rr'rs$     rcsZeZdZdZeddZeddZddZded d e e d fd d ffd d Z Z S)rrcCrxr4)r8rrzr&r&r'rGrzDictKeysVariable.set_itemscCrH)NcSrr&rrr&r&r'r!rz2DictKeysVariable.view_items_vt..rrzr&r&r'rzDictKeysVariable.view_items_vtcCrNr4rrzr&r&r'ru#r1zDictKeysVariable.python_typer+rrrdcs|dkr |j||||S|dvr"t|j|}||dj}t|S|tvrBt|dttfs4t t St t||j|djSt ||||S)Nr)rerhrrrgrjrfrir) r'r rlrGr=rr=rrrr|r)rfr0rr+rrrrr&r'r &s  zDictKeysVariable.call_method) rrrrrrGrrur-rr.r r%r&r&rr'rs   rc@$eZdZdZeddZddZdS)rrIcCrxr4)r-rrzr&r&r'rJrz DictValuesVariable.view_items_vtcCrNr4)rrzr&r&r'ruNr1zDictValuesVariable.python_typeNrrrrrrrur&r&r&r'rFs   rc@r)rrFcCrH)NcSs g|] \}}t|j|gqSr&)r rDrerkr&r&r'rXrz3DictItemsVariable.view_items_vt..rrzr&r&r'rUrzDictItemsVariable.view_items_vtcCrNr4)rrzr&r&r'ruZr1zDictItemsVariable.python_typeNrr&r&r&r'rRs   rr4)sN   $     5~F Ch@22/