U diC @sddlZddlmZmZmZmZmZddlmZddl m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZddlmZmZmZe e gdfZedZeeeeedd d Zed d ee eegeefd dd Zd5d d eeeeeeee eegeeffddd Zeedddddfeeeefee efee ge fefeeeeeee e e d ddZ edddZ!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'e0e(e1e)iZ2d dd1d2d3d4gZ3dS)6N) _MISSING_TYPEMISSINGFieldfieldfields)wraps)AnyCallableDict ForwardRefListLiteralOptionalTupleTypeTypeVarUnionget_args get_originoverload)#StrictDataclassClassValidationErrorStrictDataclassDefinitionError#StrictDataclassFieldValidationErrorTclsreturncCsdSNrrr?/tmp/pip-unpacked-wheel-ohruig0i/huggingface_hub/dataclasses.pystrict!sr"F accept_kwargs)r$rcCsdSrrr#rrr!r"%s)rr$rcs0ttttdfdd }|dk r,||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}|dk rt|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 sf|j|gD]F}z ||Wqttfk rR}zt||d|W5d}~XYqXq|||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__)kwargsrcs^ddtDfdd|D}|f||D]\}}|kr<|||qsz9strict..wrap..__init__..csi|]\}}|kr||qSrrr4kvZdataclass_fieldsrr! sz:strict..wrap..__init__..)ritems __setattr__)r(r3Zstandard_kwargsr)r*)r original_initr:r!__init__s  z&strict..wrap..__init__)rcsJ|}fdd|jD}d|}|rF|ddd|dS|S)Ncs*g|]"\}}|jkrd|d|qS)*=)r%r7r rr! 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 sTjD]H}z ||Wqttfk rL}zt|j|d|W5d}~XYqXqdS)z%Run class validators on the instance.)r&r+N)__class_validators__r.r/r__name__)r(r&r0r rr!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|f|||dS)z*Run class validators after initialization.N)rK)r(argsr3)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_validatorsr5Z validatorsZcustom_validatorr&r2r?rHZclass_validatorsr)methodrKrOr#)rrNr>rGr1r!wrapcst                 zstrict..wrapN)rr)rr$rcrr#r!r")s: T) r&defaultdefault_factoryinitreprhashcomparerSr3rc KsBt|ts|g}|dkri}||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&rdrerfrgrhrirSr3rrr!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. TNrdrerfrgrhrirSr3c s tf|||||||d|S)Nrj)rkrmrlrr!_inner%s  z"as_validated_field.._inner)rrrrr boolrr )r&rnrrlr!as_validated_fields$  rp)r)r* expected_typercCst|}t|}|tkrdSt|}r8||||nDt|trPt|||n,t|tsdt|t rhdSt d|d|dS)z.Validate that 'value' matches 'expected_type'.NzUnsupported type for field 'r') rrr_BASIC_TYPE_VALIDATORSr-rTtype_validate_simple_typer rXr/)r)r*rqoriginrMr&rrr!type_validator>s rv.)r)r*rMrc Csg}|D]J}zt|||WdStk rP}z|t|W5d}~XYqXqtd|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*rMerrorstr0rrr!_validate_unionOs "$rzcCs&||kr"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]`\}}zt|d|d||Wq4tk r}ztd|d|d|W5d }~XYq4Xq4d 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_typeiitemr0rrr!_validate_listds rc Cst|ts$td|dt|j|\}}|D]l\}}z.t|d||t|d|d||Wq4tk r}ztd|d|W5d}~XYq4Xq4dS) 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_typer8r9r0rrr!_validate_dictrs rc CsZt|ts$td|dt|jt|dkr|dtkrt|D]d\}}z t|d|d||dWqDtk r}ztd|d |d |W5d }~XYqDXqDnt|t|krtd|d t|d t|nvtt ||D]f\}\}}zt|d|d||Wqtk rR}ztd|d |d |W5d }~XYqXqd 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*rMrrr0expectedrrr!_validate_tuples  .$rc Cst|ts$td|dt|j|d}t|D]T\}}zt|d||Wq4tk r}ztd|d|W5d}~XYq4Xq4dS)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 fkrXdS|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 Validator_Trrr)N)4r^Z dataclassesrrrrr functoolsrtypingrr r r r r rrrrrrrrrxrrrrrr"rorkrprXrvrzr|rrrrrsrtrRrVrUrrrrr__all__rrrr!s @*  K  +"