a hHC @sddlZddlmZmZmZmZmZddlmZddl m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZddlmZmZmZe e gdfZedZeeeeedd d Zed d ee eegeefd dd Zd2d d eeeeeeee eegeeffddd Zeedddddfee eefee efee ge fefeeeeeee e e d ddZedddZ e!e e ddddZ"e!e ee dfddddZ#e!e ee dfdddd Z$e!e ee dfddd!d"Z%e!e ee dfddd#d$Z&e!e ee dfddd%d&Z'e!e ee dfddd'd(Z(e!e e)ddd)d*Z*eed+d,d-Z+e ed.d/d0Z,ee#ee$e-e%e.e&e/e'e0e(iZ1gd1Z2dS)3N) _MISSING_TYPEMISSINGFieldfieldfields)wraps) AnyCallableDictListLiteralOptionalTupleTypeTypeVarUnionget_args get_originoverload)#StrictDataclassClassValidationErrorStrictDataclassDefinitionError#StrictDataclassFieldValidationErrorTclsreturncCsdSNrrrW/var/www/html/assistant/venv/lib/python3.9/site-packages/huggingface_hub/dataclasses.pystrict sr!F accept_kwargs)r#rcCsdSrrr"rrr r!$s)rr#rcs0ttttdfdd }|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 ``` rcstdstdjdi}tD]x}g}|t||jd}|durt|t sb|g}|D]$}t |sftd|j d|dqf| ||||j <q(|_ jtttddfd d }|_r$jttdd fd d }|_jttdfdd }|_g} tD]f} | dsDq0t| } t| s\q0tt| jdkrtdjd| d| | q0| _tddfdd } d| _tdrtjddstdjd| _jtddfdd } 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.)selfnamevaluerc sh|j|gD]H}z ||WqttfyT}zt||d|WYd}~qd}~00q|||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__)kwargsrcsbddtDfdd|D}|fi||D]\}}|vr@|||q@dS)NcSsh|] }|jqSr)r().0frrr z9strict..wrap..__init__..csi|]\}}|vr||qSrrr3kvZdataclass_fieldsrr r6z:strict..wrap..__init__..)ritems __setattr__)r'r2Zstandard_kwargsr(r))r original_initr:r __init__s z&strict..wrap..__init__)rcsJ|}fdd|jD}d|}|rF|ddd|dS|S)Ncs*g|]"\}}|jvrd|d|qS)*=)r$r7rrr s z:strict..wrap..__repr__..z, ))__dict__r<join)r'Z standard_reprZadditional_kwargsZadditional_repr)r original_reprrr __repr__s   z&strict..wrap..__repr__Z 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.)r'rc sVjD]J}z ||WqttfyN}zt|j|d|WYd}~qd}~00qdS)z%Run class validators on the instance.)r%r*N)__class_validators__r-r.r__name__)r'r%r/rrr validates   z&strict..wrap..validateTrK"__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)rK)r'argsr2)r initial_initrr init_with_validatesz0strict..wrap..init_with_validater?)!hasattrrrJrappend_create_type_validatormetadatar, isinstancelist _is_validatorr(extendr+r=rstrr?rrHdir startswithgetattrcallableleninspect signature parametersrIrrLrKsetattr)rZfield_validatorsr4Z validatorsZcustom_validatorr%r1r?rHZclass_validatorsr(methodrKrOr")rrNr>rGr0r wrapbst                 zstrict..wrapN)rr)rr#rcrr"r r!(s: T) r%defaultdefault_factoryinitreprhashcomparerSr2rc KsBt|ts|g}|duri}||d<tf|||||||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%rdrerfrgrhrirS)rTrUr) r%rdrerfrgrhrirSr2rrr validated_fields  rkr%c sTttdddddftttfttgtftfttttttttdfdd }|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. TNrdrerfrgrhrirSr2c s tf|||||||d|S)Nrj)rkrmrlrr _inner$s  z"as_validated_field.._inner)rrrrr boolr r )r%rnrrlr as_validated_fields$  rp)r(r) expected_typercCsht|}t|}|turdSt|}r8||||n,t|trPt|||ntd|d|dS)z.Validate that 'value' matches 'expected_type'.NzUnsupported type for field 'r&) rrr_BASIC_TYPE_VALIDATORSr,rTtype_validate_simple_typer.)r(r)rqoriginrMr%rrr type_validator=s rv.)r(r)rMrc Csg}|D]L}zt|||WdStyR}z|t|WYd}~qd}~00qtd|dt|d|dd|dS)z8Validate that value matches one of the types in a Union.NField 'z ' with value z doesn't match any type in z . Errors: z; )rvr.rQrXrgrF)r(r)rMerrorstr/rrr _validate_unionLs &$rzcCs&||vr"td|d|d|dS)zValidate Literal type.rwz' expected one of , got N)r.)r(r)rMrrr _validate_literal[sr|c Cst|ts$td|dt|j|d}t|D]b\}}zt|d|d||Wq4ty}z$td|d|d|WYd }~q4d }~00q4d S) zValidate List[T] type.rwz' expected a list, got r[]Invalid item at index z in list ''N)rTrUr.rsrJ enumeratervr(r)rMZ item_typeiitemr/rrr _validate_listas rc Cst|ts$td|dt|j|\}}|D]n\}}z.t|d||t|d|d||Wq4ty}ztd|d|WYd}~q4d}~00q4dS) zValidate Dict[K, V] type.rwz' expected a dict, got z.keyr}r~zInvalid key or value in dict 'rN)rTdictr.rsrJr<rv)r(r)rMZkey_typeZ value_typer8r9r/rrr _validate_dictos rc Cs^t|ts$td|dt|jt|dkr|dturt|D]f\}}z t|d|d||dWqDty}z$td|d |d |WYd }~qDd }~00qDnt|t|krtd|d t|d t|nxtt ||D]h\}\}}zt|d|d||WqtyV}z$td|d |d |WYd }~qd }~00qd S)zValidate Tuple type.rwz' expected a tuple, got rr}r~rrz in tuple 'rNz' expected a tuple of length r{) rTtupler.rsrJr]Ellipsisrrvzip)r(r)rMrrr/expectedrrr _validate_tuple~s  2$rc Cst|ts$td|dt|j|d}t|D]V\}}zt|d||Wq4ty}ztd|d|WYd}~q4d}~00q4dS)zValidate Set[T] type.rwz' expected a set, got rz itemzInvalid item in set 'rN)rTsetr.rsrJrrvrrrr _validate_sets rc Cs<t||s8td|d|jdt|jdt|d dS)z&Validate simple type (int, str, etc.).rwz ' expected r{z (value: rDN)rTr.rJrsrg)r(r)rqrrr rts (rt)rrcstddfdd }|S)z-Create a type validator function for a field.N)r)rcstj|jdSr)rvr(rs)r)rrr r%sz)_create_type_validator..validator)r)rr%rrr rRsrR)r%rcCst|s dSt|}t|j}t|dkr4dS|djtjj tjj tjj fvrXdS|ddD]}|j tjj krddSqddS)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)r\r^r_rUr`valuesr]kind ParameterPOSITIONAL_OR_KEYWORDPOSITIONAL_ONLYVAR_POSITIONALrdempty)r%r_r`Z parameterrrr rVs   rV)r!rk Validator_Trrr)N)3r^ dataclassesrrrrr functoolsrtypingrr r r r r rrrrrrrrxrrrrrr!rorkrprXrvrzr|rrrrrsrtrRrVrUrrrrr__all__rrrr sp <*  K  +"