o )Ti5L@sUdZddlZddlZddlZddlZddlZddlZddlmZm Z m Z m Z ej dkr3ddlm ZneZddlmZmZddlmZmZdd lmZmZdd lmZmZmZgd Zejd ejd ej dej!dej"dej#diZ$dde$%DZ&iZ'e(e)efe*d<iZ+e(edfe*d<iZ,e(e-e fe*d<   dde-dede de e-de e(de e)defdd Z. ! dde e-ded"e e)e/e)fd#e-de e-def d$d%Z0 dde-d&e e-ej1fde e-defd'd(Z2d)d*Z3d+efd,d-Z4dd.e-de e-defd/d0Z5Gd1ddZ6e-de7e8e9fd2d3d4dfd5ed6e d7e e d8e ee:e)ee:d9ffd:e d;e;dZdd5ed>=<<=z==z!=cCsi|]\}}||qSr*).0kvr*r*G/home/ubuntu/.local/lib/python3.10/site-packages/jsonargparse/typing.py 7sr/registered_typesRegisteredTyperegistered_type_handlersregistration_pendingname base_type validation_fn docstring extra_attrs register_keyreturnc s|tvrt|}|j|krtd|jd|SGfddd}|dur5|D] \}} t||| q*t||fd|i} t| || S)aCreates and registers an extension of base type. Args: name: How the new type will be called. base_type: The type from which the created type is extended. validation_fn: Function that validates the value on instantiation/casting. Gets two arguments: ``type_class`` and ``value``. docstring: The ``__doc__`` attribute value for the created type. extra_attrs: Attributes set to the type class that the ``validation_fn`` can access. register_key: Used to determine the uniqueness of registered types. Raises: ValueError: If the type has already been registered with a different name. z4Same type already registered with a different name: .cs$eZdZZZfddZZS)z"extend_base_type..TypeCorecs |||t|||SN)_validation_fnsuper__new___typeclsr- __class__r*r.r?^s z*extend_base_type..TypeCore.__new__)__name__ __module__ __qualname__r=r@r? __classcell__r*r5r6rCr.TypeCoreZsrJN__doc__)r0rE ValueErroritemssetattrtypeadd_type) r4r5r6r7r8r9registered_typerJkeyvalue created_typer*rIr.r>s  rand restrictionsjoinc s<tthvr td|dvrtdt|tr|gn|}t|tr5tdd|Dr5tfdd|Ds>tdjdtt||f}d d |D}d |d  d d |D}|d urj}t |D]!\}\}} ||dkrxd|dnd7}||jt |  dd7}qg|||d} dd} t || ||| dS)aCreates or returns an already registered restricted number type class. Args: name: Name for the type or ``None`` for an automatic name. base_type: One of ``{int, float}``. restrictions: Tuples of pairs (comparison, reference), e.g. ``('>', 0)``. join: How to combine multiple comparisons, one of ``{'or', 'and'}``. docstring: Docstring for the type class. Returns: The created or retrieved type class. z-Expected base_type to be one of {int, float}.>orrUz)Expected join to be one of {'or', 'and'}.css&|]}t|tot|dkVqdS)N) isinstancetuplelenr+xr*r*r. s$z)restricted_number_type..c3s0|]}|dtvo|d|dkVqdS)rr N _operators2r]r5r*r.r_s.z}Expected restrictions to be a list of tuples each with a comparison operator (> >= < <= == !=) and a reference value of type r;cSs g|] }t|d|dfqS)rr r`r]r*r*r. s z*restricted_number_type.. cSs$g|]\}}dt|t|qS)r-) _operators1str)r+oprefr*r*r.rcs$Nr_) _restrictions _expression_joinr@cst|tr t|d|jtkr"t|tr"t|s"t|d||fdd|jD}|jdkr:t |rC|jdkrMt |sOt|d|j dSdS)Nz not a numberz not an integercsg|] \}}||qSr*r*)r+ comparisonrhvvr*r.rczArestricted_number_type..validation_fn..rUrXz! does not conform to restriction ) rZboolrLr@intfloat is_integerrkrmallanyrl)rBr-checkr*ror.r6s  $z-restricted_number_type..validation_fnr4r5r6r9r7r8)rsrtrLrZr[listrvrEsortedrW enumeraterfreplacer) r4r5rVrWr7r9 expressionnumrnrhr8r6r*rbr.rlsL  rregexcCsJt|tr t|}d|j}||td}dd}t|t||tf||dS)aCCreates or returns an already registered restricted string type class. Args: name: Name for the type or ``None`` for an automatic name. regex: Regular expression that the string must match. docstring: Docstring for the type class. Returns: The created or retrieved type class. z matching )_regexrlr@cSs&|j|st|d|jjdS)Nz# does not match regular expression )rmatchrLpatternrAr*r*r.r6s z-restricted_string_type..validation_fnry)rZrfrecompilerr)r4rr7r~r8r6r*r*r.rs   rcCs t|tSr<)rZr)rS type_classr*r*r. _is_path_type rpathcCs<t|ts tdtr|j|jkr|j|jdSt |S)NzExpected a Path instance.)relativecwd) rZrrLr getrabsolute _relative_cwdrf)rr*r*r._serialize_paths rmodec stdddt}t|ddr-ddlm}|dd d 7|d 7}|tf}|tvr9t|SGfd d d t}t |fd|i}t ||t d|S)aCreates or returns an already registered path type class. Args: mode: The required type and access permissions among ``[fdrwxcuFDRWX]``. docstring: Docstring for the type class. Returns: The created or retrieved type class. Path_zpath rjF) skip_checkr )path_skip_check_deprecation) stacklevel _skip_checkz skip_checkcs,eZdZZZZeZfddZZ S)zpath_type..PathTypecst|tr7t|ddhkr7t|dtj|df|j|jd|WddS1s0wYdStj|f|j|jd|dS)Nrr)rr)rZdictsetrr>__init___moder)selfr-r,rCr*r.r s $""z$path_type..PathType.__init__) rErFrGrlrrrr@rrHr*rr4rrCr.PathTypes rrK type_check) r _check_moderWr{r _deprecatedrrfr0rOrPr)rr7kwargskey_namerr9rrestricted_typer*rr.rs     rc @s^eZdZdededeedeeee eedffdef ddZ d d Z d d Z d dZ dS)r1r serializer deserializerdeserializer_exceptions.rcCs.||_||_|dur |n||_||_||_dSr<)rrbase_deserializerrr)rrrrrrr*r*r.rs  zRegisteredType.__init__cstfdddDS)Nc3s$|] }t|t|kVqdSr<)getattr)r+r,otherrr*r.r_(s"z(RegisteredType.__eq__..)rrr)rvrrr*rr.__eq__'zRegisteredType.__eq__cCs|||jSr<)rrrrSr*r*r.is_value_of_type*szRegisteredType.is_value_of_typec Cs\z||WS|jy-}zt|jdt|j}td|d|}||_||d}~ww)NrEz Not of type z: )rrrrrfrLparent)rrSextype_class_nameex2r*r*r.r-s zRegisteredType.deserializerN)rErFrGrrrrrO Exceptionr[rrrrr*r*r*r.r1s  cCs |j|kSr<rC)r-tr*r*r.@s rTrrrr.rfail_already_registereduniqueness_keycCst|s td|t|||||}td|}|s3|r3t|r3|t|kr+dStd|d|t|<|durA|t|<dSdS)aRegisters a new type for use in jsonargparse parsers. Args: type_class: The class to be registered. serializer: Function that converts an instance of the class to a basic type. deserializer: Function that converts a basic type to an instance of the class. Default instantiates ``type_class``. deserializer_exceptions: Exceptions that deserializer raises when it fails. type_check: Function to check if a value is of ``type_class``. Gets as arguments the value and ``type_class``. fail_already_registered: Whether to fail if type has already been registered. uniqueness_key: Key to determine uniqueness of type. z'Expected type_class to be a class, got _fail_already_registeredNzType "zC" already registered with different serializer and/or deserializer.) inspectisclassrLr1globalsrget_registered_typer2r0)rrrrrrr type_handlerr*r*r.r7s   r import_pathcsfddt<dS)NcsttgRiSr<)rrr*argsrrr*r.r^sz,register_type_on_first_use..)r3)rrrr*rr.register_type_on_first_use]rrcCsd|tvr-ddlm}|ttt|}|tvrt|Wdn1s(wYt|S)Nr)suppress) r2 contextlibrAttributeErrorrLrr3popr)rrrr*r*r.res    rcCsf|tvsJ|jtvrtd|jd|t|j<d|i}|dur'||d<t||jfi|dS)Nz Type name "z;" clashes with name already defined in jsonargparse.typing.rr)r0rErrLrr@)rrrrr*r*r.rPps   rPFr)r&rzint restricted to be >0)r7r)r'ruint restricted to be ≥0rzfloat restricted to be >0rufloat restricted to be ≥0r)r)r u$float restricted to be ≥0 and ≤1r)r(r z float restricted to be >0 and <1rz ^.*[^ ].*$z2str restricted to not-empty pattern ``^.*[^ ].*$``r z^[^@ ]+@[^@ ]+\.[^@ ]+$z?str restricted to the email pattern ``^[^@ ]+@[^@ ]+\.[^@ ]+$``frz*path to a file that exists and is readablefcz7path to a file that can be created if it does not existdwz0path to a directory that exists and is writeabledcz.raise_errorz7(?P\d+):(?P\d+):(?P\d[\.\d+]*)dayz(?P[-\d]+) day[s]*, cSsi|] \}}|t|qSr*)rt)r+rRvalr*r*r.r/rqz*timedelta_deserializer..r) timedeltar*)rZrfrr groupdictrMdatetimer)rSrrrrrr*rr.timedelta_deserializers    rzdatetime.timedelta)rrScCsddlm}||S)Nr) b64encode)base64rdecode)rSrr*r*r.bytes_serializer  rcCsddlm}||SNr) b64decode)rrrSrr*r*r.bytes_deserializers rcCsddlm}t||Sr)rr bytearrayrr*r*r.bytearray_deserializerrrzbuiltins.bytes)rrzbuiltins.bytearraycCsV|jdkr|jdkrd|jdSd|jd|jdSd|jd|jd|jdS)Nr rrange()z, )stepstartstoprr*r*r.range_serializers  rz ^(-?\d+)$z^(-?\d+),(-?\d+)$z^(-?\d+),(-?\d+),(-?\d+)$cCs|}|drT|drT|dddd}t|}|r'tt|dSt|}|r;tt|dt|dSt |}|rTtt|dt|dt|d St d ) Nrrrdrjr rYrzXExpected 'range()' or 'range(, )' or 'range(, , )') strip startswithendswithr} re_range_stoprrangersre_range_start_stopre_range_start_stop_steprL)rSrr*r*r.range_deserializers   $rc@sheZdZdZdefddZdefddZdefdd Zd e de fd d Z defd dZ defddZ dS)rz8Holds a secret string that serializes to ``**********``.rScCs ||_dSr<_valuerr*r*r.rrzSecretStr.__init__r:cCsdS)Nz **********r*rr*r*r.__str__szSecretStr.__str__cC t|jSr<)r\rrr*r*r.__len__rzSecretStr.__len__rcCst||jo |j|jkSr<)rZrDrrr*r*r.rrzSecretStr.__eq__cCrr<)hashrrr*r*r.__hash__rzSecretStr.__hash__cCs|jS)z Returns the actual secret value.rrr*r*r.get_secret_valueszSecretStr.get_secret_valueN)rErFrGrKrfrrrsrrrrrrrr*r*r*r.rsrzpydantic.SecretStrcs,ddlm}|d|dfdfdd}|S)Nr) create_modelpydantic_model.pydantic_fieldcs |djS)Nrrrrr*r. deserialize s z*pydantic_deserializer..deserialize)pydanticr)rrrr*rr.pydantic_deserializers  rcCsPt}tttttttffD]}t|ts||f}t ||dr%|d}|Sq |S)Nrr ) rfrsrtrrrzrrrZr[ issubclass)rrbaser*r*r.pydantic_serializers r>pydantic.typespydantic.networkspydantic_core._pydantic_corepydantic_extra_typescCs&tot|otddt|DS)Ncss |] }t|ddtvVqdS)rFrjN)rpydantic_type_modules)r+rr*r*r.r_&sz#is_pydantic_type..)r rrrwgetmro)rr*r*r.is_pydantic_type"s r cCs^ddlm}||r |j}t|sdSt|s-ddlm}t|t|t ||t fddSdS)Nr ) is_annotatedr)ValidationError)rrrr) _optionalsr  __origin__r rrrrrr TypeError)rr rr*r*r.register_pydantic_type*s   r)NNN)rUNr<)irKroperatorospathlibrsystypingrrrr version_infor _TypeAliasrO_commonr r rr r _pathsrr_utilrrr__all__gtgeltleeqnererMrar0rr[__annotations__r2r3rfrrzrPatternrrrrr1rLrrrrrrrrrPrrsrrrtrrrrrr r!r"r#r$r%PathLikecomplex PosixPath WindowsPath_pathrZrbytesrrrrrrrrrrrrrrr r rr*r*r*r.s2  2 L  ' -!    &