o %if@s^dZddlmZddlZddlmZddlmZmZm Z ddl m Z ddl m Z mZddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm 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.dd l/m0Z0dd l1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8dd l9m:Z:dd l;mZ>ddl?m@Z@ddlAmBZBddlCmDZDddlEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPddlQmRZRddlSmTZTddlUmVZWddlXmYZYddlZm[Z[zddlEm\Z\Wne]yddlEmNZ\YnwdZ^dZ_dZ`dZadZbdZcdZdd Zed!Zfhd"Zge[eWZhehd#krd$nd%Zid&ZVdvd+d,ZjGd-d.d.e6ZkGd/d0d0Zldwd4d5ZmGd6d7d7ZnGd8d9d9ZoGd:d;d;ZpGdd?d@Zre dAdBd@Zse dCdDd@Zte dEdFd@Zue dGdHd@Zve dGdId@Zwe dGdJd@ZxdxdQdRZydydUdVZzdzdWdXZ{d{d[d\Z|dzd]d^Z}d|d_d`Z~d|dadbZ   cd}d~dpdqZddtduZdS)z^This module includes classes and functions designed specifically for use with the mypy plugin.) annotationsN) ConfigParser)AnyCallableIterator) ErrorCode) expand_typeexpand_type_by_instance) ARG_NAMED ARG_NAMED_OPTARG_OPTARG_POS ARG_STAR2MDEFArgumentAssignmentStmtBlockCallExprClassDefContext DecoratorDictExpr EllipsisExpr ExpressionFuncDefIfStmtJsonDict MemberExprNameExprPassStmtPlaceholderNodeRefExpr StatementStrExprSymbolTableNodeTempNode TypeAliasTypeInfoVar)Options)CheckerPluginInterfaceClassDefContextFunctionContext MethodContextPluginReportConfigContextSemanticAnalyzerPluginInterface) dataclasses)deserialize_and_fixup_type)set_callable_name)make_wildcard_trigger)state)map_type_from_supertype) AnyType CallableTypeInstanceNoneType OverloadedType TypeOfAnyTypeType TypeVarType UnionTypeget_proper_type) fill_typevars)get_unique_redefinition_name) __version__)_fields)parse_mypy_version) TypeVarDef)r? pydantic-mypyzpydantic-mypy-metadatazpydantic.main.BaseModelz#pydantic_settings.main.BaseSettingszpydantic.root_model.RootModelz5pydantic._internal._model_construction.ModelMetaclasszpydantic.fields.Fieldzpydantic.dataclasses.dataclass.pydantic.functional_validators.model_validator>*pydantic.functional_serializers.serializer.pydantic.deprecated.class_validators.validator.pydantic.functional_validators.field_validator0pydantic.functional_serializers.model_serializer3pydantic.deprecated.class_validators.root_validatorrI)ribuiltins __builtins__versionstrreturn type[Plugin]cCstS)a!`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). Args: version: The mypy version string. Return: The Pydantic mypy plugin type. )PydanticPlugin)rRrWF/var/www/html/corbot_env/lib/python3.10/site-packages/pydantic/mypy.pypluginrs rYcs|eZdZdZd'fdd Zd(d d Zd)ddZd*ddZd+ddZd)ddZ d,ddZ d-dd Z d.d!d"Z d/d%d&Z ZS)0rVzThe Pydantic mypy plugin.optionsr)rTNonecs&t||_|j|_t|dSN)PydanticPluginConfig plugin_configto_data _plugin_datasuper__init__)selfrZ __class__rWrXrbs  zPydanticPlugin.__init__fullnamerS(Callable[[ClassDefContext], bool] | NonecCs:||}|rt|jtrtdd|jjDr|jSdS)zUpdate Pydantic model class.cs|]}|jtkVqdSr\)rfBASEMODEL_FULLNAME.0baserWrWrX z5PydanticPlugin.get_base_class_hook..N)lookup_fully_qualified isinstancenoder'anymro$_pydantic_model_class_maker_callbackrcrfsymrWrWrXget_base_class_hooks z"PydanticPlugin.get_base_class_hook(Callable[[ClassDefContext], None] | NonecCs|tkr|jSdS)z,Update Pydantic `ModelMetaclass` definition.N)MODEL_METACLASS_FULLNAME)_pydantic_model_metaclass_marker_callbackrcrfrWrWrXget_metaclass_hooksz!PydanticPlugin.get_metaclass_hook(Callable[[FunctionContext], Type] | NonecCs"||}|r|jtkr|jSdS)z/Adjust the return type of the `Field` function.N)rorfFIELD_FULLNAME_pydantic_field_callbackrurWrWrXget_function_hooks z PydanticPlugin.get_function_hook&Callable[[MethodContext], Type] | NonecCs|drtSdS)z-Adjust return type of `from_orm` method call.z .from_ormN)endswithfrom_attributes_callbackr{rWrWrXget_method_hooks zPydanticPlugin.get_method_hookcCs|tkr tdkr tjSdS)zMark pydantic.dataclasses as dataclass. Mypy version 1.1.1 added support for `@dataclass_transform` decorator. )rN)DATACLASS_FULLNAMEMYPY_VERSION_TUPLEr1dataclass_class_maker_callbackr{rWrWrXget_class_decorator_hooksz'PydanticPlugin.get_class_decorator_hookctxr/dict[str, Any]cCs|jS)zjReturn all plugin config data. Used by mypy to determine if cache needs to be discarded. )r`)rcrrWrWrXreport_config_datasz!PydanticPlugin.report_config_datar+boolcCst|j|j|j|j}|Sr\)PydanticModelTransformerclsreasonapir^ transform)rcr transformerrWrWrXrtsz3PydanticPlugin._pydantic_model_class_maker_callbackcCs@|jjrdS|jjj}|sJdt|jddrd|j_dSdS)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_transformrinfodeclared_metaclassgetattrtyper)rcrinfo_metaclassrWrWrXrzs   z8PydanticPlugin._pydantic_model_metaclass_marker_callbackr,r<c s|j|jddksJd|jddksJd|jd}|jd}|r0|r0t|j|jS|rF|jdd}|d}t|tsD|SS|r|jdd}t|t rY|j d}t|t r|j }t |dd}|r~td d |Dr~tfd d |D|_|SS) aoExtract 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()argsNcss|]}t|tVqdSr\)rpr?)rkargrWrWrXrmrnz:PydanticPlugin._pydantic_field_callback..c3s|]}VqdSr\rW)rk_default_any_typerWrXrms)default_return_typecallee_arg_namesr+error_default_and_default_factory_specifiedrcontext arg_typesrprr;itemsr8ret_typeralltuple) rcr default_argsdefault_factory_args default_type default_argdefault_factory_typerrrWrrXrs4       z'PydanticPlugin._pydantic_field_callbackrZr)rTr[)rfrSrTrg)rfrSrTrx)rfrSrTr})rfrSrTr)rr/rTr)rr+rTr)rr+rTr[)rr,rTr<)__name__ __module__ __qualname____doc__rbrwr|rrrrrtrzr __classcell__rWrWrdrXrVs       rVc@sJeZdZUdZdZded<ded<ded<ded<dd d ZdddZdS)r]aA Pydantic mypy plugin config holder. Attributes: init_forbid_extra: Whether to add a `**kwargs` at the end of the generated `__init__` signature. init_typed: Whether to annotate fields in the generated `__init__`. warn_required_dynamic_aliases: Whether to raise required dynamic aliases error. debug_dataclass_transform: Whether to not reset `dataclass_transform_spec` attribute of `ModelMetaclass` for testing purposes. )init_forbid_extra init_typedwarn_required_dynamic_aliasesrrrrrrrZr)rTr[cCs|jdurdSt|j}|dur:|didi}|jD]}||d}t|ts1td|t|||qdSt}| |j|jD]}|j t |dd}t|||qFdS)NtoolrHFz/Configuration value must be a boolean for key: )fallback) config_file parse_tomlget __slots__rpr ValueErrorsetattrrread getbooleanCONFIGFILE_KEY)rcrZ toml_configconfigkeysettingr^rWrWrXrb s"       zPydanticPluginConfig.__init__rcsfddjDS)z/Returns a dict of config names to their values.csi|]}|t|qSrW)r)rkrrcrWrX "z0PydanticPluginConfig.to_data..)rrrWrrXr_ zPydanticPluginConfig.to_dataNrrTr)rrrrr__annotations__rbr_rWrWrWrXr]s   r]rr-r<cCs|j}t|tr |j}t|trt|jtr|j}nt|tr"|}nd|d|jjd}t ||j |j |j S|jj t}|durF|j S|did}|dur]t|jj|j |j |j S)z1Raise an error if from_attributes is not enabled.z ctx.type: (of type )Nrfrom_attributesT)rrpr>itemr8rr9rererror_unexpected_behaviorrrrmetadatar METADATA_KEYerror_from_attributesname)rctx_type model_typedetailpydantic_metadatarrWrWrXr%s"  rc@sZeZdZdZd,ddZd-ddZd.ddZd/dd Zd0d"d#Ze d1d%d&Z d2d)d*Z d+S)3PydanticModelFieldz5Based on mypy.plugins.dataclasses.DataclassAttribute.rrSalias str | Nonehas_dynamic_aliasr has_defaultlineintcolumnr Type | Nonerr'c Cs4||_||_||_||_||_||_||_||_dSr\)rrrrrrrr) rcrrrrrrrrrWrWrXrb?s  zPydanticModelField.__init__ current_infotypedforce_optional use_aliasrr0rTrcCsJ||||}|r|||nttj}t||d|s|jr!tdStdS)zABased on mypy.plugins.dataclasses.DataclassAttribute.to_argument.N)variabletype_annotation initializerkind) to_varrr7r=explicitrrr r )rcrrrrrrrrWrWrX to_argumentSs  zPydanticModelField.to_argumentcCsj|jdur2t|jdddur2t|jjt|j|jjj t |iWdS1s-wY|jS)zABased on mypy.plugins.dataclasses.DataclassAttribute.expand_type.N self_type) rrrr5strict_optional_setrZstrict_optionalrridrB)rcrrrWrWrXres  zPydanticModelField.expand_typer(cCs.|r |jdur |j}n|j}t||||S)zextrafrozenalias_generatorrpopulate_by_namezset[str]tracked_config_fieldsrrrExpression | Statementrr0r^r]rTr[cCs||_||_||_||_dSr\)_cls_reason_apir^)rcrrrr^rWrWrXrbs z!PydanticModelTransformer.__init__rcCs|jj}tdd|jddD}|}|||\}}|dus&|dur(dS|D] }|jdur4dSq*tdd|jddD}||||||||||j ||j |j dud| d d |Dd d |D| d |jt<dS) aConfigures the BaseModel subclass according to the plugin settings. In particular: * determines the model config and fields, * adds a fields-aware signature for the initializer and construct methods * freezes the class if frozen = True * stores the fields, config, and if the class is settings in the mypy metadata for access by subclasses css|]}t|jvVqdSr\)ROOT_MODEL_FULLNAMErfrjrWrWrXrmrnz5PydanticModelTransformer.transform..NFcsrhr\)rfBASESETTINGS_FULLNAMErjrWrWrXrmrnT)rcSi|]}|j|qSrWrrrkfieldrWrWrXrrz6PydanticModelTransformer.transform..cSr rWr)rk class_varrWrWrXrr)fields class_varsr)rrrrrscollect_configcollect_fields_and_class_varsradd_initializeradd_model_construct_method set_frozenr radjust_decorator_signaturesget_values_dictrr)rcr is_root_modelrrrr is_settingsrWrWrXrs(     z"PydanticModelTransformer.transformcs|jjjD]:\}}t|jtrA|jjdttrAtj t rAj j t vrAj j t kr.TN)rrnamesrrprqroriginal_decoratorsrcalleerrfDECORATOR_FULLNAMESMODEL_VALIDATOR_FULLNAMErr enumeraterfuncis_class)rcrrvrWr"rXrs        z4PydanticModelTransformer.adjust_decorator_signaturesModelConfigDatac Cs |j}t}d}d}|jD]\}}|||}|r"d}||qd}|jjD]}t|t t fs3q)t|t r|j d} t| t rG| j dkrHq)t|jtrmt|jj|jjD]\} } | dur`qW||j| | ddqWnNt|jtr|jjD]\} } t| tsqw||| j| qwn.t|t r|j dkrq)|jjD]}t|t sq|j d} t| t sq||| j |jq|r|jd|nd}q)|s|r|r|jr|js|jjrt|j||jjd dD]&}t|j vrq|j!t"|j#|j td D] \}}|$||qq|S) zhCollects the values of the config attributes that are used by the plugin, accounting for parent classes.FTNr model_config) lax_extraConfigzYSpecifying config in two places is ambiguous, use either Config attribute or class kwargsrr)%rr,keywordsrget_config_updateupdatedefsbodyrprrlvaluesrrrvaluerziprrrr#r r failhas_alias_generatorrr^rerror_required_dynamic_aliasesrrsrradd_plugin_dependencyr4rf setdefault)rcrrhas_config_kwargshas_config_from_namespacerexpr config_datastmtlhsarg_namerkey_expr value_exprsubstmtrr rWrWrXr s                z'PydanticModelTransformer.collect_configr-rJtuple[list[PydanticModelField] | None, list[PydanticModelClassVar] | None]cCs|j}i}i}t|jjddD]b}t|jvrq|jt|j |jtd D]2\}}t |||j} | |j|j| ||<|jj|} | r]| jr]t| jts]|jd| jq+|jtd D] \}}t |||<qgqt} t} ||jD]B} || ||}t|t r| jd}|r|jdkrt|j| q| |j|||j<qt|tr| jd}| |j|||j<qt|t|fS)zACollects the fields for the model, accounting for parent classes.rr rz7BaseModel field may only be overridden by another fieldrrroot)rreversedrrsrrr r;r4rfrrrrr$rrqrpr(r8rset%_get_assignment_statements_from_blockr3$collect_field_or_class_var_from_stmtr5r error_extra_fields_on_root_modeladdlistvalues)rcr-rr found_fieldsfound_class_varsrrrrsym_nodecurrent_field_namescurrent_class_vars_namesrA maybe_fieldrBrWrWrXrWsJ          z6PydanticModelTransformer.collect_fields_and_class_varsrArIterator[AssignmentStmt]ccsT|jD] }|js||EdHq|jdur&|jjs(||jEdHdSdSdSr\)r4is_unreachablerK else_body)rcrAr4rWrWrX,_get_assignment_statements_from_if_statements zEPydanticModelTransformer._get_assignment_statements_from_if_statementblockrccs>|jD]}t|tr|Vqt|tr||EdHqdSr\)r4rprrrZ)rcr[rArWrWrXrKs   z>PydanticModelTransformer._get_assignment_statements_from_blockrr dict[str, PydanticModelClassVar]1PydanticModelField | PydanticModelClassVar | Nonec Cs|j}|jd}t|trt|jr|jdkrdS|jsKt|jt rs z@PydanticModelTransformer.get_field_arguments..)rr)rcrrrrwr argumentsrWrrXr|s   z,PydanticModelTransformer.get_field_argumentscCs.|js||t|jrdS|jrdS|jjS)a@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_presentrr9rr^r)rcrrrWrWrXr"s z1PydanticModelTransformer.should_init_forbid_extrar9cCs8|D]}|jr dSq|r|D] }|jdurdSqdS)zReturns whether any fields on the model have a "dynamic alias", i.e., an alias that cannot be determined during static analysis. TNF)rr)rr9rrWrWrXr/s z1PydanticModelTransformer.is_dynamic_alias_presentN) rrrrrr0r^r]rTr[)rTr)rTr[)rTr,)r-r,rrrTrG)rArrTrW)r[rrTrW)rArr-r,rr\rTr])rvr$rrSrrrTr) rrurr,rrrrrTr[)rrurr,rrrTr[)rrurr0rrrTr[)F)rrSrrr.rrTr)rArrTr)rArrTr) rrurrrrrwrrrrTr)rrurr,rTr)rrur9rrTr)rrrrrrrbrrrrrZrKrLrhrrrr1 staticmethodrcrgr|rrrWrWrWrXrs0   $  L A   } & -  #     rc@sDeZdZdZ     ddd d Zdd dZdddZdddZdS)r,z(Pydantic mypy plugin model config class.Nr bool | Nonerrrr9cCs"||_||_||_||_||_dSr\)rrrrr9)rcrrrrr9rWrWrXrbAs  zModelConfigData.__init__rTrcCsdd|jDS)zReturns a dict of Pydantic model config names to their values. It includes the config if config value is not `None`. cSsi|] \}}|dur||qSr\rW)rkkvrWrWrXrTsz3ModelConfigData.get_values_dict..)__dict__rrrWrWrXrOzModelConfigData.get_values_dictrrr[cCs2|durdS|D] \}}t|||q dS)z$Update Pydantic model config values.N)rrr)rcrrrrWrWrXr2Vs zModelConfigData.updaterrSr rcCs"t||durt|||dSdS)zFSet default value for Pydantic model config if config value is `None`.N)rr)rcrr rWrWrXr<]szModelConfigData.setdefault)NNNNN) rrrrrrrrr9rr)rrrTr[)rrSr rrTr[)rrrrrbrr2r<rWrWrWrXr,>s  r,z pydantic-ormzInvalid from_attributes callPydanticzpydantic-configzInvalid config valuezpydantic-aliaszDynamic alias disallowedzpydantic-unexpectedzUnexpected behaviorzpydantic-fieldzUntyped field disallowedzInvalid Field defaultsz!Extra field on RootModel subclass model_namerr*rrr[cC|jd|d|tddS)zCEmits an error when the model does not have `from_attributes=True`.rlz$" does not have from_attributes=TruecodeN)r8 ERROR_ORM)rrrrWrWrXrlrrr0cCr)z0Emits an error when the config value is invalid.zInvalid value for "Config.rlrN)r8 ERROR_CONFIG)rrrrWrWrXrqrrcC|jd|tddS)znEmits required dynamic aliases error. This will be called when `warn_required_dynamic_aliases=True`. z#Required dynamic aliases disallowedrN)r8 ERROR_ALIASrrrWrWrXr:vrr:r8CheckerPluginInterface | SemanticAnalyzerPluginInterfacecCs4d}d|d}|d|d7}|j||tddS)z Emits unexpected behavior error.z6https://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!rN)r8ERROR_UNEXPECTED)rrrlink full_messagerWrWrXr~s rcCr)z;Emits an error when there is an untyped field in the model.zUntyped fields disallowedrN)r8 ERROR_UNTYPEDrrWrWrXrarracCr)z]Emits an error when there is more than just a root field defined for a subclass of RootModel.z2Only `root` is allowed as a field of a `RootModel`rN)r8ERROR_EXTRA_FIELD_ROOT_MODELrrWrWrXrMrrMcCr)zNEmits an error when `Field` has both `default` and `default_factory` together.z>Field default and default_factory cannot be specified togetherrN)r8ERROR_FIELD_DEFAULTSrrWrWrXrrrF8SemanticAnalyzerPluginInterface | CheckerPluginInterfacerrrrrzrrtvar_defTypeVarDef | NonerrcCs|j}||jvr|j|} | jrt| jtr|jj| jt|t r)| d} n| dg} |rE|p8t t |}ttd|dtdg} n|pJt |}ttd|dtg} | |}ggg} } }|D]}|jslJd| |j| |jj||jqct| || || }|r|g|_t||ttg}||_t|||_||_|jd||_|j|_||jvrt ||j}|j||j|<|rd|_!t||j}||_|j|_d|_"t#|t$dg|}|j|_t%t&|} nt%t&|} d| _| |j|<|j'jj|dS) zhVery closely related to `mypy.plugins.common.add_method_to_class`, with a few pydantic-specific changes.zbuiltins.functionrNT__pydantic_self__z"All arguments must be fully typed.rr)(rr$r{rprqrr3r4remover0rnamed_generic_typer>rBrr(r rrrrrr8 variablesrrr3rr+rfrrrC is_decoratedrrrr$rdefn)rrrrrzrrrrrv function_typefirstrrrpr signaturer*r_namerdecrWrWrXrsZ               rrdict[str, Any] | NonecCs|dsdStjdkrddl}nzddl}Wnty*ddl}|dYdSwt|d }| |WdS1s@wYdS)zReturns a dict of config keys to values. It reads configs from toml file and returns `None` if the file is not a toml file. z.tomlN) rzJNo TOML parser installed, cannot read configuration from `pyproject.toml`.rb) rsys version_infotomllibtomli ImportErrorwarningswarnopenload)rtoml_rrfrWrWrXrs       $r)rRrSrTrU)rr-rTr<)rrSrr*rrrTr[)rrSrr0rrrTr[)rr0rrrTr[)rrSrrrrrTr[)rr*rrrTr[)NNF)rrrrrrSrrrzr<rrrrrrrTr[)rrSrTr)r __future__rr configparserrtypingrrrmypy.errorcodesrmypy.expandtyperr mypy.nodesr r r r rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r( mypy.optionsr) mypy.pluginr*r+r,r-r.r/r0 mypy.pluginsr1mypy.plugins.commonr2 mypy.semanalr3mypy.server.triggerr4 mypy.stater5 mypy.typeopsr6 mypy.typesr7r8r9r:r;r<r=r>r?r@rA mypy.typevarsrB mypy.utilrC mypy.versionrD mypy_versionpydantic._internalrEpydantic.versionrFrGrrrrir r ryr~rr(r'rrrYrVr]rrrrr,rrrrrrrrrr:rrarMrrrrWrWrWrXs    !$     4       v .[ %              X