o )Ti+@sUdZddlZddlmZddlmZddlmZddlm Z ddl m Z m Z m Z dd lmZdd lmZmZdd lmZmZmZmZdd lmZmZd gZe didZe ed<dedefddZ d0de eefdede e e!ee"e!edffde"e ee effddZ# d0de eefdede e e!ee"e!edffde efddZ$ d0dedede e e!ee"e!edffde"e ee effddZ% d0dedede e e!ee"e!edffde efddZ&Gd d d eZ'ed!d"Z( # $d1ded%ed&ed'ede"e e)ee e)eff d(d)Z* # $d1ded%ed&ed'ede"e ee eff d*d+Z+ # $d1ded%ed,e ed-ed&ed'eddfd.d/Z,dS)2z(Subcommands action and helper functions.N)Action)_SubParsersAction)contextmanager) ContextVar)NoReturnOptionalUnion)filter_non_parsing_actions)parsing_defaultssingle_subcommand) Namespace NSKeyError split_keysplit_key_root)ActionsContainerArgumentParserActionSubCommands parse_kwargs)defaultkeyreturncCsjt|d}t|jD]'}t|tr'||jvr'|j|}t|t|dS|j|dr2dSq dS)Nrr .TF) rr _actions isinstancer_name_parser_map is_branch_keydest startswith)parserrroot_keyaction subparserr#M/home/ubuntu/.local/lib/python3.10/site-packages/jsonargparse/_subcommands.pyrs  rrrexclude.cst|j}durfdd|D}|D]B}t|tsq||jvr'|dfSt|d}||jvrVt|\}}|j|}t||d\} } | durP|d| 7}| |fSqd} |D]'}|j|ksjd||jvrdd lm } m } t|| | fr||} q[|dfSq[| dfS) z6Finds an action in a parser given its destination key.Ncsg|] }t|s|qSr#)r).0ar%r#r$ (sz.find_action_and_subcommand..rr(rz--r ) ActionFail_ActionConfigLoad) r rrrrrfind_action_and_subcommandroption_stringsr*r+)rrr%actionsr! root_dest subcommandsubdestr" subaction subsubcommandfallback_actionr*r+r#r(r$r, s4           r,cCt|||ddSNr(r)r,)rrr%r#r#r$ find_actionFr7c Cst|||d\}}|dur=d|vr=t|}ttt|dD]}t|d|d|d|d\}}|dur<||fSq||fS)Nr(rr )r,rreversedrangelenjoin)rrr%r!r0partsnr#r#r$!find_parent_action_and_subcommandNs$r?cCr5r6)r?)rrr%r#r#r$find_parent_action]r8r@c@sTeZdZUdZeed<eed<dedefddZdededefd d Z dd d Z d S)rzLExtension of argparse._SubParsersAction to modify subcommands functionality. parent_parser env_prefixnamercKstd)zLRaises a ``NotImplementedError`` since jsonargparse uses ``add_subcommand``.zFIn jsonargparse subcommands are added using the add_subcommand method.)NotImplementedError)selfrCkwargsr#r#r$ add_parserkszActionSubCommands.add_parserrcKs|jdur td|j|krtd|d|jd||_|j|d|_|jj|_|j|_|jj|_|jj |_ |jj |_ |jj |_ ||_ | dd}d}d |vrY| d }||||}|j|||j|<|D]}||j|<qm|S) aAdds a parser as a subcommand parser. In contrast to `argparse.ArgumentParser.add_subparsers `_ add_parser requires to be given a parser as argument. Args: name: The name for the subcommand. parser: The parser to use for the subcommand. NzA subcommand name can't be the same as the subcommands dest: 'z'.z [options] _aliasesr#help) _subparsers ValueErrorr _prog_prefixprogrBrA default_env parser_mode_error_handler exit_on_errorloggerr0pop_ChoicesPseudoAction_choices_actionsappendr)rErCrrFrIhelp_arg choice_actionaliasr#r#r$add_subcommandos.            z ActionSubCommands.add_subcommandNc Cs|d}|dd}|||j<||jvrA|j|}||vr$||nd}tddddt} |j|fd|i| ||<dSdS)z5Adds subcommand dest and parses subcommand arguments.rr NT)_skip_validation_namespace_as_config namespacer#)rrgetclonedictr parse_args) rErr^values option_stringr0 arg_stringsr" subnamespacerFr#r#r$__call__s    zActionSubCommands.__call__N) __name__ __module__ __qualname____doc__r__annotations__strrrGr[rgr#r#r#r$res )ccst|dVdSrh)rset)rFr#r#r$parse_kwargs_contexts  rpTcfgprefixfail_no_subcommandc s|jdurdS|jtot }fddjD}dj}|vrA|durA|tr@tddn*t|dkrk|sK|rk|d|<t|dkrkt d d |d |d d rt|dkrfdd|DD]}|=q|rg}|rˈdur|rj sdSj rˈj vrt j }t|dkrdd |d} n dd |ddd} td|d| d|fdd|DfS)z6Returns subcommand names and corresponding subparsers.N)NNcs$g|]}t|tr|qSr#)rr_r r&k)rrrsr#r$r)s$z#get_subcommands..z:A specific subcommand can't be provided in defaults, got ''rr z'Multiple subcommand settings provided (z, z) without an explicit 'z' key. Subcommand 'z' will be used.csg|]}|kr|qSr#r#ru)r0r#r$r){,}z, ...}z expected "z" to be one of z, but it was not provided.csg|]}j|qSr#)rr_)r&s)r!r#r$r)rx)_subcommands_actionr r_r choicesrrr;warningswarnr< _requiredrlist) rrrrsrtrequire_singlesubcommand_keysrravailable_subcommandscandidate_subcommands_strr#)r!rrrsr0r$get_subcommandssJ      rcCs6t||||d\}}|r|dnd|r|dfSdfS)z=Returns a single subcommand name and corresponding subparser.rsrtrN)r)rrrrsrt subcommands subparsersr#r#r$get_subcommands "renvdefaultsc Cst||||d\}}|r|sdSt||D]>\}} d} ||} |r)| j|dd} n|r1| jdd} | durB| || t| || <| jdurSt| |||| d|dqdS)z(Takes care of parsing subcommand values.rNT)rr\)skip_validationr)rt) rzip parse_env get_defaults merge_configr_r rKhandle_subcommands) rrrrrrsrtrrr0r"rfrr#r#r$rs    rrh)rqT)-rlrargparserArgparseActionr contextlibr contextvarsrtypingrrrrr _commonr r _namespacer rrr_type_checkingrr__all__rrmrnboolrtypetupler,r7r?r@rrprrrrr#r#r#r$s       )     C  ;