o )Ti+0@sUdZddlZddlmZddlmZddlmZddlm Z m Z m Z m Z dgZ hdZGd d d eZd ed eefd dZd ed eefddZd ed eefddZd ed efddZde deffddZd,ddZeddZGdddejZeeeZeee d<dZ!d ed efdd Z"d ed efd!d"Z#d#ed efd$d%Z$d#eee fd efd&d'Z%dd(l&m'Z'e'd)rddl(Z(e(j)*ed*d+dSdS)-z3Classes and functions related to namespace objects.N) OrderedDict)Iterator)contextmanager)AnyCallableOptionalUnion Namespace>__orig____path____default_config__c@seZdZddZdS) NSKeyErrorcCst|jdS)Nr)strargsselfrK/home/ubuntu/.local/lib/python3.10/site-packages/jsonargparse/_namespace.py__str__szNSKeyError.__str__N)__name__ __module__ __qualname__rrrrrr s r keyreturncCs |dS)N.splitrrrr split_keys rcC |ddSNrrrrrrsplit_key_root r"cCrr )rsplitrrrrsplit_key_leaf!r#r%cCst|d}|tvS)N)r% meta_keys)rleaf_keyrrr is_meta_key%s r)cfgcCs|rt|td}|S)N skip_keysrecreate_branchesr')r*rrr remove_meta*s r/cs|}t|ttfr1t|ts1t|}t|d|D]\}}dus'|vr.t|||<q|St|tr?fdd|D}|S)N__dict__csg|]}t|qSr)r..0vr+rr 8sz%recreate_branches..) isinstancer dictrtypegetattritemsr.list)datar,new_datarvalrr+rr.0s  r.ccs*tj}tt_z dVW|t_dS|t_wN)argparser )namespace_classrrrpatch_namespace<s rAc s eZdZdZfddZdedeeedeffddZdedeedeffdd Z deddfd d Z d ed e ddffdd Z dede ddfddZ dede fddZdeddfddZdedefddZdefddZdeee ffddZdejfddZd:d"ed#edeeee ffd$d%Zd:d"ed#edeefd&d'Zd;d"edee fd(d)Zd!efd"ed*edeefd+d,Zdded3e de fd4d5Z#dedee deffd6d7Z$d>ded3e de fd8d9Z%Z&S)?r zJExtension of argparse's Namespace to support nesting and subscript access.cst|dkrtjdi|dSt|dks't|dks't|dtjtfs+tdt|dtr8|dnt |dD]\}}|||<q@dS)a2Initializer for Namespace instance. Instantiating a Namespace with initial values most commonly is done by providing keyword arguments, e.g. ``Namespace(name1=value1, name2=value2)``. Alternatively a single positional ``Namespace`` or ``dict`` object can be given. rr!zAExpected a single positional parameter of type Namespace or dict.Nr) lensuper__init__r5r?r r6 ValueErrorr9vars)rrkwargsrr= __class__rrrDIs ,2 zNamespace.__init__rrcCs t|ts td|dd|vrtd|dt|}tdd|Dr.td|dd d |D}|d }|}d }t|d krd|dd }|dd D]-}t||sbt|try||vry||}|durxt|t tfsx|d|fSqR|d|fS|||fS)aeParses a key for the nested namespace. Args: key: The key that is being parsed. Returns: Tuple with three elements corresponding to: - The leaf key. - The parent namespace object. - The parent namespace key. Raises: KeyError: When given invalid key. zKey must be a string, got: r zSpaces not allowed in keys: "z".css|]}|dkVqdS)Nrr2krrr msz'Namespace._parse_key..zEmpty nested key: "cSsg|]}t|qSr)add_clash_markrLrrrr4oz(Namespace._parse_key..r&rKr!N) r5rr ranyrBjoinhasattrr6r )rr key_splitr( parent_ns parent_keysubkeyrrr _parse_keyYs*   zNamespace._parse_keycCs<||\}}}|dust||std|d|||fS)z8Same as _parse_key but raises KeyError if key not found.NzKey "z" not found in namespace.)rXrSr )rrr(rUrVrrr_parse_required_key~s zNamespace._parse_required_keycCs@|}t|D]}tt||dtst||tt||}q|S)zCreates a nested namespace object. Args: key: The key where the nested namespace is created. Returns: The created nested namespace. N)rr5r8r setattr)rrrUrrr_create_nested_namespaces   z"Namespace._create_nested_namespacenamevalueNcs.d|vr |||dStt||dS)z1Sets an attribute to a possibly nested namespace.rN) __setitem__rC __setattr__rO)rr\r]rHrrr_szNamespace.__setattr__itemcCsH||\}}}|dur||}t|tr|||<dSt|||dS)z,Sets an item to a possibly nested namespace.N)rXr[r5r6rZ)rrr`r(rUrVrrrr^s    zNamespace.__setitem__cCs||\}}}t||S)z.Gets an item from a possibly nested namespace.)rYr8rrr(rU_rrr __getitem__s zNamespace.__getitem__cCs||\}}}|j|=dS)z1Deletes an item from a possibly nested namespace.N)rXr0rarrr __delitem__s zNamespace.__delitem__cCsBt|tsdSz ||\}}}Wn tyYdSw||jvS)z8Checks if an item is set possibly in a nested namespace.F)r5rrYKeyErrorr0rarrr __contains__s   zNamespace.__contains__cCs t|jS)z4Returns False if namespace is empty, otherwise True.)boolr0rrrr__bool__s zNamespace.__bool__cCsi}t|D]K\}}t|tr|}n7t|tr4|ikr4tdd|Dr4dd|D}nt|trM|gkrMtdd|DrMdd|D}||t |<q|S)z8Converts the nested namespaces into nested dictionaries.cs|]}t|tVqdSr>r5r r1rrrrNz$Namespace.as_dict..cSsi|] \}}||qSras_dict)r2rMr3rrr sz%Namespace.as_dict..csrir>rjr1rrrrNrkcSsg|]}|qSrrlr1rrrr4rPz%Namespace.as_dict..) rFr9r5r rmr6allvaluesr:del_clash_mark)rdicrr=rrrrms  ($zNamespace.as_dictcCs*t}|D] \}}t|||q|S)zEConverts the nested namespaces into a single argparse flat namespace.)r?r r9rZ)rflatrr=rrras_flatszNamespace.as_flatFTbranchesnestedccsxt|D]2\}}t|}t|tr4|r||fV|sq||D]\}}|dt||fVq#q||fVqdS)zRReturns a generator of all leaf (key, value) items, optionally including branches.rN)rFr9rqr5r )rrurvrr=rWsubvalrrrr9s   zNamespace.itemsccs$|j||dD]\}}|VqdS)zDReturns a generator of all leaf keys, optionally including branches.)rurvNr9)rrurvrrbrrrkeysszNamespace.keysccs ||D]\}}|VqdS)zFReturns a generator of all leaf values, optionally including branches.Nrx)rrurbr=rrrrpszNamespace.values key_filtercsfdd|D}|r:dd|DD]%}t|}tt|dD]}d|d|d}||vr8||q"q|jddd |S) zReturns a list of keys sorted by descending depth. Args: branches: Whether to include branch keys instead of only leaves. key_filter: Function that selects keys to exclude. csg|]}|s|qSrrrLrzrrr4z-Namespace.get_sorted_keys..cSsg|]}d|vr|qS)rrrLrrrr4r|r!rNcSstt| Sr>)rBr)xrrrsz+Namespace.get_sorted_keys..r)ryrrangerBrRappendsort)rrurzryrrTnumrVrr{rget_sorted_keyss zNamespace.get_sorted_keys with_metacCst||rddStdS)zCreates an new copy of the nested namespace. Args: with_meta: Whether to include metadata keys in the copy. Nr+r-)rrrrrcloneszNamespace.clone only_unsetcCst|ts|s td|r||vr|||<|S|r&t||ts&t||<|r,|dnd}|D]\}}|r>|||vrD||||<q2|S)aYSets or replaces all items from the given nested namespace. Args: value: A namespace to update multiple values or other type to set in a single key. key: Branch key where to set the value. Required if value is not namespace. only_unset: Whether to only set the value if not set in namespace. z)Key is required if value not a Namespace.rrK)r5r r getr9)rr]rrprefixrWrwrrrupdates    zNamespace.updatedefaultc Cs&z||WSttfy|YSwr>)re TypeError)rrrrrrrs  z Namespace.getcCs||\}}}||||fSr>)rYrarrrget_value_and_parent#szNamespace.get_value_and_parentcCs&||\}}}|s |S|j||Sr>)rXr0pop)rrrr(rUrbrrrr'sz Namespace.pop)FT)F)T)NFr>)'rrr__doc__rDrtuplerrXrYr[rr_r^rcrdrgrfrhr6rmr?r rtrr9ryrpr)rr:rrrrrrr __classcell__rrrHrr Fs@  %   $      clash_namesu​cCs|tvrt|}|Sr>)r clash_markrrrrrO2srOcCs|dtkr |dd}|S)Nrr!)rrrrrrq8s  rqr;cCs|D]=\}}t|trtdd|Drt|||<qt|trAt|D]\}}t|tr@tdd|Dr@t||||<q&qtdi|S)Ncsrir>r5rrLrrrrN@rkzexpand_dict..csrir>rrLrrrrNDrkr)r9r5r6ro expand_dictr: enumerater )r;rMr3nnvvrrrr>s rcCst|}t|Sr>)r.r)r;rrrdict_to_namespaceIsr) find_specyamlcCs|d|S)Nztag:yaml.org,2002:map)represent_mappingrm)dr}rrrr~Tsr~r>)+rr? collectionsrcollections.abcr contextlibrtypingrrrr__all__r'rer rr:rr"r%rgr)r6r/r.rAr setdirr__annotations__rrOrqrrimportlib.utilrr SafeDumperadd_representerrrrrs<      i