h-i)dZddlmZddlZddlZddlZddlZddlm Z m Z ddlm Z m Z ddlm Z mZmZddlmZmZmZmZmZddlmZd d lmZd d lmZd d lmZd d lmZddlm Z m!Z!m"Z"ddl m#Z#ddl$m%Z%ddl&m'Z'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0ejbr2d dl2m3Z3GddejhZ5Gdde5ejhZ6neZ7d#d$dZ8dd d%dZ9d&d Z: d'd!Z;d(d"Z)clstypes_namespacerCfieldss r.set_dataclass_fieldsrGHs$-S1L %c? VF$Cr0T) raise_errorsc d t|drtjdt|t j |}t ||t|}t|||}t||j|}dd}|jd|_ ||_ |j|_||_t!|dd} |r+||t#t%|j&d|d } n|j'|d} |j1|} |j3| } t7j8d |}| |_t=| ||j>|jd| |j@x|_! tE| | |_#|jHr5tK|jLd fd } | jOd||_&y#t($r2} |rt+||j,d | j.d Yd} ~ yd} ~ wwxYw#|j4$rt+||j,d YywxYw)aFinish building a pydantic dataclass. This logic is called on a class which has already been wrapped in `dataclasses.dataclass()`. This is somewhat analogous to `pydantic._internal._model_construction.complete_model_class`. Args: cls: The class. config_wrapper: The config wrapper instance. raise_errors: Whether to raise errors, defaults to `True`. types_namespace: The types namespace. Returns: `True` if building a pydantic dataclass is successfully completed, `False` otherwise. Raises: PydanticUndefinedAnnotation: If `raise_error` is `True` and there is an undefined annotations. __post_init_post_parse__zVSupport for `__post_init_post_parse__` has been dropped, the method will not be calledNcZd}|}|jjt|||y)NT) self_instance)r@validate_pythonr )__dataclass_self__r,r-__tracebackhide__ss r.r/z$complete_dataclass..__init__s.     00D&1IYZ0[r0z .__init____get_pydantic_core_schema__F)from_dunder_get_core_schemaunpack)ref_mode`zall referenced typesztype[PydanticDataclass] dataclassc,j|||yr))validate_assignment)instance__field__value validators r.validated_setattrz-complete_dataclass..validated_setattrs  ) )(GW Er0T)rNr9r,rr-rr2r3)rYrrZstrr[r^r2r3)(hasattrwarningswarnDeprecationWarningrget_cls_types_namespacerGrrgenerate_dataclass_signaturer>r6r/ config_dictr: __signature__getattrr rgenerate_schemarrr4name core_config clean_schemaCollectedInvalidtypingcastr<rr5plugin_settingsr@r r?rXr __setattr____get__)rDconfig_wrapperrHrErC gen_schemasigr/get_core_schemaschemaerjr]r\s @r.complete_dataclassrxUs72s./ dfx '??Do.,S1LJ 'sC,C,C^ TC\ #//0 :HCL,88CCc#A4HO $,J66TYZ%F //QV/WF!,,S1K((0 ++/ 5C$*C -DS^^S%5%5{KQ_QoQo.C#36;"GC)) s  F  F,33D#> C '  C166(!}=   & &C/EFs+?AGH H(G>>H&H/.H/c|j}t|tr|j}|dk(rt}|j }|j }|j}|t|trt|r|}nt|tr t|r|}|j}|tur=|jturtjj}ntj }|j#|||S|S)zCustom processing where the parameter default is of type FieldInfo Args: param (Parameter): The parameter Returns: Parameter: The custom processed parameter r) annotationridefault)r{ isinstancerrzrrialiasvalidation_aliasr^r!r default_factoryinspectrempty dataclasses_HAS_DEFAULT_FACTORYreplace)param param_defaultrzrir}r~r{s r.process_param_defaultsrsMMM-+%%   Jzz##(99  # 5#(>CVW\C]D (# .3FGW3X#D '' ' ',,0AA!++11&::}} w}OO Lr0c<t|j||tS)zGenerate signature for a pydantic dataclass. Args: cls: The dataclass. fields: The model fields. config_wrapper: The config wrapper instance. Returns: The dataclass signature. )initrFrrpost_process_parameter)rr/r)rDrFrrs r.rdrds ' \\&`v r0c tj|xrGt|d xr8t|jj tt |diS)a>Returns True if a class is a stdlib dataclass and *not* a pydantic dataclass. We check that - `_cls` is a dataclass - `_cls` does not inherit from a processed pydantic dataclass (and thus have a `__pydantic_validator__`) - `_cls` does not have any annotations that are not dataclass fields e.g. ```py import dataclasses import pydantic.dataclasses @dataclasses.dataclass class A: x: int @pydantic.dataclasses.dataclass class B(A): y: int ``` In this case, when we first check `B`, we make an extra check and look at the annotations ('y'), which won't be a superset of all the dataclass fields (only the stdlib fields i.e. 'x') Args: cls: The class. Returns: `True` if the class is a stdlib dataclass, `False` otherwise. r@r7)r is_dataclassr_setr% issupersetrg)_clss r.is_builtin_dataclassrsZ>   & a67 7 a )) * 5 5c'$HY[]:^6_ `r0r))rDtype[StandardDataclass]rEdict[str, Any] | Noner2r3) rD type[Any]rrz_config.ConfigWrapperrHboolrErr2r)rrr2r)rDrrFzdict[str, FieldInfo]rrrr2r)rrr2z"TypeGuard[type[StandardDataclass]])=rA __future__r _annotationsrrrmr` functoolsrrrrrr r pydantic_corer r r rrtyping_extensionsrerrorsrrFrplugin._schema_validatorrrrrrr_fieldsr_generate_schemarr _genericsr _mock_val_serr_schema_generation_sharedr _utilsr! TYPE_CHECKINGconfigr"Protocolr$r9rbrGrxrrdrr*r0r.rs62 $(**(0>011"-I0.C' #FOO:-v:03 %" f f)f f + f  fR$N *>P]$"r0