KiKUddlZddlZddlZddlZddlZddlZddlZddlmZm Z m Z ddl m Z m Z m Z ddlmZmZddlmZmZddlmZmZmZmZmZmZmZej6dk\rddlmZndZdd lmZmZm Z m!Z!m"Z"m#Z#dd l$m%Z%m&Z&dd l'm(Z(m)Z)dd l*m+Z+m,Z,m-Z-m.Z.m/Z/ej6dk\rddlmZndZerdd l0m1Z1edZ2edeZ3edeZ4e5e5e6e7e7e6e8e8e6e9e9e6dfe:e:e6e6fiZ;e:e0fnnofalseF>1tyyestrueT) target_type)lowerrboolr/s ]/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/cyclopts/_convert.py_boolrAGs2  A ** , ,--c(|j}|jdr t|dS|jdr t|dS|jdr t|dSd|vrttt |St|S)N0x0o0b.)r= startswithintroundfloatr?s r@_intrORsz  A||D1bz d 1ay d 1ay 5q?##1v rBct|dS)Nutf8)encoding)bytesr?s r@_bytesrTcs V $$rBc*tt|SN) bytearrayrTr?s r@ _bytearrayrXgs VAY rBc,tj|S)zDParse a date string. Returns ------- datetime.date )r fromisoformatr?s r@_dater[ks   a  rBc tj|S#t$r7tj|jj dddcYSwxYw)zLParse a datetime string. Returns ------- datetime.datetime  r()rrZ ValueErrorstripreplacer?s r@ _datetimerbusRF%%a(( F%%aggi&7&7S!&DEE Fs=AAcd}|jdrd}|dd}tjd|}|std|d}|D]p\}}}t |}|d k(r||z }|d k(r ||d zz }+|d k(r ||d zz }9|dk(r ||dzz }G|dk(r ||dzz }U|dk(r ||dzz }c|dk(si||dzz }r|r| }t |S)zParse a timedelta string.F-Tr^Nz((\d+\.\d+|\d+)([smhdwMy]))z!Could not parse duration string: rr/m<hidiQwi: Mi'r9i3)seconds)rKrefindallr_rNr)r/negativematchesrk_valueunits r@ _timedeltarssH||C abEjj7;G z$convert_enum_flag..sOuE> :Osr)r operatoror_)rrrvs` `r@convert_enum_flagrs(4  OO!  rB _converters converterc Ntt||tdt|D}t |\}t d|rt |z}|rtddt |dt |dk(r|dn t |dk(rtn td tfd t|DSt |k7rtd dt |d|Dcgc]}t |d}}t|} |D cgc]$} t| D cgc] } t| c} &} } } | D cgc]} t | dk(r| dn| } } tfd t|| d D}|Scc}wcc} wcc} } wcc} w)Nrrvc3*K|] }|dus| yw).Nrxs r@rz!_convert_tuple..sCaaslCs r^z4Incorrect number of arguments: expected multiple of z but got rJ)msgrz%A tuple must have 0 or 1 inner-types.c3DK|]}dk(r|dn|yw)r^rNr)rchunkconvertinner_token_count inner_types r@rz!_convert_tuple..s.  J,=,Ba N s z(Incorrect number of arguments: expected c36K|]\}}||ywrVr)rrargrs r@rz!_convert_tuple..slSGJ,lF)strict)r _converttupler token_countmaxlenrstrr_r#iterrangenextzip)rtrrvr inner_types consume_all remainderrargs_per_convertitsizerpbatchedelemoutrrrs @@@r@_convert_tuplers h)NSGC8E?CCK%0%7"{-q1K"33 JK\J]]fgjkqgrfsstu  { q $QJ   "JDE E  ):;   F +>?P>QQZ[^_e[fZgghi 8CC!KN1-CC &\?OPteDk2DH2PPCJK4c$i1n47$6KKlC U\ejDkll J D2PKs$;F"F4F FF"Fdataz Token | Nonect|}t}|jD]1\}}|j||j |j 3t|j |z }|rWt|d}djt|j } td|d|jd| ||y)aValidate that JSON data doesn't contain extra keys not in the type's fields. Parameters ---------- data : dict The JSON dictionary to validate. type_ : type The target type (dataclass, etc.) to validate against. token : Token | None Optional token for error context. Raises ------ CoercionError If the data contains keys not present in the type's fields. rz, zUnknown field "z" in JSON for z. Valid fields: )rr<ruN) r setr|addupdatenameskeyssortedjoinr__name__) rrtru field_infos valid_names field_name field_info extra_keys extra_key valid_fieldss r@_validate_json_extra_keysrs*"%(KEK"-"3"3"5- J #:++,-TYY[!K/J:&q) yy (8(8(:!;< !)N5>>:JJZ[gZhi  rBrctddlm}t||i}|jD]\}}||vs ||} | rt |j t sX|t| ttzrtj| n t | } t|j | g||} n| } | ||<|di|S)aConvert JSON dict to dataclass with proper type conversion for fields. Parameters ---------- type_ : Type The dataclass type to create. data : dict The JSON dictionary containing field values. field_infos : dict Field information from the dataclass. converter : Callable | None Optional converter function. name_transform : Callable[[str], str] Function to transform field names. Returns ------- Instance of type_ with properly converted field values. rr&rqr) cyclopts.tokenr'rr|r%hintrrzdictlistjsondumpsr) rtrrrrvr'converted_datarrrqruconverted_values r@ _convert_jsonr?s4%dE*N"-"3"3"5 9 J  $E )>zPS)TE4RV;9WDJJu$5]`af]gh")*//E7I~"^"')8N: & 9  "> ""rBerrorc D|jj}t|dr |jnt|dr|jdz nd}t d|dz }t t||dz}|||}|dkDrd|z}|t|kr|dz}||z }|dkDr|dz }d} tjd |rd } n8tjd |rd } ntjd |rd} nd|vrd} d|jd|dd|zd|j| S)aCreate a helpful error message for JSON decode errors. Parameters ---------- token : Token The token containing the invalid JSON. type_ : Type The target type we were trying to convert to. error : json.JSONDecodeError The JSON decode error that occurred. Returns ------- str A formatted error message with context and hints. poscolnor^rz...rz\bTrue\bz8 Hint: Use lowercase 'true' instead of Python's Truez \bFalse\bz: Hint: Use lowercase 'false' instead of Python's Falsez\bNone\bz. Hint: Use 'null' instead of Python's None'z9 Hint: JSON requires double quotes, not single quoteszInvalid JSON for z: z r]z^ ) rqr`hasattrrrrminrrlsearchrr) rurtr value_str error_pos snippet_start snippet_endsnippet marker_posrs r@!_create_json_decode_error_messagerps?* !!#I%UE2 7SXZaKb ahiI9r>*Mc)ni"n5K k2Gq'/S^#E/]*Jqa  D yyi(J < +L ; *@ K u~~.ggYfS:EUDVVXY^YbYbXcdhci jjrBcrt|}|s|di|Sg}i}|jD]N\}}|j|}|r1|jtj k(r|j ||fJ|||<Pt|j|jfd|d|Di|S)a%Instantiate a type with proper handling of parameter kinds. Respects POSITIONAL_ONLY, KEYWORD_ONLY, and POSITIONAL_OR_KEYWORD parameter kinds when constructing the object. This function is necessary because `inspect.signature().bind(**data)` has the same limitation we're solving: it cannot accept positional-only parameters as keyword arguments. For example, `def __init__(self, a, /, b)` requires `a` to be passed positionally, but when we have a dict `{"a": 1, "b": 2}`, we need to transform this into the call `type_(1, b=2)`. Parameters ---------- type_ : type[T] The type to instantiate. data : dict[str, Any] Dictionary mapping field names to values. Returns ------- T Instance of type_ constructed from data. c,j|dS)Nr)index)rfield_names_orders r@z'instantiate_from_dict..s 1 7 7! =rB)keyc3&K|] \}}| ywrVr)rrpvs r@rz(instantiate_from_dict..s*A1*sr) r r|getkindrPOSITIONAL_ONLYappendrrsort) rtrrpos_argskwargsrrqrrs @r@instantiate_from_dictrs0"%(K }t}H F!ZZ\' E __Z0 *//Y-F-FF OOZ/ 0!&F:  '[--/0 MM=M> ** 5f 55rBrrcd}i}|}|jD]\}}|j}t|tr||j} |dz }d} n;t |\} } | dk(r||||} |dz }n|||||| z} || z }| } | ||<| rn|t |k(sn|t |k(sJt||S)aConvert tokens to a structured type with proper positional/keyword argument handling. Respects the parameter kind of each field: - POSITIONAL_ONLY: passed as positional argument - KEYWORD_ONLY or POSITIONAL_OR_KEYWORD: passed as keyword argument This correctly handles types with keyword-only fields (e.g., dataclasses with kw_only=True). Parameters ---------- type_ : type[T] The target structured type to instantiate. token : Sequence[Token] The tokens to convert. field_infos : dict[str, FieldInfo] Field information for the structured type. convert : Callable Conversion function for nested types. Returns ------- T Instance of type_ constructed from the tokens. rr^F)r|rr%rrqrrr) rtrurrirrrrrq should_breaktokens_per_elementrs r@_convert_structured_typers< A D D"-"3"3"5 J !s +!HNNE FA L.9$.? +  !Q&eAh/QeA4F0F&GH''&L Z   E ? 14 E ??  --rBc` ddlmddlm}d}t |rHddlm}|j |\}j r d}fd}|}jrj}ndtt||tt||}t|}|tvr t|}t||turtttf|} n|t vrt ||} n|tvr*t|} t j#| | } | |} nMt$%t'|t$r|j(|} n"t+|rBD]} t-| r | |} nt'|t0rt2t5|| |t6urbd} t|D]} t9||}||k(s|} n| r || _| t5t'|t0r |d| || |t<ur't'|r |||| } nJ||g|d |i} n<|t>vrtAd\}}t'|t0st2d}|d kDrtCfd |Dr |tur|}n |d kDrtEtG|g|zddi}n|}|fd|D} ntI|tJr"tM|t'|t0r|n|g|} n~tI|tNr.converter_with_token.sh''''&,%5%5"0#6)05G)H&##$67-uh77.b5(;;rBNrrx)rrr^c3K|];}t|xr)|jjjd=yw){N)rzrqr`rK)rr8r's r@rz_convert..s6\QRJq%(LQWW]]_-G-G-LL\sAArc36K|]}d|yw)rNr)rerrs r@rz_convert..sB'+a.!4Brr)rrur<r)exception_messagerq)=ryr'cyclopts.parameterrrfrom_annotationrrvr rrrr-rrrr,rrrz __value__rr Exceptionrr_rrtyper<rrranyrrr%r rr rrqr r$implicit_valuer!rrur`rKrloads TypeErrorrJSONDecodeErrorrr validatorAssertionErrorrargs)rtrurrvrconverter_needs_tokenr convert_tuple origin_typer concrete_type default_paramr8last_coercion_errorchoiceresrcountrprgenrrrr r'rrrs @@@@r@rrs(,!E0!11%8 v   $( ! <"-I  #22Nh)NSGNiP^_MU#K888E 5/K }d38ne, 1 15e Any: "Perform conversion of string token." This allows to use the :func:`convert` function to handle the the difficult task of traversing lists/tuples/unions/etc, while leaving the final conversion logic to the caller. name_transform: Optional[Callable[[str], str]] Currently only used for ``Enum`` type hints. A function that transforms enum names and CLI values into a normalized format. The function should have signature: .. code-block:: python def name_transform(s: str) -> str: "Perform name transform." where the returned value is the name to be used on the CLI. If ``None``, defaults to ``cyclopts.default_name_transform``. Returns ------- Any Coerced version of input ``*args``. rr&c3BK|]}t|yw)rN)r)rrr's r@rzconvert.."s;qu3q6**;srr^z*Dictionary of tokens provided for unknown rJz Unreachable?r)ryr'r_rzrrrr"r rrrrr,rr-rrr IndexErrorr|rrrrrrNotImplementedError)rtrrrv convert_privrrrmaybe_origin_type value_typekrdict_convertedrritemr's @r@rrset(  fd # 6!9c(B;F;;/8yXLNiP^_M ENE | + / /u =E 22:5A /33M=QU#K888E #,ueU,V,, n ,E6** d "&$'  !%+Jgmfrfrft ^b^_abAwz1 .Y Y  02CDV~VV FD !EeYaPQQ ' (!!2FNKK*5e*<'K v. . [A vay1 1 1 $:@A$L-A A [. .v. .%n5 5/ J  "Bs,H- I8I -I?Iskip_converter_paramscddlm}|||\}}|D]F}|j|jdk(ry||urt|d\}}nd}|j|fcS|}t |}|t vr t |}|t vr*t |} t j| | }t |}|xs|tur$t|} | rtd | Dd | vfSy|xs|tury |tvs|tvrtt|dk(ryt|ry|tvr0tt|rtt|dddfSt%t!|trt|j"St%|rIt|} t| d} | d dD]&} t| }|| k7st'd | dd| | St)|ryt+|}d \}}|j-D]L}|j.|j0urd}n |j2s+t|j4\}}||z }||z}N|sy||fS)aThe number of tokens after a keyword the parameter should consume. Parameters ---------- type_: Type A type hint/annotation to infer token_count from if not explicitly specified. skip_converter_params: bool If True, don't extract converter parameters from __cyclopts__. Used to prevent infinite recursion when determining consume_all behavior. Returns ------- int Number of tokens to consume. bool If this is ``True`` and positional, consume all remaining tokens. The returned number of tokens constitutes a single element of the iterable-to-be-parsed. r)get_parameters)r$N)r^TTFc3BK|]}|dust|dyw).rN)rrs r@rztoken_count..sGQ!3,{1~a(Gs .)rFr^z=Cannot Union types that consume different numbers of tokens: r])r^F)rr&n_tokensrrr-r,rrrsumr>rrrrrzrrr_r$r valuesrVAR_POSITIONALrequiredr)rtr$r& resolved_type parametersparamrpconsume_all_from_typerrrsub_argstoken_count_target sub_type_thisrrrrq elem_countelem_consume_alls r@rr_s*2 .uLa bM:= >> %~~# !-0;=`d/e,A,-2)~~'<<rArLrOrSrTrWrXr[rbrsrrrrrrr rrrrrrrrBr@rKs(  88..%w$Mkk>:ddw$M' CL Ct Ct y~$s)S 5c?$sCx. 5dDj!1 d OOT dOOOOOO##TOOTOO""D 8"Dt$4 S%  >??@ .S.T.CC"%c%e% # ) !S!T! F F F "&#"&)"&J 7 # cUCZ( .Aw SMHW- -cUCZ( F % 6 z% i z$ T#x- ( 9( (s S()D0(cUCZ( (  (\!# # #  #  # L.# .# .#.#$ .# cUCZ( .#b5k 5k 5k   5k 5kp*6a*6S#X*61*6Z=. 7=. G =.c;&'=. =.  =.@J (7++ ,Jc C(4/ J cUCZ( J`4826 z6 z6 SMHW- - =z6s S()D0z6cUCZ(4/ z6zd"sd"4d"E#t)DTd"rB