o i6W@sddlZddlmZmZmZmZmZmZddlm Z m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZmZz ddl mZmZWn)eykz ddlmZmZWneyheddiZeddiZYnwYnwd d lmZmZm Z ee gdfZ!ed Z"ed e#e$e fd Z%e&Z'edee"dee"fddZ(eddde)deee"gee"ffddZ( dKdddeee"de)deee"eee"gee"fffddZ(dee%de#ddfddZ*e dee%defddZ+dee%de#e$e ffddZ,eedddddfd ee-e!e!fd!ee efd"eege fefd#e)d$e)d%ee)d&e)d'ee#d(e de fd)d*Z.d e!fd+d,Z/d-e$d.e d/e ddfd0d1Z0d-e$d.e d2e1e d3fddfd4d5Z2d-e$d.e d2e1e d3fddfd6d7Z3d-e$d.e d2e1e d3fddfd8d9Z4d-e$d.e d2e1e d3fddfd:d;Z5d-e$d.e d2e1e d3fddfdd?Z7d-e$d.e d/eddfd@dAZ8dBede!fdCdDZ9d e de)fdEdFZ:dGe de)fdHdIZ;ee2ee3e-e4e#e5e1e6edS)LN) _MISSING_TYPEMISSINGFieldfieldfieldsmake_dataclass) lru_cachewraps) AnnotatedAnyCallable ForwardRefLiteralOptionalTypeTypeVarUnionget_args get_originoverload) NotRequiredRequiredrr)#StrictDataclassClassValidationErrorStrictDataclassDefinitionError#StrictDataclassFieldValidationErrorT TypedDictType)boundclsreturncCdSNrr rra/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/huggingface_hub/dataclasses.pystrict/r&F accept_kwargsr)cCr"r#rr(rrr%r&3r'cs2dttdttffdd }|dur||S|S)a Decorator to add strict validation to a dataclass. This decorator must be used on top of `@dataclass` to ensure IDEs and static typing tools recognize the class as a dataclass. Can be used with or without arguments: - `@strict` - `@strict(accept_kwargs=True)` Args: cls: The class to convert to a strict dataclass. accept_kwargs (`bool`, *optional*): If True, allows arbitrary keyword arguments in `__init__`. Defaults to False. Returns: The enhanced dataclass with strict validation on field assignment. Example: ```py >>> from dataclasses import dataclass >>> from huggingface_hub.dataclasses import as_validated_field, strict, validated_field >>> @as_validated_field >>> def positive_int(value: int): ... if not value >= 0: ... raise ValueError(f"Value must be positive, got {value}") >>> @strict(accept_kwargs=True) ... @dataclass ... class User: ... name: str ... age: int = positive_int(default=10) # Initialize >>> User(name="John") User(name='John', age=10) # Extra kwargs are accepted >>> User(name="John", age=30, lastname="Doe") User(name='John', age=30, *lastname='Doe') # Invalid type => raises >>> User(name="John", age="30") huggingface_hub.errors.StrictDataclassFieldValidationError: Validation error for field 'age': TypeError: Field 'age' expected int, got str (value: '30') # Invalid value => raises >>> User(name="John", age=-1) huggingface_hub.errors.StrictDataclassFieldValidationError: Validation error for field 'age': ValueError: Value must be positive, got -1 ``` r r!cstdstdjdi}tD]<}g}|t||jd}|durKt|t s1|g}|D]}t |sEtd|j d|dq3| ||||j <q|_ jdtd td td dffd d }|_rjtdtd dffdd }|_jtd tffdd }|_g} tD]-} | dsqt| } t| sqtt| jdkrtdjd| d| | q| _dtd dffdd } d| _tdrtjddstdjd| _jtd!fdd } t d | S)"N__dataclass_fields__zClass 'z.' must be a dataclass before applying @strict. validatorzInvalid validator for field '': z.. Must be a callable taking a single argument.selfnamevaluer!c sZ|j|gD]}z||Wqttfy$}zt||d|d}~ww|||dS)z1Custom __setattr__ method for strict dataclasses.)rcauseN)__validators__get ValueError TypeErrorr)r-r.r/r+e)original_setattrrr%__strict_setattr__s z0strict..wrap..__strict_setattr__kwargscsbddtDfdd|D}|fi||D]\}}|vr.|||q dS)NcSsh|]}|jqSr)r.).0frrr% sz9strict..wrap..__init__..csi|] \}}|vr||qSrrr9kvdataclass_fieldsrr% sz:strict..wrap..__init__..)ritems __setattr__)r-r8standard_kwargsr.r/)r original_initr?r%__init__s z&strict..wrap..__init__csJ|}fdd|jD}d|}|r#|ddd|dS|S)Ncs*g|]\}}|jvrd|d|qS)*=)r*r<r$rr% s  z:strict..wrap..__repr__..z, ))__dict__rBjoin)r- standard_repradditional_kwargsadditional_repr)r original_reprrr%__repr__s    z&strict..wrap..__repr__ validate_rz' has a class validator 'z' that takes more than one argument. Class validators must take only 'self' as an argument. Methods starting with 'validate_' are considered to be class validators.c sHjD]}z||Wqttfy!}zt|j|d|d}~wwdS)z%Run class validators on the instance.)r+r0N)__class_validators__r3r4r__name__)r-r+r5r$rr%validates  z&strict..wrap..validateTrV"__is_defined_by_strict_decorator__Fz' already implements a method called 'validate'. This method name is reserved when using the @strict decorator on a dataclass. If you want to keep your own method, please rename it.cs$|g|Ri||dS)z*Run class validators after initialization.N)rV)r-argsr8)r initial_initrr%init_with_validatesz0strict..wrap..init_with_validaterF)r!N)!hasattrrrUrappend_create_type_validatormetadatar2 isinstancelist _is_validatorr.extendr1rCr strrFr rRdir startswithgetattrcallableleninspect signature parametersrTrrWrVsetattr)r field_validatorsr: validatorscustom_validatorr+r7rFrRclass_validatorsr.methodrVrZr()r rYrErQr6r%wrapqsv                zstrict..wrapN)rr)r r)rrrr(r%r&7s: schemadatacCst|}|di|dS)a Validate that a dictionary conforms to the types defined in a TypedDict class. Under the hood, the typed dict is converted to a strict dataclass and validated using the `@strict` decorator. Args: schema (`type[TypedDictType]`): The TypedDict class defining the expected structure and types. data (`dict`): The dictionary to validate. Raises: `StrictDataclassFieldValidationError`: If any field in the dictionary does not conform to the expected type. Example: ```py >>> from typing import Annotated, TypedDict >>> from huggingface_hub.dataclasses import validate_typed_dict >>> def positive_int(value: int): ... if not value >= 0: ... raise ValueError(f"Value must be positive, got {value}") >>> class User(TypedDict): ... name: str ... age: Annotated[int, positive_int] >>> # Valid data >>> validate_typed_dict(User, {"name": "John", "age": 30}) >>> # Invalid type for age >>> validate_typed_dict(User, {"name": "John", "age": "30"}) huggingface_hub.errors.StrictDataclassFieldValidationError: Validation error for field 'age': TypeError: Field 'age' expected int, got str (value: '30') >>> # Invalid value for age >>> validate_typed_dict(User, {"name": "John", "age": -1}) huggingface_hub.errors.StrictDataclassFieldValidationError: Validation error for field 'age': ValueError: Value must be positive, got -1 ``` Nr)!_build_strict_cls_from_typed_dict)rsrt strict_clsrrr%validate_typed_dicts,rwc Cst|}t|ddsA|D]2\}}t|}|tur6t|^}}t|s(t|}tt|gt |||<qt|s@t|||<qg}|D],\}}t|turht|^}}| ||t t d|didfqG| ||t t dfqGt t|j|S)N __total__Tr+r)defaultr^)ry)_get_typed_dict_annotationsrfrBrr r_is_required_or_notrequiredrtupler`r\r_TYPED_DICT_DEFAULT_VALUEr&rrU)rs type_hintskeyr/originbasemetarrrr%ru1s&     "rucCsDz ddl}||WSty!dd|jdiDYSw)z0Extract type annotations from a TypedDict class.rNcSs&i|]\}}||dur |ntdqSr#)type)r9r.r/rrr%rAXsz/_get_typed_dict_annotations..__annotations__) annotationlibget_annotations ImportErrorrLr2rB)rsrrrr%rzPs   rzTr+rydefault_factoryinitreprhashcomparer^r8c KsBt|ts|g}|duri}||d<td|||||||d|S)a< Create a dataclass field with a custom validator. Useful to apply several checks to a field. If only applying one rule, check out the [`as_validated_field`] decorator. Args: validator (`Callable` or `list[Callable]`): A method that takes a value as input and raises ValueError/TypeError if the value is invalid. Can be a list of validators to apply multiple checks. **kwargs: Additional arguments to pass to `dataclasses.field()`. Returns: A field with the validator attached in metadata Nr+ryrrrrrr^r)r_r`r) r+ryrrrrrr^r8rrr%validated_field`s rcsbttdddddfdtttfdttgtftfdtdtdttdtd ttd tffd d }|S) a Decorates a validator function as a [`validated_field`] (i.e. a dataclass field with a custom validator). Args: validator (`Callable`): A method that takes a value as input and raises ValueError/TypeError if the value is invalid. TNryrrrrrr^r8c s tf|||||||d|S)Nr)r)ryrrrrrr^r8r+rr%_inners  z"as_validated_field.._inner)rrr rr boolrdict)r+rrrr%as_validated_fields2  rr.r/ expected_typecCst|}t|}|turdSt|}r||||dSt|tr*t|||dSt|ts4t|t r6dS|t urP|t urFt d|dt |||ddS|turd|t urZdSt |||ddSt d|d|)z.Validate that 'value' matches 'expected_type'.NField 'z' is required but missing.rzUnsupported type for field 'r,)rrr _BASIC_TYPE_VALIDATORSr2r_r_validate_simple_typer rcrr}r4type_validatorr)r.r/rrrXr+rrr%rs& rrX.c Cs~g}|D]%}z t|||WdSty)}z |t|WYd}~qd}~wwtd|dt|d|dd|)z8Validate that value matches one of the types in a Union.Nrz ' with value z doesn't match any type in z . Errors: z; )rr4r\rcrrM)r.r/rXerrorstr5rrr%_validate_unions $rcCs&||vrtd|d|d|dS)zValidate Literal type.rz' expected one of , got N)r4)r.r/rXrrr%_validate_literalsrc Cst|tstd|dt|j|d}t|D]*\}}zt|d|d||WqtyD}z td|d|d|d }~wwd S) zValidate list[T] type.rz' expected a list, got r[]Invalid item at index z in list ''N)r_r`r4rrU enumeraterr.r/rX item_typeiitemr5rrr%_validate_lists rc Cst|tstd|dt|j|\}}|D]0\}}zt|d||t|d|d||WqtyJ}z td|d|d}~wwdS) zValidate dict[K, V] type.rz' expected a dict, got z.keyrrzInvalid key or value in dict 'rN)r_rr4rrUrBr)r.r/rXkey_type value_typer=r>r5rrr%_validate_dicts rc Cs@t|tstd|dt|jt|dkrQ|dturQt|D],\}}zt|d|d||dWq"tyN}z td|d |d |d }~wwd St|t|krjtd|d t|d t|tt ||D],\}\}}zt|d|d||Wqqty}z td|d |d |d }~wwd S)zValidate Tuple type.rz' expected a tuple, got rrrrrz in tuple 'rNz' expected a tuple of length r) r_r|r4rrUrhEllipsisrrzip)r.r/rXrrr5expectedrrr%_validate_tuples*  "rc Cst|tstd|dt|j|d}t|D]$\}}z t|d||Wqty>}z td|d|d}~wwdS)zValidate set[T] type.rz' expected a set, got rz itemzInvalid item in set 'rN)r_setr4rrUrrrrrr% _validate_sets rc Cs<t||std|d|jdt|jdt|d dS)z&Validate simple type (int, str, etc.).rz ' expected rz (value: rKN)r_r4rUrr)r.r/rrrr%rs (rrcsdtddffdd }|S)z-Create a type validator function for a field.r/r!Ncstj|jdSr#)rr.r)r/rrr%r+)sz)_create_type_validator..validator)r )rr+rrr%r]%sr]cCst|sdSt|}t|j}t|dkrdS|djtjj tjj tjj fvr,dS|ddD] }|j tjj kr>dSq2dS)zCheck if a function is a validator. A validator is a Callable that can be called with a single positional argument. The validator can have more arguments with default values. Basically, returns True if `validator(value)` is possible. FrrNT)rgrirjr`rkvaluesrhkind ParameterPOSITIONAL_OR_KEYWORDPOSITIONAL_ONLYVAR_POSITIONALryempty)r+rjrk parameterrrr%ra/s"  ra type_hintcCs|ttfvp t|ttfvS)z2Helper to check if a type is Required/NotRequired.)rrr)rrrr%r{Jsr{)r&rwr Validator_Trrrr#)?ri dataclassesrrrrrr functoolsrr typingr r r r rrrrrrrrrrrtyping_extensionsrrrrrrrrrcrobjectr}r&rrwrurzr`rrrr|rrrrrrrr]rar{rr__all__rrrr%s 8    ,   I2   +"""""""