a y{Jh*W@sUdZddlmZddlZddlZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZddlmZddlmZddlmZmZmZd d lmZd d lmZmZmZm Z d d l!mZd d l"m#Z#ej$d krddlm%Z%n ddl m%Z%ej&Z'ej(fddie j)GdddZ*ej(fddie j)GdddZ+ej(fddie j)GdddZ,ej(fddie j)GdddZ-e rGddde%Z.Gddde%Z/Gddde%Z0ee/ej1e.ej2fZ3ee0ej4fZ5ee6e e e fe7e e fee fZ8de9d <ed!e3e8Z:ed"e5e8Z;ed#d#d$d%d%d&d'd(d)d*d+Zed6dd7Z?Gd8d9d9ej@e%e?ZAGd:d;d;e%e>ZBGdZCGd>d?d?e%ZDGd@dAdAe%ZEe e>ge>fZFe e>ejGge>fZHeeCe>eBe>fZIeeEeDfZJeeHe>eFe>fZKed-dBdCdDdEZLedFdGdCdHdEZLedIdJdCdKdEZLdLdMdCdNdEZLedOZMe reeMd#fZNn ej(fie j)GdPdQdQZNe reeMd#fZOn ej(fie j)GdRdSdSZOdS)TzBThis module contains related classes and functions for validation.) annotationsN) partialmethod) FunctionType) TYPE_CHECKINGAnyCallableTypeVarUnioncastoverload) core_schema) AnnotatedLiteral TypeAlias)GetCoreSchemaHandler)_core_metadata _decorators _generics_internal_dataclass)PydanticUserError) )ProtocolfrozenTc@s,eZdZUdZded<dddddd Zd S) AfterValidatora8Usage docs: https://docs.pydantic.dev/2.5/concepts/validators/#annotated-validators A metadata class that indicates that a validation should be applied **after** the inner validation logic. Attributes: func: The validator function. Example: ```py from typing_extensions import Annotated from pydantic import AfterValidator, BaseModel, ValidationError MyInt = Annotated[int, AfterValidator(lambda v: v + 1)] class Model(BaseModel): a: MyInt print(Model(a=1).a) #> 2 try: Model(a='a') except ValidationError as e: print(e.json(indent=2)) ''' [ { "type": "int_parsing", "loc": [ "a" ], "msg": "Input should be a valid integer, unable to parse string as an integer", "input": "a", "url": "https://errors.pydantic.dev/2/v/int_parsing" } ] ''' ``` Kcore_schema.NoInfoValidatorFunction | core_schema.WithInfoValidatorFunctionfuncr_GetCoreSchemaHandlercore_schema.CoreSchema source_typehandlerreturncCsX||}t|jd}|r8ttj|j}tj|||jdSttj|j}tj||dSdS)Nafterschema field_namer&) _inspect_validatorrr r WithInfoValidatorFunctionZ"with_info_after_validator_functionr'NoInfoValidatorFunctionZ no_info_after_validator_functionselfr!r"r&info_argrr/P/var/www/shaz/venv/lib/python3.9/site-packages/pydantic/functional_validators.py__get_pydantic_core_schema__Is z+AfterValidator.__get_pydantic_core_schema__N__name__ __module__ __qualname____doc____annotations__r1r/r/r/r0rs )rc@s,eZdZUdZded<dddddd Zd S) BeforeValidatoraUsage docs: https://docs.pydantic.dev/2.5/concepts/validators/#annotated-validators A metadata class that indicates that a validation should be applied **before** the inner validation logic. Attributes: func: The validator function. Example: ```py from typing_extensions import Annotated from pydantic import BaseModel, BeforeValidator MyInt = Annotated[int, BeforeValidator(lambda v: v + 1)] class Model(BaseModel): a: MyInt print(Model(a=1).a) #> 2 try: Model(a='a') except TypeError as e: print(e) #> can only concatenate str (not "int") to str ``` rrrrrr cCsX||}t|jd}|r8ttj|j}tj|||jdSttj|j}tj||dSdS)Nbeforer%r() r)rr r r*Z#with_info_before_validator_functionr'r+Z!no_info_before_validator_functionr,r/r/r0r1us z,BeforeValidator.__get_pydantic_core_schema__Nr2r/r/r/r0r8Ts r8c@s,eZdZUdZded<dddddd Zd S) PlainValidatora;Usage docs: https://docs.pydantic.dev/2.5/concepts/validators/#annotated-validators A metadata class that indicates that a validation should be applied **instead** of the inner validation logic. Attributes: func: The validator function. Example: ```py from typing_extensions import Annotated from pydantic import BaseModel, PlainValidator MyInt = Annotated[int, PlainValidator(lambda v: int(v) + 1)] class Model(BaseModel): a: MyInt print(Model(a='1').a) #> 2 ``` rrrrrr cCsJt|jd}|r.ttj|j}tj||jdSttj|j}t|SdS)Nplain)r') r)rr r r*Z"with_info_plain_validator_functionr'r+Z no_info_plain_validator_function)r-r!r"r.rr/r/r0r1s  z+PlainValidator.__get_pydantic_core_schema__Nr2r/r/r/r0r:s r:c@s,eZdZUdZded<dddddd Zd S) WrapValidatoraUsage docs: https://docs.pydantic.dev/2.5/concepts/validators/#annotated-validators A metadata class that indicates that a validation should be applied **around** the inner validation logic. Attributes: func: The validator function. ```py from datetime import datetime from typing_extensions import Annotated from pydantic import BaseModel, ValidationError, WrapValidator def validate_timestamp(v, handler): if v == 'now': # we don't want to bother with further validation, just return the new value return datetime.now() try: return handler(v) except ValidationError: # validation failed, in this case we want to return a default value return datetime(2000, 1, 1) MyTimestamp = Annotated[datetime, WrapValidator(validate_timestamp)] class Model(BaseModel): a: MyTimestamp print(Model(a='now').a) #> 2032-01-02 03:04:05.000006 print(Model(a='invalid').a) #> 2000-01-01 00:00:00 ``` zScore_schema.NoInfoWrapValidatorFunction | core_schema.WithInfoWrapValidatorFunctionrrrrr cCsX||}t|jd}|r8ttj|j}tj|||jdSttj|j}tj||dSdS)Nwrapr%r() r)rr r WithInfoWrapValidatorFunctionZ!with_info_wrap_validator_functionr'ZNoInfoWrapValidatorFunctionZno_info_wrap_validator_functionr,r/r/r0r1s z*WrapValidator.__get_pydantic_core_schema__Nr2r/r/r/r0r<s $r<c@seZdZddddddZdS)_OnlyValueValidatorClsMethodr)!_OnlyValueValidatorClsMethod__cls#_OnlyValueValidatorClsMethod__valuer#cCsdSNr/)r-r@rAr/r/r0__call__sz%_OnlyValueValidatorClsMethod.__call__Nr3r4r5rCr/r/r/r0r?sr?c@s eZdZdddddddZdS)_V2ValidatorClsMethodr_core_schema.ValidationInfo)_V2ValidatorClsMethod__cls"_V2ValidatorClsMethod__input_value_V2ValidatorClsMethod__infor#cCsdSrBr/)r-rGrHrIr/r/r0rCsz_V2ValidatorClsMethod.__call__NrDr/r/r/r0rEsrEc@s"eZdZddddddddZdS)_V2WrapValidatorClsMethodrz)_core_schema.ValidatorFunctionWrapHandlerrF)_V2WrapValidatorClsMethod__cls&_V2WrapValidatorClsMethod__input_value$_V2WrapValidatorClsMethod__validator_V2WrapValidatorClsMethod__infor#cCsdSrBr/)r-rKrLrMrNr/r/r0rCsz"_V2WrapValidatorClsMethod.__call__NrDr/r/r/r0rJsrJr_PartialClsOrStaticMethod"_V2BeforeAfterOrPlainValidatorType_V2WrapValidatorType.)mode check_fieldsstrz#Literal['before', 'after', 'plain']z bool | NonezRCallable[[_V2BeforeAfterOrPlainValidatorType], _V2BeforeAfterOrPlainValidatorType])__fieldfieldsrRrSr#cGsdSrBr/rUrRrSrVr/r/r0field_validatorsrX)rSzLiteral['wrap']z6Callable[[_V2WrapValidatorType], _V2WrapValidatorType]cGsdSrBr/rWr/r/r0rX s)r9r$r=r;FieldValidatorModesr$zCallable[[Any], Any]cs\t|trtddd|gRtddDs@tddddd d fd d }|S) a<Usage docs: https://docs.pydantic.dev/2.5/concepts/validators/#field-validators Decorate methods on the class indicating that they should be used to validate fields. Example usage: ```py from typing import Any from pydantic import ( BaseModel, ValidationError, field_validator, ) class Model(BaseModel): a: str @field_validator('a') @classmethod def ensure_foobar(cls, v: Any): if 'foobar' not in v: raise ValueError('"foobar" not found in a') return v print(repr(Model(a='this is foobar good'))) #> Model(a='this is foobar good') try: Model(a='snap') except ValidationError as exc_info: print(exc_info) ''' 1 validation error for Model a Value error, "foobar" not found in a [type=value_error, input_value='snap', input_type=str] ''' ``` For more in depth examples, see [Field Validators](../concepts/validators.md#field-validators). Args: __field: The first field the `field_validator` should be called on; this is separate from `fields` to ensure an error is raised if you don't pass at least one. *fields: Additional field(s) the `field_validator` should be called on. mode: Specifies whether to validate the fields before or after validation. check_fields: Whether to check that the fields actually exist on the model. Returns: A decorator that can be used to decorate a function to be used as a field_validator. Raises: PydanticUserError: - If `@field_validator` is used bare (with no fields). - If the args passed to `@field_validator` as fields are not strings. - If `@field_validator` applied to instance methods. z`@field_validator` should be used with fields and keyword arguments, not bare. E.g. usage should be `@validator('', ...)`zvalidator-no-fieldscodecss|]}t|tVqdSrB) isinstancerT).0fieldr/r/r0 ^z"field_validator..z`@field_validator` fields should be passed as separate string args. E.g. usage should be `@validator('', '', ...)`zvalidator-invalid-fieldszHCallable[..., Any] | staticmethod[Any, Any] | classmethod[Any, Any, Any](_decorators.PydanticDescriptorProxy[Any]fr#cs<t|rtdddt|}tjd}t||S)Nz8`@field_validator` cannot be applied to instance methodszvalidator-instance-methodrZ)rVrRrS)rZis_instance_method_from_sigr%ensure_classmethod_based_on_signatureZFieldValidatorDecoratorInfoPydanticDescriptorProxyrcZdec_inforSrVrRr/r0deces  zfield_validator..dec)r\rrall)rUrRrSrVrhr/rgr0rXs>   _ModelType _ModelTypeCo) covariantc@s$eZdZdZd ddddddZdS) ModelWrapValidatorHandlerz[@model_validator decorated function handler argument type. This is used when `mode='wrap'`.Nrzstr | int | Nonerk) input_valueouter_locationr#cCsdSrBr/)r-rnror/r/r0rC}sz"ModelWrapValidatorHandler.__call__)Nr3r4r5r6rCr/r/r/r0rmzsrmc@s$eZdZdZdddddddZd S) ModelWrapValidatorWithoutInfozA @model_validator decorated function signature. This is used when `mode='wrap'` and the function does not have info argument. type[_ModelType]r%ModelWrapValidatorHandler[_ModelType]rj)cls%_ModelWrapValidatorWithoutInfo__value'_ModelWrapValidatorWithoutInfo__handlerr#cCsdSrBr/)r-rtrurvr/r/r0rCs z&ModelWrapValidatorWithoutInfo.__call__Nrpr/r/r/r0rqsrqc@s&eZdZdZdddddddd Zd S) ModelWrapValidatorzQA @model_validator decorated function signature. This is used when `mode='wrap'`.rrrrsrFrj)rt_ModelWrapValidator__value_ModelWrapValidator__handler_ModelWrapValidator__infor#cCsdSrBr/)r-rtrxryrzr/r/r0rCs zModelWrapValidator.__call__Nrpr/r/r/r0rwsrwc@s"eZdZdZddddddZdS)ModelBeforeValidatorWithoutInfozA @model_validator decorated function signature. This is used when `mode='before'` and the function does not have info argument. r)rt'_ModelBeforeValidatorWithoutInfo__valuer#cCsdSrBr/)r-rtr|r/r/r0rCsz(ModelBeforeValidatorWithoutInfo.__call__Nrpr/r/r/r0r{sr{c@s$eZdZdZdddddddZdS)ModelBeforeValidatorzUA `@model_validator` decorated function signature. This is used when `mode='before'`.rrF)rt_ModelBeforeValidator__value_ModelBeforeValidator__infor#cCsdSrBr/)r-rtr~rr/r/r0rCs zModelBeforeValidator.__call__Nrpr/r/r/r0r}sr}z|Callable[[_AnyModelWrapValidator[_ModelType]], _decorators.PydanticDescriptorProxy[_decorators.ModelValidatorDecoratorInfo]])rRr#cCsdSrBr/rRr/r/r0model_validatorsrzLiteral['before']zqCallable[[_AnyModeBeforeValidator], _decorators.PydanticDescriptorProxy[_decorators.ModelValidatorDecoratorInfo]]cCsdSrBr/rr/r/r0rszLiteral['after']z}Callable[[_AnyModelAfterValidator[_ModelType]], _decorators.PydanticDescriptorProxy[_decorators.ModelValidatorDecoratorInfo]]cCsdSrBr/rr/r/r0rsz"Literal['wrap', 'before', 'after']rcsdddfdd }|S)aUsage docs: https://docs.pydantic.dev/2.5/concepts/validators/#model-validators Decorate model methods for validation purposes. Example usage: ```py from typing import Optional from pydantic import BaseModel, ValidationError, model_validator class Square(BaseModel): width: float height: float @model_validator(mode='after') def verify_square(self) -> 'Rectangle': if self.width != self.height: raise ValueError('width and height do not match') return self s = Square(width=1, height=1) print(repr(s)) #> Square(width=1.0, height=1.0) try: Square(width=1, height=2) except ValidationError as e: print(e) ''' 1 validation error for Square __root__ width and height do not match (type=value_error) ''' ``` For more in depth examples, see [Model Validators](../concepts/validators.md#model-validators). Args: mode: A required string literal that specifies the validation mode. It can be one of the following: 'wrap', 'before', or 'after'. Returns: A decorator that can be used to decorate a function to be used as a model validator. rrarbcs"t|}tjd}t||S)Nr)rrdZModelValidatorDecoratorInforerfrr/r0rhs  zmodel_validator..decr/)rRrhr/rr0rs1AnyTypec@s@eZdZdZedddddZedddd d d ZejZd S) InstanceOfuGeneric type for annotating a type that is an instance of a given class. Example: ```py from pydantic import BaseModel, InstanceOf class Foo: ... class Bar(BaseModel): foo: InstanceOf[Foo] Bar(foo=Foo()) try: Bar(foo=42) except ValidationError as e: print(e) """ [ │ { │ │ 'type': 'is_instance_of', │ │ 'loc': ('foo',), │ │ 'msg': 'Input should be an instance of Foo', │ │ 'input': 42, │ │ 'ctx': {'class': 'Foo'}, │ │ 'url': 'https://errors.pydantic.dev/0.38.0/v/is_instance_of' │ } ] """ ``` ritemr#cCst||fSrB)r rtrr/r/r0__class_getitem__QszInstanceOf.__class_getitem__rrrsourcer"r#cCslddlm}tt|p|}z ||}Wn|yB|YS0tjdd|d|d<tj||dSdS)Nr)PydanticSchemaGenerationErrorcSs||SrBr/vhr/r/r0er`z9InstanceOf.__get_pydantic_core_schema__..functionr& serialization)Z python_schemaZ json_schema)Zpydanticrr Zis_instance_schemar get_origin#wrap_serializer_function_ser_schemaZjson_or_python_schema)rtrr"rZinstance_of_schemaoriginal_schemar/r/r0r1Us     z'InstanceOf.__get_pydantic_core_schema__N) r3r4r5r6 classmethodrr1object__hash__r/r/r/r0r/s  rc@s<eZdZdZdddddZeddddd d ZejZd S) SkipValidationaIf this is applied as an annotation (e.g., via `x: Annotated[int, SkipValidation]`), validation will be skipped. You can also use `SkipValidation[int]` as a shorthand for `Annotated[int, SkipValidation]`. This can be useful if you want to use a type annotation for documentation/IDE/type-checking purposes, and know that it is safe to skip validation for one or more of the fields. Because this converts the validation schema to `any_schema`, subsequent annotation-applied transformations may not have the expected effects. Therefore, when used, this annotation should generally be the final annotation applied to a type. rrcCst|tfSrB)r rrr/r/r0r}sz SkipValidation.__class_getitem__rrrcs:||tjfddgd}tj|tjddddS)Ncs|SrBr/)Z_crrr/r0rr`z=SkipValidation.__get_pydantic_core_schema__..)Zjs_annotation_functionscSs||SrBr/rr/r/r0rr`r)metadatar)rZbuild_metadata_dictr Z any_schemar)rtrr"rr/rr0r1sz+SkipValidation.__get_pydantic_core_schema__N) r3r4r5r6rrr1rrr/r/r/r0rps   r)Pr6 __future__rZ _annotations dataclassessys functoolsrtypesrtypingrrrrr r r Z pydantic_corer Z _core_schematyping_extensionsr rrrr _internalrrrrZannotated_handlerserrorsr version_inforZinspect_validatorr) dataclassZ slots_truerr8r:r<r?rErJr*r+Z _V2Validatorr>Z_V2WrapValidatorr staticmethodrOr7rPrQrXrYrjrkZValidatorFunctionWrapHandlerrmrqrwr{r}ZModelAfterValidatorWithoutInfoZValidationInfoZModelAfterValidatorZ_AnyModelWrapValidatorZ_AnyModeBeforeValidatorZ_AnyModelAfterValidatorrrrrr/r/r/r0s   $       7+$2 ,   ]     :<