U eY @sddlZddlmZddlmZmZmZmZmZm Z m Z m Z m Z ddlmZddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm 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-ddl.m/Z/ddl0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7ddl8m9Z9dd l:m;Z;dd lm?Z?m@Z@mAZAmBZBmCZCmDZDm Z mEZEmFZFmGZGmHZHmIZIdd lJmKZKdd lLmMZMddlNmOZPddlQmRZRzddl>mSZSWn"eTk rddl>mGZSYnXdZUdZVeWddZXeXdZYeXdZZeXdZ[eXdZ\eXdZ]e^e e_dfdddZ`e`ePZaeadkr,dnd Zbd!ZOe^d"dd#d$ZcGd%d&d&e5ZdGd'd(d(Zee4e d)d*d+ZfGd,d-d-ZgGd.d/d/ZhGd0d1d1Zied2d3d4Zjed5d6d4Zked7d8d4Zled9d:d4Zmed;dd?d@Zpe^e7eddAdBdCZqe7eddDdEdFZre^e e1e7feddGdHdIZse7eddDdJdKZte1eddDdLdMZudZe2e^eee ee eeSevevddO dPdQZwe ee(fe^dRdSdTZxe ee(fe^dRdUdVZye^eee^efdWdXdYZzdS)[N) ConfigParser) AnyCallableDictListOptionalSetTupleTypeUnion) ErrorCode) ARG_NAMED ARG_NAMED_OPTARG_OPTARG_POS ARG_STAR2MDEFArgumentAssignmentStmtBlockCallExprClassDefContext Decorator EllipsisExprFuncBaseFuncDefJsonDict MemberExprNameExprPassStmtPlaceholderNodeRefExprStrExpr SymbolNodeSymbolTableNodeTempNodeTypeInfo TypeVarExprVar)Options)CheckerPluginInterfaceClassDefContextFunctionContext MethodContextPluginReportConfigContextSemanticAnalyzerPluginInterface) dataclasses)set_callable_name)make_wildcard_trigger) AnyType CallableTypeInstanceNoneType Overloaded ProperTyper TypeOfAnyTypeType TypeVarType UnionTypeget_proper_type) fill_typevars)get_unique_redefinition_name) __version__)is_valid_field) TypeVarDef)r= pydantic-mypyzpydantic-mypy-metadataz.main.BaseModelz.env_settings.BaseSettingsz.main.ModelMetaclassz .fields.Fieldz.dataclasses.dataclass.)versionreturncCsttt|dddS)N+r.)tuplemapint partitionsplitrGrQ4/tmp/pip-unpacked-wheel-6q_yhcy8/pydantic/v1/mypy.pyparse_mypy_versionVsrS)ribuiltins __builtins__zTypingType[Plugin]cCstS)z `version` is the mypy version string We might want to use this to print a warning if the mypy version being used is newer, or especially older, than we expect (or need). )PydanticPluginrPrQrQrRpluginasrXcseZdZeddfdd ZeddddZeeee gdfdd d Z ed dd d Z eeee ge fdddZeeee gdfdddZeeeefdddZe ddddZe ddddZeddddZZS)rWNoptionsrHcs&t||_|j|_t|dSN)PydanticPluginConfig plugin_configto_data _plugin_datasuper__init__)selfrZ __class__rQrRrals  zPydanticPlugin.__init__z+Optional[Callable[[ClassDefContext], None]])fullnamerHcCs:||}|r6t|jtr6tdd|jjDr6|jSdS)Ncss|]}t|tkVqdSr[) get_fullnameBASEMODEL_FULLNAME.0baserQrQrR usz5PydanticPlugin.get_base_class_hook..)lookup_fully_qualified isinstancenoder'anymro$_pydantic_model_class_maker_callbackrbresymrQrQrRget_base_class_hookqs  z"PydanticPlugin.get_base_class_hookcCs|tkr|jSdSr[)MODEL_METACLASS_FULLNAME)_pydantic_model_metaclass_marker_callbackrbrerQrQrRget_metaclass_hookysz!PydanticPlugin.get_metaclass_hookz+Optional[Callable[[FunctionContext], Type]]cCs"||}|r|jtkr|jSdSr[)rlreFIELD_FULLNAME_pydantic_field_callbackrrrQrQrRget_function_hook~s z PydanticPlugin.get_function_hookcCs|drtSdS)Nz .from_orm)endswithfrom_orm_callbackrwrQrQrRget_method_hooks zPydanticPlugin.get_method_hookcCs|tkrtdkrtjSdS)zMark pydantic.dataclasses as dataclass. Mypy version 1.1.1 added support for `@dataclass_transform` decorator. )rN)DATACLASS_FULLNAMEMYPY_VERSION_TUPLEr2Zdataclass_class_maker_callbackrwrQrQrRget_class_decorator_hooksz'PydanticPlugin.get_class_decorator_hookctxrHcCs|jS)zjReturn all plugin config data. Used by mypy to determine if cache needs to be discarded. )r_)rbrrQrQrRreport_config_datasz!PydanticPlugin.report_config_datacCst||j}|dSr[)PydanticModelTransformerr] transform)rbrZ transformerrQrQrRrqs z3PydanticPlugin._pydantic_model_class_maker_callbackcCs<|jjr dS|jjj}|s"tdt|jddr8d|j_dS)zReset dataclass_transform_spec attribute of ModelMetaclass. Let the plugin handle it. This behavior can be disabled if 'debug_dataclass_transform' is set to True', for testing purposes. Nz-callback not passed from 'get_metaclass_hook'dataclass_transform_spec) r]debug_dataclass_transformclsinfoZdeclared_metaclassAssertionErrorgetattrtyper)rbrZinfo_metaclassrQrQrRrvs   z8PydanticPlugin._pydantic_model_metaclass_marker_callbackr c s |j|jddkstd|jddks2td|jd}|jd}|r`|r`t|j|jS|r|jdd}|d}t|t s|Sn|r|jdd}t|t rt dkr|j d}n | d}t|t r|j}t|dd }|rtd d |Drtfd d |D|_|SS) ax Extract the type of the `default` argument from the Field function, and use it as the return type. In particular: * Check whether the default and default_factory argument is specified. * Output an error if both are specified. * Retrieve the type of the argument which is specified, and use it as return type for the function. rdefaultz0"default" is no longer first argument in Field()rdefault_factoryz9"default_factory" is no longer second argument in Field())riargsNcss|]}t|tVqdSr[)rmr=)riargrQrQrRrksz:PydanticPlugin._pydantic_field_callback..c3s|] }VqdSr[rQ)ri_Zdefault_any_typerQrRrks)default_return_typeZcallee_arg_namesrr+error_default_and_default_factory_specifiedapicontext arg_typesrmrr9ritemsr6ret_typerallrK) rbrZ default_argsZdefault_factory_argsZ default_typeZ default_argZdefault_factory_typerrrQrrRrzs6         z'PydanticPlugin._pydantic_field_callback)__name__ __module__ __qualname__r*rastrrtrrr,rxr{r.r r~rr0rrrrqrvr-rz __classcell__rQrQrcrRrWks  rWc@s`eZdZUdZeed<eed<eed<eed<eed<eddd d Zee e fd d d Z dS)r\)init_forbid_extra init_typedwarn_required_dynamic_aliaseswarn_untyped_fieldsrrrrrrNrYcCs|jdkrdSt|j}|dk rr|didi}|jD]4}||d}t|tsbtd|t|||q:n:t}| |j|jD] }|j t |dd}t|||qdS)NZtoolrEFz/Configuration value must be a boolean for key: )fallback) config_file parse_tomlget __slots__rmbool ValueErrorsetattrrread getbooleanCONFIGFILE_KEY)rbrZZ toml_configconfigkeyZsettingr]rQrQrRras       zPydanticPluginConfig.__init__rHcsfddjDS)Ncsi|]}|t|qSrQ)r)rirrbrQrR sz0PydanticPluginConfig.to_data..)rrrQrrRr^szPydanticPluginConfig.to_data) rrrrr__annotations__r*rarrrr^rQrQrQrRr\s r\rcCs|j}t|tr|j}t|tr4t|jtr4|j}n.NFT)rcSsi|]}|j|qSrQ)name serializerifieldrQrQrRrAsz6PydanticModelTransformer.transform..)fieldsr)rrradjust_validator_signaturescollect_configcollect_fieldsrorpadd_initializeradd_construct_method set_frozenrrset_values_dictrr)rbrrrr is_settingsrQrQrRr,s    z"PydanticModelTransformer.transformcCsh|jjjjD]R\}}t|jtr|jjd}t|t rt|j t r|j j t dkrd|jj_qdS)a[When we decorate a function `f` with `pydantic.validator(...), mypy sees `f` as a regular method taking a `self` instance, even though pydantic internally wraps `f` with `classmethod` if necessary. Teach mypy this by marking any function whose outermost decorator is a `validator()` call as a classmethod. rz.class_validators.validatorTN)rrrnamesrrmrnrZoriginal_decoratorsrcalleerre _NAMESPACEfuncis_class)rbrrsZ first_decrQrQrRrEs   z4PydanticModelTransformer.adjust_validator_signaturesModelConfigDatac Cs|j}|j}t}|jjD]d}t|ts*q|jdkr|jjD] }t|tsLq<| | |q<|j r|j s|j jrt|j|q|jjddD]L}t|jkrq|jtt||jtdD]\}}|||qq|S)zz Collects the values of the config attributes that are used by the plugin, accounting for parent classes. ZConfigrNr)rrrdefsbodyrmrrrupdateget_config_updatehas_alias_generatorrr]rerror_required_dynamic_aliasesrrrprradd_plugin_dependencyr4rfr setdefault) rbrrrstmtsubstmtrrvaluerQrQrRrWs2      z'PydanticModelTransformer.collect_configPydanticModelField) model_configrHc s|j}|jj}g}t}|jjD]}t|ts0q |jd}t|tr t |j sPq |j sj|j j rjt|j||jj|j }|dkrq |j} t| trq t| tsq | jrq ||||} ||\} } | r|js|j jrt|j||t|j | | | |j|jd| |j q |!} |jj"ddD]}t#|j$kr>q*g}|j%t&t'||j$t#d(D]^\}|krt)||}| ||n(fdd| D\}| *|||qh|| } q*| S)zR Collects the fields for the model, accounting for parent classes rNr is_requiredaliashas_dynamic_aliaslinecolumnrrc3s|]}|jkr|VqdSr[r)riarrQrRrks z:PydanticModelTransformer.collect_fields..)+rrsetrrrmrlvaluesrrCrZ new_syntaxr]rerror_untyped_fieldsrrrrrnr!r)Z is_classvarget_is_requiredget_alias_inforrrappendrrraddcopyrprrrr4rfr deserializeremove)rbrrrrZ known_fieldsrlhsrsrnrrrZ all_fieldsrZsuperclass_fieldsdatarrQrrRrvst               z'PydanticModelTransformer.collect_fields)rrrrHc Cs|j}|jj}|jdk }|p,t|jo*|j }|j||||d}|||sntd} | t | t t j dtd|jjjkrt|d|tdS)z Adds a fields-aware `__init__` method to the class. The added `__init__` will be annotated with types vs. all `Any` depending on the plugin settings. Ttypedforce_all_optional use_aliaskwargsNra)rr]rrrrget_field_argumentsshould_init_forbid_extrar)rrr5r;explicitrrrr add_methodr8) rbrrrrrrrZinit_argumentsvarrQrQrRrs"   z(PydanticModelTransformer.add_initializer)rrHc Cs,|j}|jtd|jtdg}t|tg}ttd||dt}|j |dddd}|g|}|jtd}d }|j j d |} t d krt || d g|ttj} t|| g|ttj} nt|| d g|} t|| g|} tt| |j jj|<t| t r | } nt | } t|d || | | dddS)z Adds a fully typed `construct` classmethod to the class. Similar to the fields-aware __init__ method, but always uses the field names (not aliases), and does not treat settings fields as optional. z.setz.strZ _fields_setNTFrz.objectZ_PydanticBaseModelrJ)rrZ construct) return_type self_typetvar_defis_classmethod)rr named_type BUILTINS_NAMEr>r8rr)rrrrerr=r5r;Zfrom_omitted_genericsr(rDr%rrrrmr) rbrrZset_strZoptional_set_strZfields_set_argumentZconstruct_argumentsZobj_typeZself_tvar_nameZ tvar_fullnameZtvdZself_tvar_exprr rQrQrRrsP$  z-PydanticModelTransformer.add_construct_method)rrrHc Cs|j}|jj}|D]}|j|j}|dk r|j}t|trD||_ qt|t rb|j j sb|j qz t|}Wntk rt|}YnXd|d|jd} t| |j |jq|j|dd}||_||_ t|dt||_tt||jt|<qdS)z Marks all fields as properties so that attempts to set them trigger mypy errors. This is the same approach used by the attrs and dataclasses plugins. Nzsym_node.node: rrF)rrJ)rrrrrrrnrmr)Z is_propertyr!rZfinal_iterationdeferr TypeErrorreprrdrto_varrfr _fullnamer%r) rbrrrrrZsym_noderZvar_strrrQrQrRrs*   z#PydanticModelTransformer.set_frozen)rrHcCs|jd}t|tr |j|jks$dS|jdkrt|jtrH|jjdk}n0t|jtrb|jjdk}nt |j|j j |dSt |dS|jdkrd}t|jtr|jj dkrd }t |d St|jtr|jj d krt f|j|jj d kiSt |j|j j |dS) z Determines the config update due to a single statement in the Config class definition. Warns if a tracked config attribute is set to a value the plugin doesn't know how to interpret (e.g., an int) rNrZforbid) forbid_extrarTz builtins.NoneF)r) builtins.Truezbuiltins.Falser)rrmrrrrvaluer#rrerror_invalid_config_valuerrrre)rbrrrrrQrQrRr7s(       z*PydanticModelTransformer.get_config_update)rrrrHcCs|j}t|tr.t|j|jj}t| St|t rt|j t r|j j t krt|j|jD]4\}}|dksv|dkr|jtkS|dkr^dSq^t|j|jj}t| St|tS)zg Returns a boolean indicating whether the field defined in `stmt` is a required field. NrrF)rrmr&r?rrrrtype_has_implicit_defaultrrr"reryzipr arg_namesrdr)rrrexprZ value_typerrrQrQrRrSs  " z(PydanticModelTransformer.get_is_required)type_rHcCs4t|trdSt|tr0tdd|jDr0dSdS)z Returns True if the passed type will be given an implicit default value. In pydantic v1, this is the case for Optional types and Any (with default value None). Tcss"|]}t|tpt|tVqdSr[)rmr8r5)rirrQrQrRrkwszEPydanticModelTransformer.type_has_implicit_default..F)rmr5r>ror)rrQrQrRrms z2PydanticModelTransformer.type_has_implicit_default)rrHcCs|j}t|trdSt|tr6t|jtr6|jjtks:dSt|j D]:\}}|dkrVqD|j |}t|t rx|j dfSdSqDdS)a  Returns a pair (alias, has_dynamic_alias), extracted from the declaration of the field defined in `stmt`. `has_dynamic_alias` is True if and only if an alias is provided, but not as a string literal. If `has_dynamic_alias` is True, `alias` will be None. )NFrF)NT) rrmr&rrr"rery enumeraterrr#r)rriZarg_namerrQrQrRr~s$     z'PydanticModelTransformer.get_alias_info)rrrrrHcs&|jjjfdd|D}|S)z Helper function used during the construction of the `__init__` and `construct` method signatures. Returns a list of mypy Argument instances for use in the generated signatures. cs(g|] }r|js|jdqS))rforce_optionalr)r to_argumentrrrrrrQrR sz@PydanticModelTransformer.get_field_arguments..)rrr)rbrrrr argumentsrQr!rRrs  z,PydanticModelTransformer.get_field_arguments)rrrHcCs.|js||t|jrdS|jr&dS|jjS)aH Indicates whether the generated `__init__` should get a `**kwargs` at the end of its signature We disallow arbitrary kwargs if the extra config setting is "forbid", or if the plugin config says to, *unless* a required dynamic alias is present (since then we can't determine a valid signature). FT)ris_dynamic_alias_presentrrrr]r)rbrrrQrQrRrs z1PydanticModelTransformer.should_init_forbid_extra)rrrHcCs8|D]}|jrdSq|r4|D]}|jdkrdSqdS)z Returns whether any fields on the model have a "dynamic alias", i.e., an alias that cannot be determined during static analysis. TNF)rr)rrrrQrQrRr$s z1PydanticModelTransformer.is_dynamic_alias_present)"rrrrrrrr,r\rarrrrrrrrrrrr staticmethodrrrr:rr rrrrr$rQrQrQrRrs>  T6! rc@steZdZeeeeeeedddZeee dddZ eeeee ddd Z e d d d Zeee dd ddZdS)rrcCs(||_||_||_||_||_||_dSr[r)rbrrrrrrrQrQrRras zPydanticModelField.__init__)rrrHcCs,|j}|r|jdk r|j}t|||jjSr[)rrr)r)rbrrrrQrQrRrszPydanticModelField.to_var)rrrrrHcCsR|r"||jjdk r"||jj}n ttj}t||||d|sF|jsJtnt dS)N)variabletype_annotationZ initializerkind) rrr5r;rrrrrr )rbrrrrr'rQrQrRr s  zPydanticModelField.to_argumentrcCs|jSr[)__dict__rrQrQrRrszPydanticModelField.serialize)rrrHcCs |f|Sr[rQ)rrrrQrQrRrszPydanticModelField.deserializeN)rrrrrrrMrar'r)rrr rr classmethodrrQrQrQrRrs  rc@szeZdZdeeeeeeeeeeeedddZeeefdddZ edddd d Z eedd d d Z dS)rNrrrrrrcCs(||_||_||_||_||_||_dSr[r+)rbrrrrrrrQrQrRras zModelConfigData.__init__rcCsdd|jDS)NcSsi|]\}}|dk r||qSr[rQ)rikvrQrQrRrsz3ModelConfigData.set_values_dict..)r)rrrQrQrRrszModelConfigData.set_values_dict)rrHcCs2|dkr dS|D]\}}t|||qdSr[)rrr)rbrr,r-rQrQrRrszModelConfigData.update)rrrHcCst||dkrt|||dSr[)rr)rbrrrQrQrRrszModelConfigData.setdefault)NNNNNN) rrrrrrarrrrrrrQrQrQrRrs" rz pydantic-ormzInvalid from_orm callZPydanticzpydantic-configzInvalid config valuezpydantic-aliaszDynamic alias disallowedzpydantic-unexpectedzUnexpected behaviorzpydantic-fieldzUntyped field disallowedzInvalid Field defaults) model_namerrrHcCs|jd|d|tddS)N"z" does not have orm_mode=Truecode)fail ERROR_ORM)r.rrrQrQrRrsr)rrrrHcCs|jd|d|tddS)NzInvalid value for "Config.r/r0)r2 ERROR_CONFIG)rrrrQrQrRrsr)rrrHcCs|jd|tddS)Nz#Required dynamic aliases disallowedr0)r2 ERROR_ALIASrrrQrQrRrsr)rrrrHcCs4d}d|d}|d|d7}|j||tddS)Nz6https://github.com/pydantic/pydantic/issues/new/choosez7The pydantic mypy plugin ran into unexpected behavior:  z&Please consider reporting this bug at z so we can try to fix it!r0)r2ERROR_UNEXPECTED)rrrlinkZ full_messagerQrQrRr!s rcCs|jd|tddS)NzUntyped fields disallowedr0)r2 ERROR_UNTYPEDr6rQrQrRr+srcCs|jd|tddS)Nz>Field default and default_factory cannot be specified togetherr0)r2ERROR_FIELD_DEFAULTSr6rQrQrRr/srF) rrrr r r r is_newrHcCs|jj}||jkr@|j|} | jr@t| jtr@|jjj | j|pJt |}|sT|rpt t dt |dtg} n |pzt |}t t d|dtg} | |}ggg} } } |D]:}|jstd| |j| t|j| |jq|jtd}t| | | ||}|r|g|_t||ttg}||_t|||_||_t |d||_!|j"|_"||jkrt#||j}|j||j|<|rd|_$t ||j}||_|j!|_!d|_%t&|t'dg|}|j"|_"t(t)|} n t(t)|} d| _| |j|<|j*jj|dS) z} Adds a new method to a class. This can be dropped if/when https://github.com/python/mypy/issues/7301 is merged _clsNZ__pydantic_self__z"All arguments must be fully typed.z .functionrJTr*)+rrrZplugin_generatedrmrnrrrrr@rr)r<Zmake_normalizedrr'rrrr&r(rr rr6 variablesrr r3rrrfrrrAZ is_decoratedr rrr%rZdefn)rrrr r r r r<rrsfirstrrZ arg_kindsrZ function_type signaturerZr_namer-decrQrQrRr3sV            r)xrHcCs|j}t|r|S|Szc Used for compatibility with mypy 0.740; can be dropped once support for 0.740 is dropped. )recallablerBfnrQrQrRrfsrfcCs|j}t|r|S|SrC)rrDrErQrQrRrsr)rrHc Cs|dsdSd}tjdkr&ddl}n^z ddl}WnPtk rd}z ddl}Wn*tk r|ddl}|dYYdSXYnXt ||}| |W5QRSQRXdS)Nz.tomlrb) rrzJNo TOML parser installed, cannot read configuration from `pyproject.toml`.) r|sys version_infoZtomllibZtomli ImportErrortomlwarningswarnopenload)rZ read_modeZtoml_rOrfrQrQrRrs"       r)NNFF){rK configparserrtypingrrrrrrr r Z TypingTyper Zmypy.errorcodesr Z mypy.nodesr rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)Z mypy.optionsr*Z mypy.pluginr+r,r-r.r/r0r1Z mypy.pluginsr2Z mypy.semanalr3Zmypy.server.triggerr4Z mypy.typesr5r6r7r8r9r:r;r<r=r>r?Z mypy.typevarsr@Z mypy.utilrAZ mypy.versionrBZ mypy_versionZpydantic.utilsrCrDrMrrrrrgrruryrrrMrSrrrXrWr\r}rrrr3r4r5r8r:r;rrrrrrrrrfrrrQrQrQrRs , | $   8           u%.%          S