[ #gtddlZddlZddlZddlmZmZmZmZmZGddejZ GddejZ y)N)ListOptionalSequenceAnyUnionc eZdZdZd fd Z d dej dejdee e e dfde e ddf d Z xZS) HelpfulSubparserActionz^Class to either select a unique plugin based on a substring, or identify the alternatives.returnNc$t||i|yN)super__init__)selfargskwargs __class__s M/home/panchajanya/Development/work/volatility3/volatility3/cli/volargparse.pyrzHelpfulSubparserAction.__init__s $)&)parser namespacevalues option_stringcZd}g}||D] }|s|}||gz }|jtjk7rt||j||jDcgc] }||vs| } }t | dkr8d|ddj |jd} tj|| t | dkDr.d|ddj | d} tj|| |j| d }t|d | d |j|d\} }t| jD]\} }t|| ||rSt|jtjgt|tjj|yycc}w) Nzinvalid choice z (choose from , )zplugin z matches multiple plugins (rplugin)destargparseSUPPRESSsetattr_name_parser_maplenjoin ArgumentErrorparse_known_argsvarsitems setdefault_UNRECOGNIZED_ARGS_ATTRgetattrextend) rrrrr parser_name arg_stringsvaluenamematched_parsersmsg subnamespacekeys r__call__zHelpfulSubparserAction.__call__s   ""'KE7*K   99)) ) Ityy+ 6"22 kT6ID    ! ##K=tyyI^I^?_>``abC((s3 3  ! #K=(CDIIoD^C__`aC((s3 3&&q'9: 8_Q%78%+$;$;K$N! k|,224JC IsE *5  O & &x'G'G L Ix?? @ G G T 1 s  F(%F()r Nr )__name__ __module__ __qualname____doc__rr ArgumentParser Namespacerstrrrrr6 __classcell__rs@rr r sm*(, .U''.U%%.Uc8C=$./ .U  } .U  .Urr cLeZdZdefdZdej deddffd ZxZ S)HelpfulArgParserr c^|j|}tj||}|dtjdtj tjdtj tjdi}|j|j}|.tjdd|j|jz}|jr!|ddj|jd}t j||t|jd S) Nzexpected one argumentzexpected at most one argumentzexpected at least one argumentzexpected %s argumentzexpected %s argumentsz (from: rrr)_get_nargs_patternrematchgettextr OPTIONAL ONE_OR_MOREgetnargsngettextchoicesr%r&r$group)ractionarg_strings_pattern nargs_patternrE nargs_errorsr3s r_match_argumentz HelpfulArgParser._match_argumentKs//7 (;< =goo&=>!!7??3R#S$$goo6V&WL ""6<<0C{$$.0Gll# ~~Xdii&?%@B((5 55;;q>""rrNr0NcFt|tst| ||y)a_This is called to ensure a value is correct/valid In normal operation, it would check that a value provided is valid and return None If it was not valid, it would throw an ArgumentError When people provide a partial plugin name, we want to look for a matching plugin name which happens in the HelpfulSubparserAction's __call_method To get there without tripping the check_value failure, we have to prevent the exception being thrown when the value is a HelpfulSubparserAction. This therefore affects no other checks for normal parameters. N) isinstancer r _check_value)rrNr0rs rrUzHelpfulArgParser._check_valuefs"&"89 G  /r) r7r8r9intrRr ActionrrUr>r?s@rrArAJs2#c#68??34rrA) r rFrDtypingrrrrr_SubParsersActionr r;rArrr[s=  775UX775Up+x..+r