o 7fP@sfddlmZddlmZddlmZddlmZddl m Z ddl m Z m Z mZmZddlmZmZmZmZmZmZmZddlmZdd lmZdd lmZmZdd lm 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 eZ/d*ddZ0d+ddZ1d*ddZ2d+ddZ3d*ddZ4d+ddZ5d+d d!Z6d*d"d#Z7d+d$d%Z8d+d&d'Z9d+d(d)Z:dS),) annotations)partial)CallableN)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] | NonecCs0ddlm}m}|dkr|jS|dkr|jSdS)Nrctypessingledispatchz _ctypes.Arrayzfunctools.singledispatch) mypy.pluginsr*r+array_constructor_callback'create_singledispatch_function_callbackselfr%r*r+r1M/home/ertert/spirit/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,r6r7evolve_function_sig_callbackfields_function_sig_callbackreplace_function_sig_callbackr0r%r6r7r1r1r2get_function_signature_hook5sz)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)typing.Mapping.getcSh|]}|dqSz .setdefaultr1.0nr1r1r2 Iz:DefaultPlugin.get_method_signature_hook..cSr?z.popr1rAr1r1r2rDKrEz_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)r0r%r*r+typed_dict_updatesrCr1r1r2get_method_signature_hookBs( z'DefaultPlugin.get_method_signature_hook&Callable[[MethodContext], Type] | NonecCsddlm}m}|dkrtS|dkrtS|dkrtS|dvr tS|ddtDvr+tS|d dtDvr6t S|d dtDvrAt S|d krH|j S|d krO|j S||j krW|jS||jkr_|jSdS) Nrr)r>zbuiltins.int.__pow__zbuiltins.int.__neg__)zbuiltins.tuple.__mul__zbuiltins.tuple.__rmul__cSr?r@r1rAr1r1r2rDirEz0DefaultPlugin.get_method_hook..cSr?rFr1rAr1r1r2rDkrEcSr?)z .__delitem__r1rAr1r1r2rDmrEz_ctypes.Array.__getitem__z_ctypes.Array.__iter__)r,r*r+typed_dict_get_callbackint_pow_callbackint_neg_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_argumentr/r1r1r2get_method_hook^s0  zDefaultPlugin.get_method_hook)Callable[[AttributeContext], Type] | NonecCsPddlm}m}|dkr|jS|dkr|jS||jvr|jS||jvr&|jSdS)Nr)r*enumsz_ctypes.Array.valuez_ctypes.Array.raw) r,r*rbarray_value_callbackarray_raw_callbackENUM_NAME_ACCESSenum_name_callbackENUM_VALUE_ACCESSenum_value_callback)r0r%r*rbr1r1r2get_attribute_hookys  z DefaultPlugin.get_attribute_hook(Callable[[ClassDefContext], None] | NonecCsRddlm}m}||jvr|jS||jvs$||jvs$||jvs$||jvr'|j SdS)Nrr5) r,r6r7dataclass_makersdataclass_tag_callbackattr_class_makersattr_dataclass_makersattr_frozen_makersattr_define_makersattr_tag_callbackr;r1r1r2get_class_decorator_hooks     z&DefaultPlugin.get_class_decorator_hook(Callable[[ClassDefContext], bool] | NonecCsddlm}m}m}||jvr|jS||jvr|jS||jvr"|j S||j vr.t |j ddS||j vr;t |j dddS||j vrHt |j dddSdS)Nr)r6r7 functoolsT)auto_attribs_default)rufrozen_default)ru slots_default)r,r6r7rtrkdataclass_class_maker_callbackfunctools_total_ordering_makers'functools_total_ordering_maker_callbackrmattr_class_maker_callbackrnrrorp)r0r%r6r7rtr1r1r2get_class_decorator_hook_2s$      z(DefaultPlugin.get_class_decorator_hook_2N)r%r&r'r()r%r&r'r4)r%r&r'r=)r%r&r'rR)r%r&r'ra)r%r&r'rj)r%r&r'rs) __name__ __module__ __qualname____doc__r3r<rQr`rirrr|r1r1r1r2r$(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 isinstancetyperlenargsrr variablesvaluer"itemsgetrr copy_modifiedrMr r)r signaturekey value_typer default_argtvr1r1r2rGs8  rGrrcCsbt|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"rrappendrrrMrr)rkeys output_typesrrrr1r1r2rSs:   2  rScCs|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_typerrrrrrrrrrrr rr)rrstr_typerrrtypr1r1r2rIs&   rIcCsPt|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_typesrrr1r1r2rXs4    2rXcCs|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)rrrrrrrrrrrrrr)rrrrrr1r1r2rHBs rHcCs t|jtrt|jdkrt|jddkrt|jddkrt|jdd|jdd}|durC|jjt j |j t j dttjS|jdd}g}|D]:}|jj|}|durm|jj|j||j ttjSt||s|jj|||j ttjS||qNt|S|jS)z@Type check TypedDict.setdefault and infer a precise return type.rrrNr)rrrrrrrrrrrrrrrrrrrrrr+typeddict_setdefault_arguments_inconsistentrrr)rr default_typerrrr1r1r2rWYs8    rWcCst|jtrbt|jdkrbt|jddkrbt|jdd|jdd}|dur:|jjt j |j t j dttjS|D]%}||jjvrP|jj|j||j q<||jjvra|jj|j||j q<|jS)z!Type check TypedDict.__delitem__.rrNr)rrrrrrrrrrrrrrrrrrrrrrr)rrrr1r1r2rYs&    rYcCsR|j}t|jtrt|jdkrt|jd}t|ts|S|}|jt d}|j r|j dr|j j t|j j|j dd|d}Wdn1sQwYg}t|tra|g}nt|trqddt|D}g}|D]$}|j|j|jB|j@d}|j jjs|jt|jd}||qu|rt|}|j|gd S|S) zTry to infer a better signature type for methods that update `TypedDict`. This includes: `TypedDict.update`, `TypedDict.__or__`, `TypedDict.__ror__`, and `TypedDict.__ior__`. rrr) type_contextNcSsg|] }t|tr|qSr1)rr)rBtr1r1r2 s z8typed_dict_update_signature_callback..) item_namesr)rrrrrrr" as_anonymousrrMrrr filter_errorsget_expression_typer!r#relevant_itemsrrroptions extra_checkslistrr)rrarg_typeinferred possible_tdsrtditemr1r1r2rOsD       rOcCst|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)rargexponentr1r1r2rTs2   rTcCst|jtr@|jjdur@|jjj}|jjj}t|tr=t|jj dr*t | |dS|jj t | |j|jj |jj ddS|jSt|jt rZ|jj}|jj}t|trZt | |dS|jS)zInfer a more precise return type for int.__neg__. This is mainly used to infer the return type as LiteralType if the original underlying object is a LiteralType object N)rfallback)rrlinecolumn)last_known_value)rrrrrrintrrrrrrrr)rrrr1r1r2rUs$     rUcCst|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)rrrr1r1r2rVs    rV)rrr'r)rrr'r); __future__rrtrtypingrmypy.errorcodes errorcodesrmypyr mypy.nodesrrrr mypy.pluginr r r r rrrmypy.plugins.commonr mypy.subtypesr mypy.typeopsrr mypy.typesrrrrrrrrrrrr r!r"r#r$rGrSrIrXrHrWrYrOrTrUrVr1r1r1r2s0    $ D ( '  #  +  *