问题描述
我已经为此苦苦了很长时间,所以我想为另一个命令做子命令,该命令将忽略与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()
的“良好”格式
注意:我想使用基本库,所以我将继续使用argparse:v
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)