xgdZddlmZddlZddlmZddlmZddlm Z m Z m Z ddl m Z ddlmZmZdd lmZmZmZmZmZmZmZmZmZmZmZmZmZmZm 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.m/Z/m0Z0m1Z1dd l2m3Z3dd l4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:dd l;mZ>ddl?m@Z@ddlAmBZBddlCmDZDddlEmFZFddlGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQddlRmSZSddlTmUZUddlVmWZXddlYmZZZddl[m\Z\dZ]dZ^dZ_dZ`dZadZbdZcdZdd Zehd!Zfefd"hz Zge\eXZhd#Zid$ZWdmd)ZjGd*d+e8ZkGd,d-Zldnd1ZmGd2d3ZnGd4d5ZoGd6d7ZpGd8d9eDZqGd:d;Zrdod?Zse d@dAdBZte dCdDdBZue dEdFdBZve dGdHdBZwe dIdJdBZxe dIdKdBZye dIdLdBZzdpdSZ{dqdVZ|drdWZ}dsdZZ~drd[Zdtd\Z dudvdiZdwdlZdS)xz^This module includes classes and functions designed specifically for use with the mypy plugin.) annotationsN)Iterator) ConfigParser)AnyCallablecast) ErrorCode) expand_typeexpand_type_by_instance) ARG_NAMED ARG_NAMED_OPTARG_OPTARG_POS ARG_STAR2 INVARIANTMDEFArgumentAssignmentStmtBlockCallExprClassDefContext DecoratorDictExpr EllipsisExpr ExpressionFuncDefIfStmtJsonDict MemberExprNameExprPassStmtPlaceholderNodeRefExpr StatementStrExprSymbolTableNodeTempNode TypeAliasTypeInfoVar)Options)CheckerPluginInterfaceClassDefContext MethodContextPluginReportConfigContextSemanticAnalyzerPluginInterface)deserialize_and_fixup_type)set_callable_name)make_wildcard_trigger)state)TypeTranslator)map_type_from_supertype) AnyType CallableTypeInstanceNoneTypeType TypeOfAnyTypeType TypeVarType UnionTypeget_proper_type) fill_typevars)get_unique_redefinition_name) __version__)_fields)parse_mypy_version 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_validator3pydantic.deprecated.class_validators.root_validatorrI0pydantic.functional_serializers.model_serializerrNbuiltinsversionstrreturn type[Plugin]ctS)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)rQs H/var/www/html/nourish/venv/lib/python3.11/site-packages/pydantic/mypy.pypluginrXms  cTeZdZdZdfd Zdd Zdd Zdd ZddZddZ ddZ xZ S)rVzThe Pydantic mypy plugin.optionsr,rSNonect||_|j|_t |dSN)PydanticPluginConfig plugin_configto_data _plugin_datasuper__init__)selfr[ __class__s rWrdzPydanticPlugin.__init__sH1':: .6688 !!!!!rYfullnamerR(Callable[[ClassDefContext], None] | Nonec||}|r@t|jtr&|jt r|jSdS)zUpdate Pydantic model class.N)lookup_fully_qualified isinstancenoder*has_baseBASEMODEL_FULLNAME$_pydantic_model_class_maker_callback)rergsyms rWget_base_class_hookz"PydanticPlugin.get_base_class_hooksZ))(33  A:ch11 Ax  !344 A@@trYc*|tkr|jSdS)z,Update Pydantic `ModelMetaclass` definition.N)MODEL_METACLASS_FULLNAME)_pydantic_model_metaclass_marker_callbackrergs rWget_metaclass_hookz!PydanticPlugin.get_metaclass_hooks / / /A AtrY&Callable[[MethodContext], Type] | Nonec>|drtSdS)z-Adjust return type of `from_orm` method call.z .from_ormN)endswithfrom_attributes_callbackrus rWget_method_hookzPydanticPlugin.get_method_hooks#   [ ) ) ,+ +trYctxr1dict[str, Any]c|jS)zjReturn all plugin config data. Used by mypy to determine if cache needs to be discarded. )rb)rer|s rWreport_config_dataz!PydanticPlugin.report_config_datas   rYr.czt|j|j|j|j}|dSr^)PydanticModelTransformerclsreasonapir` transform)rer| transformers rWroz3PydanticPlugin._pydantic_model_class_maker_callbacks6.sw CGTM_`` rYc|jjrdS|jjj}|s Jdt |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)rer|info_metaclasss rWrtz8PydanticPlugin._pydantic_model_metaclass_marker_callbacksm   7  F8NNNNN~ >&(BD I I @;?N  8 8 8 @ @rYr[r,rSr\)rgrRrSrh)rgrRrSrw)r|r1rSr})r|r.rSr\) __name__ __module__ __qualname____doc__rdrqrvr{rrort __classcell__rfs@rWrVrV|s##""""""   !!!!     @ @ @ @ @ @ @ @rYrVcPeZdZUdZdZded<ded<ded<ded<dd ZddZdS)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_aliasesrboolrrrrr[r,rSr\c |jdSt|j}||didi}|jD]P}||d}t |t st d|t|||QdSt}| |j|jD]0}| t|d}t|||1dS)NtoolrHFz/Configuration value must be a boolean for key: )fallback) config_file parse_tomlget __slots__rkr ValueErrorsetattrrread getbooleanCONFIGFILE_KEY)rer[ toml_configconfigkeysettingr`s rWrdzPydanticPluginConfig.__init__s$   & F !455  " __VR0044_bIIF~ , , **S%00!'400^$%\WZ%\%\]]]c7++++  , , )NNM   w2 3 3 3~ , ,'22>3QV2WWc7++++ , ,rYr}c*fdjDS)z/Returns a dict of config names to their values.c2i|]}|t|S)r).0rres rW z0PydanticPluginConfig.to_data..s%BBBCWT3''BBBrY)rres`rWrazPydanticPluginConfig.to_datasBBBB4>BBBBrYNrrSr})rrrrr__annotations__rdrarrYrWr_r_sI ''''####,,,,&CCCCCCrYr_r|r/r=c|j}t|tr|j}t|tr"t|jt r|j}nMt|t r|}n5d|d|jjd}t||j |j |j S|jj t}||j S|jt"s|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)rrkr?itemr:ret_typer;rfrerror_unexpected_behaviorrcontextdefault_return_typemetadatar METADATA_KEYrmrnerror_from_attributesname)r|ctx_type model_typedetailpydantic_metadatars rWrzrzsGxH(H%%!=(L))'j9JH.U.U'& Hh ' '' PhPP(2D2MPPP!&#'3;???&&"044\BB && ? # #$6 7 7'&&'++Hb99==>OPPOd""jo2CGS[III  ""rYcdeZdZdZd-dZd.dZ d/d0d"Z d1d2d$Zd3d&Ze d4d(Z d5d+Z d,S)6PydanticModelFieldz5Based on mypy.plugins.dataclasses.DataclassAttribute.rrRalias str | None is_frozenrhas_dynamic_alias has_defaultstrict bool | Nonelineintcolumnr Type | Nonerr*c ||_||_||_||_||_||_||_||_| |_| |_ dSr^) rrrrrrrrrr) rerrrrrrrrrrs rWrdzPydanticModelField.__init__sP  "!2&     rY current_infotyped model_strictforce_optional use_aliasrr2force_typevars_invariantis_root_model_rootrSrc  |||||} |j|n|j} |s| r|||d} nttj} t | | d|rtn|s|jrtntS)zABased on mypy.plugins.dataclasses.DataclassAttribute.to_argument.NT)include_root_type)variabletype_annotation initializerkind) to_varrr r9r>explicitrrrr r ) rerrrrrrrrrrrs rW to_argumentzPydanticModelField.to_arguments;;|S)=UVV!%!4$+  :F :"..|STX.YYOO%i&899O+!V#1TT5ET--9     rYFrc|rFt|jtr,|j}t|_||_|j/|jj"tj |j j 5t|}t|tsJ|r+|jD]#}t|tr t|_$t|j|jjj|i}|rot|trZt#|jrFt%t&|jdj} t)| |} t+|| g}|cdddS#1swxYwY|jS)zABased on mypy.plugins.dataclasses.DataclassAttribute.expand_type.Nroot)rkrr@ copy_modifiedrvariancer self_typer6strict_optional_setr[strict_optionalrCr;argsr id is_root_modelrr=r rA) rerrrr modified_typefilled_with_typevarsarg expanded_type root_typeexpanded_root_types rWr zPydanticModelField.expand_type2s $ * $)[11 * $ 7 7 9 9 )2 &) 9 TY%8%D *3;+FGG % %'4\'B'B$!"6AAAAA+53855%c;775+4CL +DI 8K8NPd7e f f $SM8)L)LSQ^_l_qQrQrS!%T=+=f+E+J K KI)@M)Z)Z&$-}>P.Q$R$RM$! % % % % % % % % % % % % % % % %"ysextrafrozenralias_generatorrpopulate_by_namevalidate_by_namevalidate_by_aliaszset[str]tracked_config_fieldsrrrExpression | Statementrr2r`r_rSr\c>||_||_||_||_dSr^)_cls_reason_apir`)rerrrr`s rWrdz!PydanticModelTransformer.__init__s'   *rYrc6|jj}t|}|}|||\}}||dS|D] }|jdS |t}|||||| ||||| ||j |j du| d|Dd|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 NFT)r cBi|]}|j|Srrr)rfields rWrz6PydanticModelTransformer.transform..s&IIIuz5??#4#4IIIrYcBi|]}|j|Srr)r class_vars rWrz6PydanticModelTransformer.transform..s(]]]Y9>9+>+>+@+@]]]rY)fields class_varsr)rrrcollect_configcollect_fields_and_class_varsrrmBASESETTINGS_FULLNAMEadd_initializeradd_model_construct_method set_frozenrr adjust_decorator_signaturesget_values_dictrr)reris_a_root_modelrrrr is_settingss rWrz"PydanticModelTransformer.transformsAy~'--$$&&!??XX >Z/5  Ez!uu"mm$9::  VV[/JJJ '' _UUU  &-42GHHH ((***JI&III]]R\]]],,..' '  l# trYc|jjjD]}t |jt r|jjdt trt j trfj j tvrSj j tkr-tfdtjDsd|jj_dS)aWhen we decorate a function `f` with `pydantic.validator(...)`, `pydantic.field_validator` or `pydantic.serializer(...)`, 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()`, `field_validator()` or `serializer()` call as a `classmethod`. rc3K|]8\}}j|dkot|to |jdkV9dS)modeafterN) arg_namesrkr&value)rir first_decs rW zGPydanticModelTransformer.adjust_decorator_signatures..sg   &3&/2f<rCQXAYAYr^a^gkr^r      rYTN)rrnamesvaluesrkrlroriginal_decoratorsrcalleer!rg(IMPLICIT_CLASSMETHOD_DECORATOR_FULLNAMESMODEL_VALIDATOR_FULLNAMEany enumeraterfuncis_class)rerpr/s @rWr$z4PydanticModelTransformer.adjust_decorator_signaturess9>'..00 2 2C#(I.. 2H8; y(332"9#3X>>2"(15]]]"(15MMM    *3IN*C*C   N.2CHM*# 2 2rYModelConfigDatac |j}t}d}d}|jD]4\}}|||}|rd}||5d}|jjD]}t|ttfs t|tr|j d} t| tr | j dkrdt|jtrZt!|jj|jjD]3\} } | ||| | d4nt|jt&rV|jjD]I\} } t| t(s||| j| Jnt|tr|j dkrk|jjD]n}t|ts|j d} t| ts;||| j |jo|r|jd|nd} |s|r8|r6|jr/|js(|js!|jjrt;|j||jjd dD]}t@|j!vr|j"tG|j$|j!t@d D]\}}|%|||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 kwargsr)&rr;keywordsitemsget_config_updateupdatedefsbodyrkrrlvaluesr!rrvaluerzipr,rrr&r-rfailhas_alias_generatorrr r`rerror_required_dynamic_aliasesrmrorradd_plugin_dependencyr5rg setdefault)rerrhas_config_kwargshas_config_from_namespacerexpr config_datastmtlhsarg_namerkey_expr value_exprsubstmtrr-s rWrz'PydanticModelTransformer.collect_configsi ""!$)!,,,.. + +JD$00t<?? $// Tl1o!#x00CH4N4Ndk844 Z),T[-BDKDT)U)U]] ##+$ d&<&>! !/!,C%c844! MM$"8"87>"R"RSSSS   o(, % %  @ 9 @ @. @ 0 @5;4K @&D  @ /ty$???HL$ / /D4=00 I + +,A$-,P,P Q Q Q#}\:8DJJLL / / e!!$.... / rYr=rJtuple[list[PydanticModelField] | None, list[PydanticModelClassVar] | None]c|j}i}i}t|jjddD]c}t|jvr|jt|j |jtd D]\}}t |||j} | |j|j| ||<|jj|} | rA| jr:t#| jt$s |jd| j|jtd D]"\}}t( |||<#et+} t+} ||jD]} || ||}|| jd}t#|t4sJt#|trH|r!|jdkrt9|j| x| |j|||j<t#|t(r$| |j|||j<t=|t=|fS) zACollects the fields for the model, accounting for parent classes.r@rz7BaseModel field may only be overridden by another fieldrNrr) rreversedrrMrrrrNr5rgrBrrrr1rrlrkr+rJrset%_get_assignment_statements_from_blockrE$collect_field_or_class_var_from_stmtrGr!r error_extra_fields_on_root_modeladdlistr2)rer=rr found_fieldsfound_class_varsrrrrsym_nodecurrent_field_namescurrent_class_vars_namesrT maybe_fieldrUs rWrz6PydanticModelTransformer.collect_fields_and_class_varsMsi79 =?SX\!B$/00 Q QD4=00 I + +,A$-,P,P Q Q Q"mL9(CIIKK   d*66tT49MM 11#(DIFFF%* T"8>--d33 jPS6T6TINNQ  #mL9,GMMOO Q Q d)>)J)J4)P)P && Q),-0UU >>sxHH 9 9DCCD,XhiiK",q/Cc8,, , ,,+'9:: 9 9SX%7%74TYEEEE'++CH555-8L**K)>?? 9(,,SX666-8 *L''))**D1A1H1H1J1J,K,KKKrYrTrIterator[AssignmentStmt]c#K|jD]$}|js||Ed{V%|j.|jjs$||jEd{VdSdSdSr^)rFis_unreachabler_ else_body)rerTrFs rW,_get_assignment_statements_from_if_statementzEPydanticModelTransformer._get_assignment_statements_from_if_statementsI L LD& LEEdKKKKKKKKK > %dn.K %AA$.QQ Q Q Q Q Q Q Q Q Q & % % %rYblockrc#K|jD]L}t|tr|Vt|tr||Ed{VMdSr^)rFrkrrrn)rerorTs rWr_z>PydanticModelTransformer._get_assignment_statements_from_blocks}J S SD$// S D&)) SLLTRRRRRRRRR  S SrYrr dict[str, PydanticModelClassVar]1PydanticModelField | PydanticModelClassVar | Nonec H|j}|jd}t|tr$t j|jr |jdkrdS|jst|jtrbt|jj trCt|jj j tr|jj j j tvrdS|j|vrdSt|j|dS|jd}t|tsdSt j|jr |jdkrdS|jj|j}|dS|j}t|t&rdSt|t(r|jd|dSt|t,sdS|jrt1|jSt3|j}t|t6r+|jj dkr|jd|||} ||} |jr|jrk|jrd|j |jd} | r| |_n9|jd |tCtDj#|_|jr| rt1|jS|$|\} } | r/|j%s(|j&s!|j'j(rtS|j||*|}|+||j|}tY|j| | | | ||j-|j.||j S) a^Get pydantic model field from statement. Args: stmt: The statement. model_config: Configuration settings for the model. class_vars: ClassVars already known to be defined on the model. Returns: A pydantic model field if it could find the field in statement. Otherwise, `None`. rr=NzFType aliases inside BaseModel definitions are not supported at runtimezdataclasses.InitVarz%InitVar is not supported in BaseModelT)is_finalzGNeed type argument for Final[...] with non-literal default in BaseModel) rrrrrrrrrr)/rrGrkr!rFis_valid_field_namer new_syntaxrHrr4rgDECORATOR_FULLNAMESerror_untyped_fieldsrrr1rrlr#r)rJr+ is_classvarrrBrr;get_has_default get_strictrt is_inferredanalyze_simple_literal_typer9r> from_errorget_alias_inforr r`rrLis_field_frozen_infer_dataclass_attr_init_typerrr)rerTr=rrrUrprl node_typerrrrrr init_types rWr`z=PydanticModelTransformer.collect_field_or_class_var_from_stmtsil1o#x(( 0KCH0U0U Y\YaesYsYs4 4;11 t{18<< t{18(CC K&-6:MMM tx:%%t !D 1 1 14l1o#x(( 4*3844 N8R8R4hn  ** ;4x dO , ,  4 dI & &  INNX   4$$$  4   3(22 2$DI.. i * * y~/FJ_/_/_ INN7    **400 && 8  $2B )77 d7SSC :  ]$I$899 = 3[ 3(22 2#'#6#6t#<#<   <!2 <6B6S <"@ < +49d ; ; ;((.. 88chMM !/#;    rYrpr'rrRrrrc|j}|jr|St|j}t|ts|S|jd}|r!t|jtr|jd}|sJ|jr)tt|j|j|}nttj St|tr8|jtttgkrt!|jd|S|jd|jjd|n)|jd|jjd||S)zvInfer __init__ argument type for an attribute. In particular, possibly use the signature of __set__. __set__rPz(Unsupported signature for "__set__" in ""zUnsupported "__set__" in ")rimplicitrBrkr;rrlrget_containing_type_infor8r9r> unannotatedr: arg_kindsrr arg_typesrrJr) rerprrdefaulttsetter super_info setter_types rWrz8PydanticModelTransformer._infer_dataclass_attr_init_type(sq ( < N CH % % !X&& NI&&  U&+w// UV<> 3;3H3KQOOOINN#\afk#\#\#\^effff JAFKJJJGTTTrYrlist[PydanticModelField]rr'c Nd|jjjvr|jjjdjsdS|jj}t |j}|jp|j o|j du}t |j o|j }| |||||||d} |ro|j tj} t#| t$sJd| jvr*| jdj} t#| t&sJ| | j| j} t#| t*sJt-| jD]\} }|*|ds|ds2|j | j| }|2|dkr,|t9t:j}t?||}| tC||dtD|#||sPt?d }| tC|tIt:j%dtLtO|j |jd| tQ dS) zAdds a fields-aware `__init__` method to the class. The added `__init__` will be annotated with types vs. all `Any` depending on the plugin settings. rdNFT)rrrequires_dynamic_aliasesrr'rr____cli_settings_sourcekwargs)r return_type))rrr1plugin_generatedr`rrrrr rrKget_field_argumentsrrjr rlrkr*rrr:r8r, startswith anal_typeracceptChangeExplicitTypeOfAnyr>from_omitted_genericsr+appendrrshould_init_forbid_extrar9rr add_methodr<)rerrr'rrrrrrbase_settings_nodebase_settings_init_node func_typearg_idxrVanalyzed_variable_typervars rWr!z(PydanticModelTransformer.add_initializerNs - - -din6J:6V6g - F"-FM** 0KF4KLvQWQiqvQv #'(B(b6KbGb#c#c '' %%=#'%)(     _!%!A!ABW!X!X!] 0(;; ; ;;/555*<*B:*N*S'!"97CCCCC*6;R;W;c 7 >>>>-6y7J-K-K _ _)#+x/B/B4/H/H+PXPcPcdgPhPh+$151D1DYEXY`Ea1b1b.1=(NdBdBd6L5R5R 7 8W X X662$'x1G#H#H HX7MtU\$]$]^^^^,,VV<< Uh--C KKgi.@&A&A4SS T T T49di$HJJWWWWWWrYc 8|jtd|jtdg}t|t g}t t d||dt}tj |jj j 5| |dt|jdd||}dddn #1swxYwY|||sPt d} |t | t#t$jdt(|r||gzn|g|z}t+|j|jd |t/|jjd dS) zAdds a fully typed `model_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.str _fields_setNTF)rrrrr'rrmodel_construct)rris_classmethod)r named_type BUILTINS_NAMErAr<rr+rr6rr[rrrrrrr9r>rrrrrCr) rerrr'rset_stroptional_set_strfields_set_argumentrrs rWr"z3PydanticModelTransformer.add_model_construct_methods)&&-'='='= @T@TXeUkUkUk@l@l?mnn$gxzz%:;;&s=:J'K'KM]_celmm  &ty'8'H I I  ++!&-00).'+,D               ,,VV<< Uh--C KKgi.@&A&A4SS T T T/<^t*+++CVBWZ^B^ I I %din55       s*/C%%C),C)r c|jj}|D]C}|j|j}||j}t |tr|p|j|_ Ot |tr&|j j s|j  t|}n#t$rt!|}YnwxYwd|d|jd} t%| |j |j|||d}||_||_ |jdz|jz|_t-t.||j|j<EdS)zMarks 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)r.)rrr1rrrlrkr+r is_propertyr#rfinal_iterationdeferrR TypeErrorreprrfrrrg _fullnamer'r) rerrr rrrfrvar_strrs rWr#z#PydanticModelTransformer.set_frozensk y~ B BEz~~ej11H#mc3'' L&,&?COO_55 Ldi>W LIOO%%%%,"%c(($,,,"&s)),SwRR#-RRRF-fdiKKKKll4l>>"( $  3ch > '6tS'A'A 38$$/ B BsB**CCFrrr>ModelConfigData | Nonec ||jvrdS|dkrlt|tr |jdk}n;t|tr |jdk}n|st ||j|dSt|S|dkr4d}t|tr |j dkrd}t| St|tr |j d vrtd i||j d kiSt ||j|dS) zDetermines the config update due to a single kwarg in the ConfigDict definition. Warns if a tracked config attribute is set to a value the plugin doesn't know how to interpret (e.g., an int) Nr forbid) forbid_extrar T builtins.NoneF)rK) builtins.Truebuiltins.Falserr) rrkr&r-r rerror_invalid_config_valuerr;r!rg)rerrr>rrKs rWrCz*PydanticModelTransformer.get_config_updates6 t1 1 14 7??#w'' "yH4 C,, "x83  E/tTYDDDt" === = $ $ $"& #x(( ,S\_-L-L&+#"7JKKK K c8 $ $ N9\)\)\"MMdCLO,K%LMM M"4C888trYc|j}t|trdSt|trt|jt rx|jjtkrct|j |j D]F\}}||dkr|j tucS|dkr#t|to |jdk cSGdSt|t S)zUReturns a boolean indicating whether the field defined in `stmt` is a required field.FNrdefault_factoryr)rHrkr(rr4r$rgFIELD_FULLNAMErIrr,rfrr!rTrRrrs rWrzz(PydanticModelTransformer.get_has_defaults{ dH % % 5 dH % % *T['*J*J t{OcguOuOu !DN;; _ _ T<49#4#4= <<<<,,, *3 9 9 ]clo>]^^^^-5dL1111rYrcP|j}t|trt|jtro|jjt krZt|j|j D]?\}}|dkr t|tr|jdkrdS|jdkrdSdSdS)zEReturns a the `strict` value of a field if defined, otherwise `None`.rrTrFN) rHrkrr4r$rgrrIrr,r!rs rWr{z#PydanticModelTransformer.get_stricts{ dH % % *T['*J*J t{OcguOuOu DN;;   T8##c8,,%|66#tt)999$uutttrYtuple[str | None, bool]c|j}t|trdSt|tr/t|jt r|jjtksdSd|jvr&|j |j d}n1d|jvr&|j |j d}ndSt|tr |j dfSdS)aReturns 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. )NFvalidation_aliasrF)NT) rHrkr(rr4r$rgrr,rindexr&r-)rTrRrs rWrz'PydanticModelTransformer.get_alias_infos{ dH % % ; tX & & +5dk7+K+K PTP[PdhvPvPv;  / /)DN001CDDECC  & &)DN0099:CC; c7 # # 9e# #:rYcl|j}t|trdSt|tr/t|jt r|jjtksdSt|j D]:\}}|dkr/|j |}t|to |jdkcS;dS)alReturns whether the field is frozen, extracted from the declaration of the field defined in `stmt`. Note that this is only whether the field was declared to be frozen in a ` = Field(frozen=True)` sense; this does not determine whether the field is frozen because the entire model is frozen; that is handled separately. Fr r) rHrkr(rr4r$rgrr8r,rr!)rTrRr.rVrs rWrz(PydanticModelTransformer.is_field_frozen2s{ dH % % 5 tX & & +5dk7+K+K PTP[PdhvPvPv5$T^44 U UKAx8##il!#x00TS\_5TTTT$urYrrrrrlist[Argument]c \ jj  f d|D} | S)zHelper function used during the construction of the `__init__` and `model_construct` method signatures. Returns a list of mypy Argument instances for use in the generated signatures. c  g|]:} r|j | p jo |jdk;S)r)rrrrrrr)rrrr) rrrrrr'rrrerrs rW z@PydanticModelTransformer.get_field_arguments..[s~      #("9    )7F;#I)A#0#IUZ65I       rY)rr) rerrrrrr'rr argumentsrs ` ``````` @rWrz,PydanticModelTransformer.get_field_argumentsKspy~                  rYc|js1|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)rr is_dynamic_alias_presentrrKrr`r)rerrs rWrz1PydanticModelTransformer.should_init_forbid_extraks] ' 6+B ,,VT&:T5U5UVV u   4!33rYrKcF|D] }|jrdS |r|D] }|jdS dS)zReturns whether any fields on the model have a "dynamic alias", i.e., an alias that cannot be determined during static analysis. TNF)rr)rrKrs rWrz1PydanticModelTransformer.is_dynamic_alias_presentxsZ   E& tt     ;&44'urYN) rrrrrr2r`r_rSr\)rSr)rSr\)rSr;)r=r;rrrSrZ)rTrrSrj)rorrSrj)rTrr=r;rrqrSrr)rpr'rrRrrrSr) rrrr;r'rrrrSr\)rrrr2r rrSr\r)rrRrrr>rrSr)rTrrSr)rTrrSr)rTrrSr)rrrrrrrrrrr'rrrrrrSr)rrrr;rSr)rrrKrrSr)rrrrrrrdrr$rrrnr_r`rr!r"r#rC staticmethodrzr{rrrrrrrYrWrrsK ' ' '     + + + +""""H22226JJJJXBLBLBLBLHRRRRSSSSG G G G R$$$$L4X4X4X4Xl& & & & PBBBB>!!!!!F222\2*   \ \:\B*/@ 4 4 4 4   \   rYrc,eZdZdZd fd Zd d ZxZS) rzrrr)rers rW visit_anyz!ChangeExplicitTypeOfAny.visit_anys. =I. . .??t/@?AA AHrY)rrrSr\)rr9rSr=)rrrrrdrrrs@rWrrsWFFrYrcDeZdZdZ ddd ZddZddZddZdS)r;z(Pydantic mypy plugin model config class.Nrrr rr rrrKrc v||_||_||_||_||_||_||_||_dSr^)rr rr rrrKr) rerr rr rrrKrs rWrdzModelConfigData.__init__sF) . 0!2 0#6  rYrSr}cHd|jDS)zReturns a dict of Pydantic model config names to their values. It includes the config if config value is not `None`. ci|] \}}||| Sr^r)rkvs rWrz3ModelConfigData.get_values_dict..sHHHA!-1---rY)__dict__rBrs rWr%zModelConfigData.get_values_dicts' IH!4!4!6!6HHHHrYrrr\c|dS|D]\}}t|||dS)z$Update Pydantic model config values.N)r%rBr)rerrrs rWrDzModelConfigData.updatesV > F**,,2244  DAq D!Q      rYrrRr-rcLt||t|||dSdS)zFSet default value for Pydantic model config if config value is `None`.N)rr)rerr-s rWrOzModelConfigData.setdefaults2 4   % D#u % % % % % & %rY)NNNNNNNN)rrr rrrr rrrrrrKrrrr)rrrSr\)rrRr-rrSr\)rrrrrdr%rDrOrrYrWr;r;s22%)"'+(,)-(,+/"(IIII    &&&&&&rYr;rr*rc6|tS)zXReturn whether the type info is a root model subclass (or the `RootModel` class itself).)rmROOT_MODEL_FULLNAME)rs rWrrs ==, - --rYz 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\cH|d|d|tdS)zCEmits an error when the model does not have `from_attributes=True`.rz$" does not have from_attributes=TruecodeN)rJ ERROR_ORM)rrrs rWrrs-HH A A A A7QZH[[[[[rYrr2cH|d|d|tdS)z0Emits an error when the config value is invalid.zInvalid value for "Config.rrN)rJ ERROR_CONFIG)rrrs rWrrs,HH 1$ 1 1 17HNNNNNrYc@|d|tdS)znEmits required dynamic aliases error. This will be called when `warn_required_dynamic_aliases=True`. z#Required dynamic aliases disallowedrN)rJ ERROR_ALIASrrs rWrLrLs# HH 2G+HNNNNNrYr8CheckerPluginInterface | SemanticAnalyzerPluginInterfacecbd}d|d}|d|dz }|||tdS)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)rJERROR_UNEXPECTED)rrrlink full_messages rWrrsO DDWVWWWL\T\\\\LHH\7)9H:::::rYc@|d|tdS)z;Emits an error when there is an untyped field in the model.zUntyped fields disallowedrN)rJ ERROR_UNTYPEDrs rWrxrxs!HH (' HFFFFFrYc@|d|tdS)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)rJERROR_EXTRA_FIELD_ROOT_MODELrs rWraras"HH A7QmHnnnnnrYF8SemanticAnalyzerPluginInterface | CheckerPluginInterfacerrrrrrrtvar_defTypeVarType | Nonerc,|j}||jvrR|j|} | jr>t| jt r$|jj| jt|tr| d} n| dg} |rE|ptt|}ttd|dt dg} n6|pt|}ttd|dt g} | |z}ggg}} } |D]f}|js Jd| |j| |jj||jgt-| || || }|r|g|_t ||t1t3g}||_t5|||_||_|jdz|z|_|j|_||jvr*tA||j}|j||j|<|rwd|_!t||j}||_|j|_d|_"tG|tIdg|}|j|_tKtL|} ntKtL|} 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)(rr1rrkrlrrErFremover2rnamed_generic_typer?rCrr+rrrrrrr: variablesrr"r4rr:rgrrrD is_decoratedrrr!r'rdefn)rrrrrrrrrrp function_typefirstrr,rr signaturer9r_namerdecs rWrrs 8D tzj   +Jsx$A$A + HM  * * *#677H':;; ../BBGG O>-*=*=!>!> #f++y$FFG4t!4!4 #122ItWMMN 4r?r@rArB mypy.typevarsrC mypy.utilrD mypy.versionrE mypy_versionpydantic._internalrFpydantic.versionrGrrrnr rrsrDATACLASS_FULLNAMEr6rwr5MYPY_VERSION_TUPLErrXrVr_rzrrrrr;rrrrrrERROR_FIELD_DEFAULTSrrrrLrrxrarrrrYrWr4sdd"""""" $$$$$$%%%%%%&&&&&&&&&&%%%%%%@@@@@@@@!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!D! +*****555555,,,,,,000000                        ('''''222222444444&&&&&&////// ' .=5R(5K,?BtAu+u((' 55      3@3@3@3@3@V3@3@3@l+C+C+C+C+C+C+C+C\####4MTMTMTMTMTMTMTMT`        2a a a a a a a a H     n   (&(&(&(&(&(&(&(&V.... In&Dj Q Q y*,BJOO i(*DjQQ 924I:VV *,F SS  y!13KZXX(y)9;^`jkk\\\\ OOOO OOOO;;;;GGGG oooo"#' U%U%U%U%U%prY