9tj+dZddlmZddlZddlmZddlmZddlm Z m Z ddl m Z ddl mZmZdd lmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm 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/m0Z0dd l1m2Z2dd l3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:dd l;mZ>ddl?m@Z@ddlAmBZBddlCmDZDddlEmFZFddlGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQddlRmSZSddlTmUZUddlVmWZXddlYmZZZddl[m\Z\dZ]dZ^dZ_dZ`dZadZbdZcdZdd Zed!Zfhd"Zgegd#hz Zhe\eXZid$Zjd%ZWdnd*ZkGd+d,e8ZlGd-d.Zmdod2ZnGd3d4ZoGd5d6ZpGd7d8ZqGd9d:eDZrGd;d<Zsdpd@Zte dAdBdCZue dDdEdCZve dFdGdCZwe dHdIdCZxe dJdKdCZye dJdLdCZze dJdMdCZ{dqdTZ|drdWZ}dsdXZ~dtd[Zdsd\Zdud]Z dvdwdjZdxdmZdS)yz^This module includes classes and functions designed specifically for use with the mypy plugin.) annotationsN)Iterator) ConfigParser)AnyCallable) ErrorCode) expand_typeexpand_type_by_instance) ARG_NAMED ARG_NAMED_OPTARG_OPTARG_POS ARG_STAR2 INVARIANTMDEFArgumentAssignmentStmtBlockCallExprClassDefContext DecoratorDictExpr EllipsisExpr ExpressionFuncDefIfStmtJsonDict MemberExprNameExprPassStmtPlaceholderNodeRefExpr StatementStrExprSymbolTableNodeTempNode TypeAliasTypeInfoVar)Options)CheckerPluginInterfaceClassDefContextDynamicClassDefContext 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.BaseModelzpydantic.main.create_modelz#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 K/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/pydantic/mypy.pypluginrXos  cdeZdZdZdfd Zdd Zdd Zdd ZddZddZ ddZ ddZ 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#   [ ) ) ,+ +trY/Callable[[DynamicClassDefContext], None] | Nonec*|tkr|jSdS)zARecognize `create_model()` calls as dynamic BaseModel subclasses.N)CREATE_MODEL_FULLNAME_pydantic_create_model_callbackrus rWget_dynamic_class_hookz%PydanticPlugin.get_dynamic_class_hooks , , ,7 7trYctxr1dict[str, Any]c|jS)zjReturn all plugin config data. Used by mypy to determine if cache needs to be discarded. )rb)rers 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)rerinfo_metaclasss rWrtz8PydanticPlugin._pydantic_model_metaclass_marker_callbacksm   7  F8NNNNN~ >&(BD I I @;?N  8 8 8 @ @rYr.ct}t|jj|jjD]\}}|dkrt |t r}|jvt |jtr |jj }Nt |jtr5t |jj tr|jj j j }|j |}|t |jts=|j t}|t |jtsdS|j}t|}t |tsJ|j |j||jj}|j|_|j |jt+t,|dS)zFMake variables assigned from `create_model()` usable as types by mypy.__base__N)rnzipcall arg_namesargsrkr#rlr)rgr*rr;rlookup_fully_qualified_or_nonerCbasic_new_typeinfonamelinemetaclass_typeadd_symbol_table_noder&r) rer base_fullnamearg_namearg_exprbase_sym base_info base_instancers rWrz.PydanticPlugin._pydantic_create_model_callbacks+ "%ch&8#(-"H"H E E Hh:%%*Xw*G*G%HMLehmX66E$,M$:MM s33E 8=CUW_8`8`E$,M$6$;$DM799-HH  :hmX#F#F w==>PQQHz(-'J'JM %i00 -22222w))#(M38=QQ'6 %%chd0K0KLLLLLrYr[r+rSr\)rgrRrSrh)rgrRrSrw)rgrRrSr|)rr1rSr)rr-rSr\)rr.rSr\)__name__ __module__ __qualname____doc__rdrqrvr{rrrortr __classcell__rfs@rWrVrV~s##""""""    !!!!     @ @ @ @MMMMMMMMrYrVcPeZdZUdZdZded<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++++ , ,rYrc*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_rr/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: z (of type )Nrfrom_attributesT)rrkr?itemr:ret_typer;rfrerror_unexpected_behaviorrcontextdefault_return_typemetadatar METADATA_KEYrmrnerror_from_attributesr)rctx_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 | Nonerintcolumnr 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>explicitrr rr r ) rerrrrrrrrrrrs rW to_argumentzPydanticModelField.to_argument8s;;|S)=UVV!%!4$+  :F :"..|STX.YYOO%i&899O+!V#1TT5ET--9     rYFrc|rFt|jtr,|j}t|_||_|j,|jjtj |j j 5t|}t|tsJ|r+|jD]#}t|tr t|_$t|j|jjj|i}|rlt|trWt#|jrC|jdj} | |cdddSt%| |} t'|| g}|cdddS#1swxYwY|jS)zABased on mypy.plugins.dataclasses.DataclassAttribute.expand_type.Nroot)rkrr@ copy_modifiedrvariancer self_typer6strict_optional_setr[strict_optionalrCr;rr id is_root_modelr rA) rerrrr modified_typefilled_with_typevarsarg expanded_type root_typeexpanded_root_types rWr zPydanticModelField.expand_typeUs $ * $)[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!. 26 : ?I (,! % % % % % % % %"*AM)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)rcBi|]}|j|Srrr)rfields rWrz6PydanticModelTransformer.transform..s&IIIuz5??#4#4IIIrYcBi|]}|j|Srr+)r class_vars rWrz6PydanticModelTransformer.transform..s(]]]Y9>9+>+>+@+@]]]rY)fields class_varsr)r&rrcollect_configcollect_fields_and_class_varsrrmBASESETTINGS_FULLNAMEadd_initializeradd_model_construct_method set_frozenr(radjust_decorator_signaturesget_values_dictrr)reris_a_root_modelrr/r0r, 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)rrkr%value)rir first_decs rW zGPydanticModelTransformer.adjust_decorator_signatures..sg   &3&/2f<rCQXAYAYr^a^gkr^r      rYTN)r&rnamesvaluesrkrlroriginal_decoratorsrcalleer rg(IMPLICIT_CLASSMETHOD_DECORATOR_FULLNAMESMODEL_VALIDATOR_FULLNAMEany enumeraterfuncis_class)rerprAs @rWr7z4PydanticModelTransformer.adjust_decorator_signatures s9>'..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)&r&rMkeywordsitemsget_config_updateupdatedefsbodyrkrrlvaluesr rrvaluerrrrrr%r?r(failhas_alias_generatorr!r r`rerror_required_dynamic_aliasesrmrorradd_plugin_dependencyr5rg setdefault)rerrhas_config_kwargshas_config_from_namespacerexpr config_datastmtlhsrrkey_expr value_exprsubstmtrr?s rWr1z'PydanticModelTransformer.collect_config'si ""!$)!,,,.. + +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!!$.... / rYrOrJtuple[list[PydanticModelField] | None, list[PydanticModelClassVar] | None]c|j}i}i}t|jjddD]j}t|jvr|jt|j |jtd D]\}}t |||j} | |j|j| ||<|jj|} | rH| jrAt#| jt$t&fs |jd| j|jtd D]"\}}t* |||<#lt-} t-} ||jD]} || ||}|| jd}t#|t6sJt#|trH|r!|jdkrt;|j| x| |j|||j<t#|t*r$| |j|||j<t?| t?| fS) zACollects the fields for the model, accounting for parent classes.rRr/z7BaseModel field may only be overridden by another fieldr0Nrr)!r&reversedrr^rrr(r_r5rgrTrrrrCrrlrkr*r"r[rset%_get_assignment_statements_from_blockrW$collect_field_or_class_var_from_stmtrYr r error_extra_fields_on_root_modeladdlistrD)rerOrr found_fieldsfound_class_varsrrr r,sym_nodecurrent_field_namescurrent_class_vars_namesre maybe_fieldrfs rWr2z6PydanticModelTransformer.collect_fields_and_class_varsssi79 =?SX\!B$/00 Q QD4=00 I + +,A$-,P,P Q Q Q"mL9(CIIKK   d*66tT49MM 11#(DIFFF%* T"8>--d33 jQTVePf6g6gINNQ  #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,KKKrYrerIterator[AssignmentStmt]c#K|jD]$}|js||Ed{V%|j.|jjs$||jEd{VdSdSdSr^)rXis_unreachablero else_body)rererXs 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^)rXrkrrr~)rerres rWroz>PydanticModelTransformer._get_assignment_statements_from_blocks}J S SD$// S D&)) SLLTRRRRRRRRR  S SrYrr0 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`. rrONzFType 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)/r&rYrkr rFis_valid_field_namer new_syntaxrZrrFrgDECORATOR_FULLNAMESerror_untyped_fieldsr(rrCrrlr"r(r[r* is_classvarrrBrr;get_has_default get_strictr is_inferredanalyze_simple_literal_typer9r> from_errorget_alias_infor!r r`rr]is_field_frozen_infer_dataclass_attr_init_typerrr)rererOr0rrfrprl node_typerrrrrr init_types rWrpz=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_typesr(r[r) rerprrdefaulttsetter super_info setter_types rWrz8PydanticModelTransformer._infer_dataclass_attr_init_typeNsq ( < N CH % % !X&& NI&&  U&+w// UV<> 3;3H3KQOOOINN#\afk#\#\#\^effff JAFKJJJGTTTrYr/list[PydanticModelField]rr:c Jd|jjjvr|jjjdjsdS|jj}t |j}|jp|j o|j du}t |j o|j }| |||||||d} |rm|j tj} t#| t$sJd| jvr(| jdj} t#| t&sJ| | j| j} t#| t*sJt-| jD]\} }|*|ds|ds2|j | j| }|0|dvr,|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_source_build_sourceskwargs)r return_type))r&rrCplugin_generatedr`rrrr!r r"r\get_field_argumentsr(rjr3rlrkr)rrr:rJr startswith anal_typeracceptChangeExplicitTypeOfAnyr>from_omitted_genericsr*appendrr should_init_forbid_extrar9rr add_methodr<)rer/rr:rrrrrrbase_settings_nodebase_settings_init_node func_typearg_idxranalyzed_variable_typervars rWr4z(PydanticModelTransformer.add_initializerts - - -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=(OCC6L5R5R 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*r r6rr[rrrrrrr9r>rrrr&rCr) rer/rr:rset_stroptional_set_strfields_set_argumentrrs rWr5z3PydanticModelTransformer.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)rc|jj}|D]}|j|j}|f|j}t |tr|p|j|_ Nt |tr%|j j s|j |||d}||_||_ |jdz|jz|_t#t$||j|j<dS)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. NF)r.)r&rrCrrrlrkr*r is_propertyr"r(final_iterationdeferrrg _fullnamer&r)rer/rrrr,rvrs rWr6z#PydanticModelTransformer.set_frozens y~ B BEz~~ej11H#mc3''&&,&?COO_55&di>W&IOO%%% ll4l>>"( $  3ch > '6tS'A'A 38$$# B BrYFrrrPModelConfigData | 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) Nrforbid) forbid_extrarT builtins.NoneF)r\) builtins.Truebuiltins.Falserr) r#rkr%r?rrerror_invalid_config_valuer(rMr rg)rerrrPrr\s rWrUz*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)rZrkr'rrFr#rgFIELD_FULLNAMErrrrfrr rercrrs rWrz(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) rZrkrrFr#rgrrrrr rs rWrz#PydanticModelTransformer.get_strict's{ 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) rZrkr'rrFr#rgrrrindexr%r?)rercrs rWrz'PydanticModelTransformer.get_alias_info7s{ 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. Frr) rZrkr'rrFr#rgrrJrrr )rercr@rrs rWrz(PydanticModelTransformer.is_field_frozenUs{ 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)rrr(r) rr,rrrr:rrrerrs rW z@PydanticModelTransformer.get_field_arguments..~s~      #("9    )7F;#I)A#0#IUZ65I       rY)r&r) rer/rrrrr:rr argumentsrs ` ``````` @rWrz,PydanticModelTransformer.get_field_argumentsnspy~                  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)r!r is_dynamic_alias_presentrr\rr`r)rer/rs rWrz1PydanticModelTransformer.should_init_forbid_extras] ' 6+B ,,VT&:T5U5UVV u   4!33rYr\cF|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)r/r\r,s rWrz1PydanticModelTransformer.is_dynamic_alias_presentsZ   E& tt     ;&44'urYN) rrrr$rr2r`r_rSr\)rSr)rSr\)rSrM)rOrMrrrSrj)rerrSrz)rrrSrz)rerrOrMr0rrSr)rpr&rrRrrrSr) r/rrrMr:rrrrSr\)r/rrr2rrrSr\r)rrRrrrPrrSr)rerrSr)rerrSr)rerrSr)r/rrrrrrrrrr:rrrrrrSr)r/rrrMrSr)r/rr\rrSr)rrrrr#rrdrr7r1r2r~rorprr4r5r6rU staticmethodrrrrrrrrrYrWrrsK ' ' '     + + + +""""H22226JJJJXBLBLBLBLHRRRRSSSSG G G G R$$$$L7X7X7X7Xr& & & & PBBBB2!!!!!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)rMz(Pydantic mypy plugin model config class.Nrrrrr r"r!r\rc v||_||_||_||_||_||_||_||_dSr^)rrrr r"r!r\r) rerrrr r"r!r\rs rWrdzModelConfigData.__init__sF) . 0!2 0#6  rYrSrcHd|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__rTrs rWr8zModelConfigData.get_values_dicts' IH!4!4!6!6HHHHrYrrr\c|dS|D]\}}t|||dS)z$Update Pydantic model config values.N)r8rTr)rerrrs rWrVzModelConfigData.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 rWr`zModelConfigData.setdefaults2 4   % D#u % % % % % & %rY)NNNNNNNN)rrrrrrr rr"rr!rr\rrrr)rrrSr\)rrRr?rrSr\)rrrrrdr8rVr`rrYrWrMrMs22%)"'+(,)-(,+/"(IIII    &&&&&&rYrMrr)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)r[ 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)r[ 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)r[ ERROR_ALIASrrs rWr]r]s# 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)r[ERROR_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)r[ ERROR_UNTYPEDrs rWrrs!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)r[ERROR_EXTRA_FIELD_ROOT_MODELrs rWrqrqs"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|gnd}t ||t/t1g}||_t3|||_||_|jdz|z|_|j|_||jvr*t?||j}|j||j|<|rwd|_ t||j}||_|j|_d|_!tE|tGd g|}|j|_tItJ|} ntItJ|} d| _| |j|<|j&jj|dS) zhVery closely related to `mypy.plugins.common.add_method_to_class`, with a few pydantic-specific changes.zbuiltins.functionr&NT__pydantic_self__z"All arguments must be fully typed.) variablesrr)'rrCrrkrlrrWrXremover2rnamed_generic_typer?rCrr*rrrrrrr:rr!r4rrLrgrrrD is_decoratedrrr r&rdefn)rrrrrrr rrrp function_typefirstrrrr signaturerKr_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~r3rrsrDATACLASS_FULLNAMErHrrGMYPY_VERSION_TUPLErrXrVr_rzrrrrrMrrrrrrERROR_FIELD_DEFAULTSr rrr]rrrqrrrrYrWrCsdd"""""" $$$$$$%%%%%% %%%%%%@@@@@@@@!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!D! +*****555555,,,,,,000000                        ('''''222222444444&&&&&&////// ' .4=5R(5K,?BtAu+u((' 55      TMTMTMTMTMVTMTMTMn+C+C+C+C+C+C+C+C\####4PTPTPTPTPTPTPTPTf        2^ ^ ^ ^ ^ ^ ^ ^ B     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