-:jTeUddlZddlZddlZddlmZddlmZmZmZm Z m Z ddl m Z m Z ddlmZmZmZmZmZmZmZmZmZmZ ddlmZmZd d lmZm Z m!Z!eegdfZ"ed Z#ed e$e%efZ&e'Z(edee#dee#fdZ)eddde*deee#gee#ffdZ)d>dddee#dzde*dee#eee#gee#fzfdZ)dee&de$ddfdZ+e dee&defdZ,dee&de$e%effdZ-eedddddfde.e"e"zdeded e*d!e*d"e*dzd#e*d$e$dzd%edefd&Z/de"fd'Z0d(e%d)ed*eddfd+Z1d(e%d)eddfd,Z2d(e%d)ed-e3ed.fddfd/Z4d(e%d)ed-e3ed.fddfd0Z5d(e%d)ed-e3ed.fddfd1Z6d(e%d)ed-e3ed.fddfd2Z7d(e%d)ed-e3ed.fddfd3Z8d(e%d)ed-e3ed.fddfd4Z9d(e%d)ed-e3ed.fddfd5Z:d(e%d)ed*eddfd6Z;d7ede"fd8Zee4ee5e.e6e$e7e3e8e?e9ejje:iZBe$eee%ee3ed.fgdffeCd<<e4eBej<gd=ZEy#e$r- ddlmZmZn#e$reddiZed diZYnwxYwYwxYw)?N)Callable)MISSINGFieldfieldfieldsmake_dataclass) lru_cachewraps) AnnotatedAny ForwardRefLiteralTypeTypeVarUnionget_args get_originoverload) NotRequiredRequiredrr)#StrictDataclassClassValidationErrorStrictDataclassDefinitionError#StrictDataclassFieldValidationErrorT TypedDictType)boundclsreturncyNr)rs ]/var/www/html/tokenscope/api/venv/lib/python3.12/site-packages/huggingface_hub/dataclasses.pystrictr$0s%(F accept_kwargsr'cyr"rr&s r#r$r$4sLOr%c\dttdttffd }|||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 ``` rr c tdstdjdi}tD]}g}|j t ||j jd}|Rt|ts|g}|D])}t|rtd|jd|d|j||||j<|_ jdtd t d td dffd }|_rj"}t%|d td dffd }|_tdsd td dfd} | _j(t%d t ffd } j*j,dur| _g} t/D]} | j1dst3| } t5| s-t7t9j:| j<dk7rtdjd| d| j | | _dt@d dffd }d|_!tdr0t3jDddstdjd|_"j"t%dfd }tGd|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|jj|gD] } || |||y#ttf$r}t |||d}~wwxYw)z1Custom __setattr__ method for strict dataclasses.)rcauseN)__validators__get ValueError TypeErrorr)r.r/r0r,eoriginal_setattrs r#__strict_setattr__z0strict..wrap..__strict_setattr__so"0044T2> Z Ze$ Z T4 / #I.Z=DPQRXYYZs7AAAkwargsct Dchc]}|j}}|jDcic] \}}||vs ||}}}t|dkDr%t d j dt|dt D]}|j|vr%t ||j||j6|jtur"t ||j|jj|jtur&t ||j|jtd|jdi}|jD]\} } | |vs | || <|jdi|ycc}wcc}}w)NrzWhen `accept_kwargs=True`, z, accepts only keyword arguments, but found `z` positional args.zMissing required field - ''r) rr/itemslenr5__name__setattrdefaultrdefault_factoryr6 __post_init__) r.argsr:fdataclass_fieldskvstandard_kwargsadditional_kwargsr/r0rs r#__init__z&strict..wrap..__init__sw5;3K#@qAFF#@ #@4:LLN"\DAqaK[F[1a4"\"\ t9q=$5cll^D&&)$i[0BD  PAvv0affoaff.EF'1affaii8**'9affa.?.?.AB'*DQVVHA(NOOP%'!#)<<>8KD%#3327)$/8#""7%67=$A"\sE= FFrCcN|jD]\}}t|||y)z2Default __post_init__ to accept additional kwargs.N)r=r@)r.r:r/r0s r#rCz+strict..wrap..__post_init__s''-||~3 edE23r%c|}|jjDcgc]\}}|jvrd|d|}}}dj|}|r |ddd|dS|Scc}}w)N*=z, ))__dict__r=r+join)r. standard_reprrGrHrJadditional_reprr original_reprs r#__repr__z&strict..wrap..__repr__s!.d 3 !% 3 3 5%1 8 88s!A5M%!% #')),=">GX-,-R/@Bj]jj%s A*T 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.cjD] } || y#ttf$r}t|j||d}~wwxYw)z%Run class validators on the instance.)r,r2N)__class_validators__r5r6rr?)r.r,r7rs r#validatez&strict..wrap..validatesW 55 l ldO l#I.l= HZHZbcdjkklsA AA r["__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.c@|g|i|j|y)z*Run class validators after initialization.N)r[)r.rDr:r initial_inits r#init_with_validatez0strict..wrap..init_with_validates#  / / / LL r%rK)r N)$hasattrrr?rappend_create_type_validatormetadatar4 isinstancelist _is_validatorr/extendr3 __setattr__r strrKr rCrW__dataclass_params__reprdir startswithgetattrcallabler>inspect signature parametersrZrr\r[r@)rfield_validatorsrE validatorscustom_validatorr,r9 original_initrKrCrWclass_validatorsr/methodr[r_r^rVr8r's` @@@r#wrapzstrict..wrappss230#,,'UV  :< 2AJ   4Q7 8 zz~~k: +!"2D9(8'9$!1I(3<;AFF83ykQA !!"23'1 QVV $ 2.?? 0S 0 0C 0D 0- LLM = ! 8 8 8" 8D$CL303#3$3 %2! LLM = ! k# k" k '',,4' H ,D??;/S$'FF#7$$V,778A=4cll^+DTFK>>  # #F + ,$4  l1 l l7;3 3 #3<<)MuU4cll^,NN   || |     Z!34 r%)rr)rr'rys ` r#r$r$8s8ph$q'hd1ghV491T1r%schemadatac*t|}|di|y)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)rzr{ strict_clss r#validate_typed_dictrsX36:Jr%c t|}t|dds|jD]t\}}t|}|turDt |^}}t |s t|}tt|gt|z||<]t |rit|||<vg}|jD]p\}}t|tur7t |^}}|j||ttd|difN|j||ttfrtt|j|S)N __total__Tr,r)rArc)rA)_get_typed_dict_annotationsrnr=rr r_is_required_or_notrequiredrtuplererar_TYPED_DICT_DEFAULT_VALUEr$rr?)rz type_hintskeyr0originbasemetars r#r}r}Ps9-V4J 6; -$**, 5JC&F"&uo t248&t,D"+E4&4:2E,F"G 307"-e"4 3 5F &&(R U e  )"5/KD4 FMM3e4MYdfjklfmXn&op q FMM3u5N'OP Q R .&9 ::r%c  ddl}|j|S#t$rU|jj dij Dcic]\}}|||n t dncc}}wc}}cYSwxYw)z0Extract type annotations from a TypedDict class.rN__annotations__) annotationlibget_annotations ImportErrorrRr4r=type)rzrr/r0s r#rros}  ,,V44  &223DbIOOQ  e 5,%$t* <    s6A5 A(' A54A5Tr,rArBinitrkhashcomparercr:c `t|ts|g}|i}||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 r,rArBrrkrrrcr)rdrer) r,rArBrrkrrrcr:s r#validated_fieldrs[4 i &K %H[  '       r%cttdddddfdtdtdtdtdtdzdtd tdzd tffd }|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. TNrArBrrkrrrcr:c ,tf|||||||d|S)Nr)r) rArBrrkrrrcr:r,s r#_innerz"as_validated_field.._inners9   +    r%)rr booldict)r,rs` r#as_validated_fieldrsz&  $      Tk    +  , Mr%r/r0 expected_typect|}t|}|tury| t||ytj |x}r ||||yt |trt|||yt |tst |try|tur(|turtd|dt|||dy|tur|turyt|||dytd|d|)z.Validate that 'value' matches 'expected_type'.NField 'z' is required but missing.rzUnsupported type for field 'r-)rrr _validate_none_BASIC_TYPE_VALIDATORSr4rdr_validate_simple_typer rirrr6type_validatorr)r/r0rrrDr,s r#rrs  &F M "D  tU#,008 8 8$t$ M4 (dE=9 M: .*]C2P 8  - -gdV+EFG GtUDG, ;  - - tUDG,6tfC OPPr%cP|$td|dt|jy)zValidate None type. 'None' is not a type, it's a special value. Type should be `NoneType` instead. But in type annotations 'None' is accepted so we must support it. Nrz' expected None, got )r6rr?)r/r0s r#rrs3  '$'tE{?S?S>TUVVQIU#R4 R dV1QCq>4 ;R R4QCz$qIJPQ Q Rs A"" B+A>>Bc0t|ts$td|dt|j|\}}|j D])\}} t |d||t |d|d||+y#t$r}td|d|d}~wwxYw) zValidate dict[K, V] type.rz' expected a dict, got z.keyrrzInvalid key or value in dict 'r<N)rdrr6rr?r=r)r/r0rDkey_type value_typerGrHr7s r#_validate_dictrs eT "'$'>tE{?S?S>TUVV Hj M1 M dV4=!X 6 dV1QE+Q ;M MEllipsisrrzip)r/r0rDrrr7expecteds r#_validate_tupler)si eU #'$'?U @T@T?UVWW 4yA~$q'X- ' WGAt W$q1~tT!W= W Tc%j '$'DSYKvVYZ_V`Uabcc#,S-=#> W Ah W$q1~tX> W  W"8;tfA NOUVV W W"8;tfA NOUVV Ws0 C+D+ D 4DD  D0D++D0ct|ts$td|dt|j|d}t |D]\}} t |d||y#t$r}td|d|d}~wwxYw)zValidate set[T] type.rz' expected a set, got rz itemzInvalid item in set 'r<N)rdsetr6rr?rrrs r# _validate_setr@s eS !'$'=d5k>R>R=STUUQIU#D4 D dV5>4 ;D D3D6;4 ;V V4QC~dV1MNTU U Vs#A99 BBBc D|turNt|tr>td|d|jdt |jdt |d t||s>td|d|jdt |jdt |d y)z&Validate simple type (int, str, etc.).rz ' expected rz (value: rQN)rrdrr6r?rrk)r/r0rs r#rr`s 5$ 7dV;}'='=&>fT%[EYEYDZZcdhindocppq r   e] +dV;}'='=&>fT%[EYEYDZZcdhindocppq r   ,r%rc$dtddffd }|S)z-Create a type validator function for a field.r0r NcHtj|jyr")rr/r)r0rs r#r,z)_create_type_validator..validatorpsuzz5%**5r%)r )rr,s` r#rbrbls666 r%ct|sytj|}t|jj }t |dk(ry|djtjjtjjtjjfvry|ddD]+}|jtjjk(s+yy)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)rorprqrerrvaluesr>kind ParameterPOSITIONAL_OR_KEYWORDPOSITIONAL_ONLYVAR_POSITIONALrAempty)r,rqrr parameters r#rfrfvs I !!),Ii**1134J :!!}//))((" ^    1 1 7 7 7 r% type_hintcP|ttfvxst|ttfvS)z2Helper to check if a type is Required/NotRequired.)rrr)rs r#rrs' ;/ / eJy4IhXcMd4der%r)r$rr Validator_Trrrr")Fcollections.abcrrptypesr dataclassesrrrrr functoolsr r typingr r r rrrrrrrrrrtyping_extensionsrrrrrrrrrirobjectrr$rrr}rrerrrrrrrrrrrrrrbrfrrrrrr UnionType__all__rr%r#rs $EE&    2,ud{#  CLtCH~> "H (Q(DG( ( $)OTOhQy$q'7I.JO Oc2c2Q$c2c2$q'T\^bcd^e]fhlmnho]oTpJpc2L/] 3/4/D/d ;d=.A;d; ;<  ](;  S#X  $" (K ;.( ((  (  ( + ((Tk(( (V+DQQSQQQ6UUSUTU # c sCx T  OC O O5c? Ot O R RS Rc3h RD R M MS Mc3h MD MW#WcWsCxWTW. D DC DuS#X D4 DVSVVE#s(OVV$    C      %KST6f3f4f  ? .. ?OO0RS(CeCHo+F+L"MMN+:u' Y22;; 2 B+="b1 2 2s6H//I!5H>=I!>II!II! I!