o 2gW@sUddlmZddlmZddlmZmZddlmZ ddl m Z ddl m Z mZmZmZddlmZmZmZmZmZmZmZddlmZdd lmZdd lmZmZdd l m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/Gd d d eZ0d3ddZ1d4ddZ2d3ddZ3d4ddZ4d3ddZ5d4ddZ6d4d d!Z7e8d"d#hZ9d$e:d%<d3d&d'Z;d4d(d)Zd4d1d2Z?dS)7) annotations)partial)CallableFinalN)message_registry)DictExprIntExprStrExpr UnaryExpr)AttributeContextClassDefContextFunctionContextFunctionSigContext MethodContextMethodSigContextPlugin)try_getting_str_literals) is_subtype)is_literal_type_likemake_simplified_union)TPDICT_FB_NAMESAnyType CallableType FunctionLikeInstance LiteralTypeNoneType TupleTypeType TypedDictType TypeOfAny TypeVarType UnionTypeget_proper_typeget_proper_typesc@sVeZdZdZdddZdd d Zdd d ZdddZdddZd ddZ d!ddZ dS)" DefaultPluginz/Type checker plugin that is enabled by default.fullnamestrreturn(Callable[[FunctionContext], Type] | NonecCs\ddlm}m}m}|dkr|jS|dkr|jS|dkr%ddl}|jjj S|dkr,|j SdS)Nr)ctypesenumssingledispatchz _ctypes.Arrayzfunctools.singledispatchzfunctools.partialz enum.member) mypy.pluginsr*r+r,array_constructor_callback'create_singledispatch_function_callbackmypy.plugins.functoolsplugins functoolspartial_new_callbackenum_member_callback)selfr&r*r+r,mypyr7L/home/garg/my-data/venv/lib/python3.10/site-packages/mypy/plugins/default.pyget_function_hook+s zDefaultPlugin.get_function_hook3Callable[[FunctionSigContext], FunctionLike] | NonecCs>ddlm}m}|dvr|jS|dvr|jS|dkr|jSdS)Nrattrs dataclasses)z attr.evolvez attrs.evolvez attr.assocz attrs.assoc)z attr.fieldsz attrs.fieldszdataclasses.replace)r-r<r=evolve_function_sig_callbackfields_function_sig_callbackreplace_function_sig_callbackr5r&r<r=r7r7r8get_function_signature_hook;sz)DefaultPlugin.get_function_signature_hook1Callable[[MethodSigContext], FunctionLike] | NonecCsddlm}m}|dkrtS|ddtDvrtS|ddtDvr$tS|dkr+|jS||jkr3|j St }tD]}| |d| |d | |d | |d q8||vr]t SdS) Nrr*r,typing.Mapping.getcSh|]}|dqSz .setdefaultr7.0nr7r7r8 Oz:DefaultPlugin.get_method_signature_hook..cSrFz.popr7rHr7r7r8rKQrLz_ctypes.Array.__setitem__z.updatez.__or__z.__ror__z.__ior__) r-r*r,!typed_dict_get_signature_callbackr(typed_dict_setdefault_signature_callback!typed_dict_pop_signature_callbackarray_setitem_callback#SINGLEDISPATCH_CALLABLE_CALL_METHOD%call_singledispatch_function_callbacksetadd$typed_dict_update_signature_callback)r5r&r*r,typed_dict_updatesrJr7r7r8get_method_signature_hookHs( z'DefaultPlugin.get_method_signature_hook&Callable[[MethodContext], Type] | NonecCsddlm}m}|dkrtS|dkrtS|dkrtS|dkr tS|dvr&tS|dd tDvr1t S|d d tDvr.cSrFrMr7rHr7r7r8rKsrLcSrF)z .__delitem__r7rHr7r7r8rKurLz_ctypes.Array.__getitem__z_ctypes.Array.__iter__zfunctools.partial.__call__)r-r*r,typed_dict_get_callbackint_pow_callbackint_neg_callbackint_pos_callbacktuple_mul_callbackrtyped_dict_setdefault_callbacktyped_dict_pop_callbacktyped_dict_delitem_callbackarray_getitem_callbackarray_iter_callbackSINGLEDISPATCH_REGISTER_METHOD singledispatch_register_callbackREGISTER_CALLABLE_CALL_METHOD4call_singledispatch_function_after_register_argumentr0r1r2partial_call_callback)r5r&r*r,r6r7r7r8get_method_hookds:   zDefaultPlugin.get_method_hook)Callable[[AttributeContext], Type] | NonecCsPddlm}m}|dkr|jS|dkr|jS||jvr|jS||jvr&|jSdS)Nr)r*r+z_ctypes.Array.valuez_ctypes.Array.raw) r-r*r+array_value_callbackarray_raw_callbackENUM_NAME_ACCESSenum_name_callbackENUM_VALUE_ACCESSenum_value_callback)r5r&r*r+r7r7r8get_attribute_hooks  z DefaultPlugin.get_attribute_hook(Callable[[ClassDefContext], None] | NonecCsRddlm}m}||jvr|jS||jvs$||jvs$||jvs$||jvr'|j SdS)Nrr;) r-r<r=dataclass_makersdataclass_tag_callbackattr_class_makersattr_dataclass_makersattr_frozen_makersattr_define_makersattr_tag_callbackrAr7r7r8get_class_decorator_hooks     z&DefaultPlugin.get_class_decorator_hook(Callable[[ClassDefContext], bool] | NonecCsddl}ddlm}m}||jvr|jS||jjjvr |jjj S||j vr(|j S||j vr4t |j ddS||jvrAt |j dddS||jvrNt |j dddSdS)Nrr;T)auto_attribs_default)r|frozen_default)r| slots_default)r0r-r<r=rsdataclass_class_maker_callbackr1r2functools_total_ordering_makers'functools_total_ordering_maker_callbackruattr_class_maker_callbackrvrrwrx)r5r&r6r<r=r7r7r8get_class_decorator_hook_2s&      z(DefaultPlugin.get_class_decorator_hook_2N)r&r'r(r))r&r'r(r:)r&r'r(rC)r&r'r(rY)r&r'r(rj)r&r'r(rr)r&r'r(r{) __name__ __module__ __qualname____doc__r9rBrXrirqrzrr7r7r7r8r%(s    ! r%ctxrr(rcCs|j}t|jtrt|jdkrt|jddkrt|jddtrt|jdkrt|jdkrt|jddkr|jddj }t |jj |}|j }|r|jdd}t|trot|trot|j dkro|jtd}|jd}t|ts{J|j|jdt||gg|dS|S)zTry to infer a better signature type for TypedDict.get. This is used to get better type context for the second argument that depends on a TypedDict value type. r required_keys arg_typesret_type)default_signature isinstancetyperlenargsr r variablesvaluer#itemsgetrr copy_modifiedrTr!r)r signaturekey value_typer default_argtvr7r7r8rNs8  rNrrcCsbt|jtrt|jdkrt|jddkrt|jdd|jdd}|dur,|jSg}|D]l}t|jj |}|durD|jSt|jdkrQ| |q0t|jdkrt|jddkrt|jddkr|jdd}t|t rt|j dkrt|tr| |j tdq0| || |jddq0t|jdkr| tt|S|jS)zJInfer a precise return type for TypedDict.get with literal first argument.rrNrr)rrrrrrrdefault_return_typer#rrappendrrrTrr)rkeys output_typesrrrr7r7r8rZs:   2  rZcCs|j}|jdg}t|jtrmt|jdkrmt|jddkrmt|jddtrmt|j dkrmt|j dkrmt|jddkrm|jddj }|jj |}|rm|j d}t|ts^Jt||g}|j||g|dS|j||j dgdS)zTry to infer a better signature type for TypedDict.pop. This is used to get better type context for the second argument that depends on a TypedDict value type. builtins.strrrrrr)rapinamed_generic_typerrrrrr rrrrrr!rr)rrstr_typerrrtypr7r7r8rPs&   rPcCsPt|jtrt|jdkrt|jddkrt|jdd|jdd}|dur:|jjt j |j t j dttjSg}|D]3}||jjvrQ|jj|j||j |jj|}|r`||q>|jj|j||j ttjSt|jddkrt|St|jdkrt|jddkrt|jddkrtg||jddS|jS)z=Type check and infer a precise return type for TypedDict.pop.rrNcoder)rrrrrrrrfailr$TYPEDDICT_KEY_MUST_BE_STRING_LITERALcontextcodes LITERAL_REQrr from_errorrmsgtypeddict_key_cannot_be_deletedrrrtypeddict_key_not_foundrr)rr value_typesrrr7r7r8r`,s4    2r`cCs|j}|jdg}t|jtrSt|jdkrSt|jddkrSt|jddtrSt|j dkrSt|jddkrS|jddj }|jj |}|rS|j ||gdS|j ||j dgdS)zTry to infer a better signature type for TypedDict.setdefault. This is used to get better type context for the second argument that depends on a TypedDict value type. rrrrr)rrrrrrrrr rrrrr)rrrrrr7r7r8rOOs rOcCsHt|jtrt|jdkrt|jddkrt|jddkrt|jdd|jdd}|durC|jjt j |j t j dttjS|jjt|@}|rW|jjj||j d|jdd}g}|D]:}|jj|}|dur|jj|j||j ttjSt||s|jj|||j ttjS||qbt|S|jS)z@Type check TypedDict.setdefault and infer a precise return type.rrrNr)r)rrrrrrrrrrrrrrrr r readonly_keysrTrreadonly_keys_mutatedrrrr+typeddict_setdefault_arguments_inconsistentrrr)rrassigned_readonly_keys default_typerrrr7r7r8r_fs>    r_cCst|jtrht|jdkrht|jddkrht|jdd|jdd}|dur:|jjt j |j t j dttjS|D]+}||jjvsJ||jjvrV|jj|j||j q<||jjvrg|jj|j||j q<|jS)z!Type check TypedDict.__delitem__.rrNr)rrrrrrrrrrrrrrrr rrrrrrrr)rrrr7r7r8ras&   razupdate of TypedDictz__ior__ of TypedDictr_TP_DICT_MUTATING_METHODScCsx|j}t|jtrt|jdkrt|jd}t|ts|S|}|jt d}|j r|j dr|j t vr9d|_ |jjjddddt|jj|j dd|d}Wd n1s^wY|j rid |_ g}t|trt|g}nt|trd d t|D}g}|D]$}|j|j|jB|j@d}|jjjs|jt|jd }||q|rt|}|j|gdS|S)zTry to infer a better signature type for methods that update `TypedDict`. This includes: `TypedDict.update`, `TypedDict.__or__`, `TypedDict.__ror__`, and `TypedDict.__ior__`. rrrTcSs |jtjkS)N)rrTYPEDDICT_READONLY_MUTATED)nameinfor7r7r8s z6typed_dict_update_signature_callback..) filter_errorssave_filtered_errors) type_contextNFcSsg|] }t|tr|qSr7)rr)rItr7r7r8 s z8typed_dict_update_signature_callback..) item_namesr)rrrrrrr# as_anonymousrrTrrr to_be_mutatedrrrget_expression_typer"r$relevant_itemsrrroptions extra_checkslistrr)rrarg_typeinferred possible_tdsrtditemr7r7r8rVsR        rVcCst|jdkrTt|jddkrTt|jddkrT|jdd}t|tr)|j}nt|tr?|jdkr?t|jtr?|jj }n|j S|dkrM|j dgS|j dgS|j S)z1Infer a more precise return type for int.__pow__.rrr-z builtins.intzbuiltins.float) rrrrrrr opexprrrr)rargexponentr7r7r8r[s2   r[ multiplierintcCst|jtrA|jjdurA|jjj}|jjj}t|tr>t|jj dr+t |||dS|jj t ||||jj |jj ddS|jSt|jt r\|jj}|jj}t|tr\t |||dS|jS)zInfer a more precise return type for int.__neg__ and int.__pos__. This is mainly used to infer the return type as LiteralType if the original underlying object is a LiteralType object. Nr)rfallback)rrlinecolumn)last_known_value)rrrrrrrrrrrrrrr)rrrrr7r7r8r\s*     r\cCs t|dS)z{Infer a more precise return type for int.__pos__. This is identical to __neg__, except the value is not inverted. r)r\)rr7r7r8r]s r]cCst|jts |jSt|jdd}t|tr3|jdur3|jj}t|t r0|jj |jj |dS|jSt|jt rL|j}t|t rL|jj |jj |dS|jS)zInfer a more precise return type for tuple.__mul__ and tuple.__rmul__. This is used to return a specific sized tuple if multiplied by Literal int rN)r) rrrrr#rrrrrrrr)rrrr7r7r8r^s    r^)rrr(r)rrr(r)r)rrrrr(r)@ __future__rr2rtypingrrmypy.errorcodes errorcodesrr6r mypy.nodesrrr r mypy.pluginr r r rrrrmypy.plugins.commonr mypy.subtypesr mypy.typeopsrr mypy.typesrrrrrrrrrrr r!r"r#r$r%rNrZrPr`rOr_ra frozensetr__annotations__rVr[r\r]r^r7r7r7r8s4   $ D  ( '  #  /  3