子命令的Python argparse顺序和默认值

如何解决子命令的Python argparse顺序和默认值

我已经为此苦苦了很长时间，所以我想为另一个命令做子命令，该命令将忽略与subparsers = parser.add_subparsers()相反的顺序，因为可以在参数末尾执行子解析器还是一个人。

例如test.py通过SSH连接到设备并执行一些命令，我可以在终端中看到输出，但是就像我想保存它一样。作为filename.test_1（默认名称为command_output）、. txt或.csv（默认格式为txt）。

我希望我可以以各种可能的方式制作--save（例如，仅在-n之后才执行-f和--save-> error: argument -f/--format: can't be used before --save）：

python3 test.py --save
python3 test.py --save -n file_name
python3 test.py --save -n file_name -f csv
python3 test.py --save -n file_name --command "cat /etc/passwd"
python3 test.py --command "cat /etc/passwd" --save -f csv

我基于此代码，但无法将其重写为自己的需求。问题之一是--save不是store_true参数：

#!/usr/bin/python3

import argparse
from collections import OrderedDict

def open_gui():
    print("GUI has been opened.")
    return 0

def test_information():
    print("Some information.")
    return 0

class ParentAction(argparse.Action):
    def __init__(self,*args,**kwargs):
        super().__init__(*args,default=OrderedDict(),**kwargs)

        self.children = []

    def __call__(self,parser,namespace,values,option_string=None):
        items = getattr(namespace,self.dest)
        nspace = type(namespace)()
        for child in self.children:
            if child.default is not None:
                setattr(nspace,child.name,child.default)
        items[values] = nspace

class ChildAction(argparse.Action):
    def __init__(self,parent,sub_action='store',**kwargs)

        self.dest,self.name = parent.dest,self.dest
        self.action = sub_action
        self._action = None
        self.parent = parent

        parent.children.append(self)

    def get_action(self,parser):
        if self._action is None:
            action_cls = parser._registry_get('action',self.action,self.action)
            self._action = action_cls(self.option_strings,self.name)
        return self._action

    def __call__(self,self.dest)
        try:
            last_item = next(reversed(items.values()))
        except stopiteration:
            raise argparse.ArgumentError(self,"can't be used before {}".format(self.parent.option_strings[0]))
        action = self.get_action(parser)
        action(parser,last_item,option_string)

def main(command_line=None):

    print('')

    parser = argparse.ArgumentParser(prog='SSHtest',description='It is example description',add_help=False)

    single_group = parser.add_argument_group('single use arguments')
    single_group_exception = parser.add_argument_group('exception for single use arguments')
    multi_group = parser.add_argument_group('multiple use arguments')

    single_group.add_argument('--help','-h',action='help',help='show this help message and exit')
    single_group.add_argument('--version','-v',action='version',version='%(prog)s 1.0.1a201009')
    single_group.add_argument('--information','-i',action='store_true',help='show alias information and exit')
    single_group.add_argument('--gui',help='open a GUI application from default web browser and exit')

    multi_group.add_argument('--debug',help='print debug info')
    multi_group.add_argument('--command',type=str,help='execute custom command and exit')

    parent = parser.add_argument('-s','--save',action=ParentAction)
    parser.add_argument('-n','--name',action=ChildAction,parent=parent,default='command_output',Metavar='NAME',help='set file name (default: %(default)s)')
    parser.add_argument('-f','--format',choices=['txt','csv','xlsx'],default='txt',Metavar='FORMAT',help='set format of file,default is %(default)s (%(choices)s)')

    args = parser.parse_args(command_line)

    if args.information:
        test_information()
        parser.exit()

    if args.debug:
        print("debug: " + str(args))

    if args.gui:
        open_gui()

if __name__ == '__main__':
    main()

也许有一个简单的方法？我还想保持--help

的“良好”格式

注意：我想使用基本库，所以我将继续使用argparse：v

子命令的Python argparse顺序和默认值

如何解决子命令的Python argparse顺序和默认值

相关推荐