U e @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>mGZSYnXdZUdZVdZWdZXdZYdZZdZ[e\e e]dfdddZ^e^ePZ_e_dkrdndZ`d ZOe\d!dd"d#ZaGd$d%d%e5ZbGd&d'd'Zce4e d(d)d*ZdGd+d,d,ZeGd-d.d.ZfGd/d0d0Zged1d2d3Zhed4d5d3Zied6d7d3Zjed8d9d3Zked:d;d3Zled:dd?Zne\e7edd@dAdBZoe7eddCdDdEZpe\e e1e7feddFdGdHZqe7eddCdIdJZre1eddCdKdLZsdYe2e\eee ee eeSetetddN dOdPZue ee(fe\dQdRdSZve ee(fe\dQdTdUZwe\eee\efdVdWdXZxdS)ZN) 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-metadatazpydantic.main.BaseModelz"pydantic.env_settings.BaseSettingszpydantic.main.ModelMetaclasszpydantic.fields.Fieldzpydantic.dataclasses.dataclass.)versionreturncCsttt|dddS)N+r.)tuplemapint partitionsplitrFrPN/var/www/html/assets/scripts/venv/lib/python3.8/site-packages/pydantic/mypy.pyparse_mypy_versionUsrR)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). )PydanticPluginrOrPrPrQplugin`srWcseZdZeddfdd ZeddddZeeee gdfdd d Z ed dd d Z eeee ge fdddZeeee gdfdddZeeeefdddZe ddddZe ddddZeddddZZS)rVNoptionsrGcs&t||_|j|_t|dSN)PydanticPluginConfig plugin_configto_data _plugin_datasuper__init__)selfrY __class__rPrQr`ks  zPydanticPlugin.__init__z+Optional[Callable[[ClassDefContext], None]])fullnamerGcCs:||}|r6t|jtr6tdd|jjDr6|jSdS)Ncss|]}t|tkVqdSrZ) get_fullnameBASEMODEL_FULLNAME.0baserPrPrQ tsz5PydanticPlugin.get_base_class_hook..)lookup_fully_qualified isinstancenoder'anymro$_pydantic_model_class_maker_callbackrardsymrPrPrQget_base_class_hookps  z"PydanticPlugin.get_base_class_hookcCs|tkr|jSdSrZ)MODEL_METACLASS_FULLNAME)_pydantic_model_metaclass_marker_callbackrardrPrPrQget_metaclass_hookxsz!PydanticPlugin.get_metaclass_hookz+Optional[Callable[[FunctionContext], Type]]cCs"||}|r|jtkr|jSdSrZ)rkrdFIELD_FULLNAME_pydantic_field_callbackrqrPrPrQget_function_hook}s z PydanticPlugin.get_function_hookcCs|drtSdS)Nz .from_orm)endswithfrom_orm_callbackrvrPrPrQget_method_hooks zPydanticPlugin.get_method_hookcCs|tkrtdkrtjSdS)zMark pydantic.dataclasses as dataclass. Mypy version 1.1.1 added support for `@dataclass_transform` decorator. )r~N)DATACLASS_FULLNAMEMYPY_VERSION_TUPLEr2Zdataclass_class_maker_callbackrvrPrPrQget_class_decorator_hooksz'PydanticPlugin.get_class_decorator_hookctxrGcCs|jS)zjReturn all plugin config data. Used by mypy to determine if cache needs to be discarded. )r^)rarrPrPrQreport_config_datasz!PydanticPlugin.report_config_datacCst||j}|dSrZ)PydanticModelTransformerr\ transform)rarZ transformerrPrPrQrps 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)rarZinfo_metaclassrPrPrQrus   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()r~default_factoryz9"default_factory" is no longer second argument in Field())riargsNcss|]}t|tVqdSrZ)rlr=)rhargrPrPrQrjsz:PydanticPlugin._pydantic_field_callback..c3s|] }VqdSrZrP)rh_Zdefault_any_typerPrQrjs)default_return_typeZcallee_arg_namesrr+error_default_and_default_factory_specifiedapicontext arg_typesrlrr9ritemsr6ret_typerallrJ) rarZ default_argsZdefault_factory_argsZ default_typeZ default_argZdefault_factory_typerrrPrrQrys6         z'PydanticPlugin._pydantic_field_callback)__name__ __module__ __qualname__r*r`strrsrrr,rwrzr.r r}rr0rrrrprur-ry __classcell__rPrPrbrQrVjs  rVc@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_fieldsrrrrrrNrXcCs|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__rlbool ValueErrorsetattrrread getbooleanCONFIGFILE_KEY)rarYZ toml_configconfigkeyZsettingr\rPrPrQr`s       zPydanticPluginConfig.__init__rGcsfddjDS)Ncsi|]}|t|qSrP)r)rhrrarPrQ sz0PydanticPluginConfig.to_data..)rrrPrrQr]szPydanticPluginConfig.to_data) rrrrr__annotations__r*r`rrrr]rPrPrPrQr[s r[rcCs|j}t|tr|j}t|tr4t|jtr4|j}n.NFT)rcSsi|]}|j|qSrP)name serializerhfieldrPrPrQr@sz6PydanticModelTransformer.transform..)fieldsr)rrradjust_validator_signaturescollect_configcollect_fieldsrnroadd_initializeradd_construct_method set_frozenrrset_values_dictrr)rarrrr is_settingsrPrPrQr+s    z"PydanticModelTransformer.transformcCsb|jjjjD]L\}}t|jtr|jjd}t|t rt|j t r|j j 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#pydantic.class_validators.validatorTN)rrrnamesrrlrmrZoriginal_decoratorsrcalleerrdfuncis_class)rarrrZ first_decrPrPrQrDs    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. ZConfigr~Nr)rrrdefsbodyrlrrrupdateget_config_updatehas_alias_generatorrr\rerror_required_dynamic_aliasesrrrorradd_plugin_dependencyr4rer setdefault) rarrrstmtsubstmtrrvaluerPrPrQrVs2      z'PydanticModelTransformer.collect_configPydanticModelField) model_configrGc 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_aliaslinecolumnr~rc3s|]}|jkr|VqdSrZr)rharrPrQrjs z:PydanticModelTransformer.collect_fields..)+rrsetrrrlrlvaluesrrCrZ new_syntaxr\rerror_untyped_fieldsrrrrrmr!r)Z is_classvarget_is_requiredget_alias_inforrrappendrrraddcopyrorrrr4rer deserializeremove)rarrrrZ known_fieldsrlhsrrrmrrrZ all_fieldsrZsuperclass_fieldsdatarrPrrQrust               z'PydanticModelTransformer.collect_fields)rrrrGc 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_aliaskwargsNr`)rr\rrrrget_field_argumentsshould_init_forbid_extrar)rrr5r;explicitrrrr add_methodr8) rarrrrrrrZinit_argumentsvarrPrPrQrs"   z(PydanticModelTransformer.add_initializer)rrGc 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_PydanticBaseModelrI)r~r construct) return_type self_typetvar_defis_classmethod)rr named_type BUILTINS_NAMEr>r8rr)rrrrdrr=r5r;Zfrom_omitted_genericsr(rDr%rrrrlr) rarrZset_strZoptional_set_strZfields_set_argumentZconstruct_argumentsobj_typeZself_tvar_nameZ tvar_fullnameZtvdZself_tvar_exprr rPrPrQrsP$  z-PydanticModelTransformer.add_construct_method)rrrGc 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)rrI)rrrrrrrmrlr)Z is_propertyr!rZfinal_iterationZdeferr TypeErrorreprrcrto_varrer _fullnamer%r) rarrrrrZsym_noderZvar_strrrPrPrQrs*   z#PydanticModelTransformer.set_frozen)rrGcCs|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)rrlrrrrvaluer#rrerror_invalid_config_valuerrrrd)rarrrrrPrPrQr6s(       z*PydanticModelTransformer.get_config_update)rrrrGcCs|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)rrlr&r?rrrrtype_has_implicit_defaultrrr"rdrxzipr arg_namesrcr)rrrexprZ value_typerrrPrPrQrRs  " z(PydanticModelTransformer.get_is_required)type_rGcCs4t|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|tVqdSrZ)rlr8r5)rhrrPrPrQrjvszEPydanticModelTransformer.type_has_implicit_default..F)rlr5r>rnr)rrPrPrQrls z2PydanticModelTransformer.type_has_implicit_default)rrGcCs|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) rrlr&rrr"rdrx enumeraterrr#r)rriZarg_namerrPrPrQr}s$     z'PydanticModelTransformer.get_alias_info)rrrrrGcs&|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_argumentrrrrrrPrQ sz@PydanticModelTransformer.get_field_arguments..)rrr)rarrrr argumentsrPr rQrs  z,PydanticModelTransformer.get_field_arguments)rrrGcCs.|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)rarrrPrPrQrs z1PydanticModelTransformer.should_init_forbid_extra)rrrGcCs8|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)rrrrPrPrQr#s z1PydanticModelTransformer.is_dynamic_alias_present)"rrrrrrrr,r[r`rrrrrrrrrrrr staticmethodrrrr:rr rrrrr#rPrPrPrQrs>  T6! rc@steZdZeeeeeeedddZeee dddZ eeeee ddd Z e d d d Zeee dd ddZdS)rrcCs(||_||_||_||_||_||_dSrZr)rarrrrrrrPrPrQr`s zPydanticModelField.__init__)rrrGcCs,|j}|r|jdk r|j}t|||jjSrZ)rrr)r)rarrrrPrPrQrszPydanticModelField.to_var)rrrrrGcCsR|r"||jjdk r"||jj}n ttj}t||||d|sF|jsJtnt dS)N)variabletype_annotationZ initializerkind) rrr5r;rrrrrr )rarrrrr&rPrPrQrs  zPydanticModelField.to_argumentrcCs|jSrZ)__dict__rrPrPrQrszPydanticModelField.serialize)rrrGcCs |f|SrZrP)rrrrPrPrQrszPydanticModelField.deserializeN)rrrrrrrLr`r'r)rrrrr classmethodrrPrPrPrQrs  rc@szeZdZdeeeeeeeeeeeedddZeeefdddZ edddd d Z eedd d d Z dS)rNrrrrrrcCs(||_||_||_||_||_||_dSrZr*)rarrrrrrrPrPrQr`s zModelConfigData.__init__rcCsdd|jDS)NcSsi|]\}}|dk r||qSrZrP)rhkvrPrPrQrsz3ModelConfigData.set_values_dict..)r(rrrPrPrQrszModelConfigData.set_values_dict)rrGcCs2|dkr dS|D]\}}t|||qdSrZ)rrr)rarr+r,rPrPrQrszModelConfigData.update)rrrGcCst||dkrt|||dSrZ)rr)rarrrPrPrQrszModelConfigData.setdefault)NNNNNN) rrrrrr`rrrrrrrPrPrPrQrs" rz pydantic-ormzInvalid from_orm callZPydanticzpydantic-configzInvalid config valuezpydantic-aliaszDynamic alias disallowedzpydantic-unexpectedzUnexpected behaviorzpydantic-fieldzUntyped field disallowedzInvalid Field defaults) model_namerrrGcCs|jd|d|tddS)N"z" does not have orm_mode=Truecode)fail ERROR_ORM)r-rrrPrPrQrsr)rrrrGcCs|jd|d|tddS)NzInvalid value for "Config.r.r/)r1 ERROR_CONFIG)rrrrPrPrQrsr)rrrGcCs|jd|tddS)Nz#Required dynamic aliases disallowedr/)r1 ERROR_ALIASrrrPrPrQrsr)rrrrGcCs4d}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!r/)r1ERROR_UNEXPECTED)rrrlinkZ full_messagerPrPrQr s rcCs|jd|tddS)NzUntyped fields disallowedr/)r1 ERROR_UNTYPEDr5rPrPrQr*srcCs|jd|tddS)Nz>Field default and default_factory cannot be specified togetherr/)r1ERROR_FIELD_DEFAULTSr5rPrPrQr.srF) rrrrr r r is_newrGcCs|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 .functionrITr))+rrrZplugin_generatedrlrmrrrrr@rr)r<Zmake_normalizedrr&rrrr%r'rr r r6 variablesrr r3rrrerrrAZ is_decoratedr rrr%rZdefn)rrrrr r r r;rrrfirstrrZ arg_kindsrZ function_type signaturerZr_namer,decrPrPrQr2sV            r)xrGcCs|j}t|r|S|Szc Used for compatibility with mypy 0.740; can be dropped once support for 0.740 is dropped. )rdcallablerAfnrPrPrQresrecCs|j}t|r|S|SrB)rrCrDrPrPrQrsr)rrGc 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_infoZtomllibtomli ImportErrortomlwarningswarnopenload)rZ read_modeZtoml_rOrfrPrPrQrs"       r)NNFF)yrJ 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.utilsrCrDrMrrrfrrtrxrrrLrRrr rWrVr[r|rrrr2r3r4r7r9r:rrrrrrrrrerrrPrPrPrQs , | $   8     u%.%          S