zi)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/m0Z0ej1r.d dl2m3Z3Gddej4Z5Gdde5ej4Z6neZ7d6d7d"Z8d#d$d8d*Z9d9d-Z:d:d2Z;d;d5Z)rBrDrGfieldss r1set_dataclass_fieldsrIHs3-S11L %c? V V VF$Cr3T) raise_errors type[Any]config_wrapper_config.ConfigWrapperrJboolc ~ t|drtjdt|t j|}t ||t|}t|||}t||j |}dd }|j d |_ ||_ |j |_||_t!|d d} |r1||t#t%|jd|d} n||d} n:#t($r-} |rt+||jd| jdYd} ~ dSd} ~ wwxYw||} || } n'#|j$rt+||jdYdSwxYwt7jd|}| |_t=| ||j|j d| |j x|_! tE| | |_#|j$r;tK|j&d fd } | 'd||_&dS)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 calledN__dataclass_self__r9r(rr*r+r,cbd}|}|jt|||dS)NT) self_instance)r@validate_pythonr )rQr(r*__tracebackhide__ss r1r2z$complete_dataclass..__init__s:    00D&1I1IYZ0[[[[[r3z .__init____get_pydantic_core_schema__F)from_dunder_get_core_schemaunpack)ref_mode`zall referenced typesztype[PydanticDataclass] dataclassinstance__fieldstr__valuec6|||dSr.)validate_assignment)r]r^r` validators r1validated_setattrz-complete_dataclass..validated_setattrs!  ) )(GW E E E E Er3T)rQr9r(rr*rr+r,)r]rr^r_r`r_r+r,)(hasattrwarningswarnDeprecationWarningrget_cls_types_namespacerIrrgenerate_dataclass_signaturer>r6r2 config_dictr: __signature__getattrr rgenerate_schemarrr4name core_config clean_schemaCollectedInvalidtypingcastr<rr5plugin_settingsr@r r?rbr __setattr____get__)rBrLrJrDrG gen_schemasigr2get_core_schemaschemaerprdrcs @r1complete_dataclassr}Us2s.//  dfx   '?DDo...,S11LJ 'sC,C^ T TC\\\\ #/:::HCL,8CCc#A4HHO  X$_,J6TYZZZ%FF //QV/WWF &   C}16}}}===uuuuu  !,,S11K((00  &C/EFFFuu +/ 5 5C$*C -DS^S%5{KQ_Qo..C#36;"G"GC)? s   F F F F F  F,33D#>> 4s+A D E "EE "E88 FFparamrc|j}t|tr|j}|dkrt}|j}|j}|j}|'t|trt|r|}n&t|trt|r|}|j}|tur,|j turtj j}n tj}||||S|S)zCustom processing where the parameter default is of type FieldInfo Args: param (Parameter): The parameter Returns: Parameter: The custom processed parameter rN) annotationrodefault)r isinstancerrrroaliasvalidation_aliasr_r!r default_factoryinspectrempty dataclasses_HAS_DEFAULT_FACTORYreplace)r~ param_defaultrrorrrs r1process_param_defaultsrsMM-++P%    Jz#(9  # 5#(>(> #CVW\C]C] #DD (# . . $3FGW3X3X $#D ' ' ' ',0AAA!+1&:}} w}OOO Lr3rHdict[str, FieldInfo]rrc<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. )initrHrLpost_process_parameter)rr2r)rBrHrLs r1rjrjs' ' \&`v   r3_cls"TypeGuard[type[StandardDataclass]]c tj|oSt|d oBt|jtt |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_dataclassresetr% issupersetrm)rs r1is_builtin_dataclassrsf>  && a677 7 a ) * * 5 5c'$HY[]:^:^6_6_ ` `r3r.)rBrCrDrEr+r,) rBrKrLrMrJrNrDrEr+rN)r~rr+r)rBrCrHrrLrr+r)rrKr+r)=rA __future__r _annotationsrrrsrf functoolsrrrrrr r pydantic_corer r r rrtyping_extensionsrerrorsrrHrplugin._schema_validatorrrrrrr_fieldsr_generate_schemarr _genericsr _mock_val_serr_schema_generation_sharedr _utilsr! TYPE_CHECKINGconfigr"Protocolr$r9rhrIr}rrjrr/r3r1rs66222222 $$$$$$$$((((((((**********('''''000000>>>>>>0000001111111111""""""------IIIIIIII000000......CCCCCC'''''' #3######FO:::::-v::::03 % % % % %" ffffffR$$$$N$""""""r3