我有一个小脚本,我需要它能够接受带有值和值的参数。
./cha.py --pretty-xml
./cha.py --pretty-xml=5
我有这个。
parser.add_argument('--pretty-xml', nargs='?', dest='xml_space', default=4)
但是当我在xml_space中使用--pretty-xml时将是'none'。如果我不在xml_space中写入此参数则存储默认值。我需要完全相反的。
答案 0 :(得分:4)
使用const
关键字:
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("--pretty-xml", nargs="?", type=int, dest="xml_space", const=4)
print(parser.parse_args([]))
print(parser.parse_args(['--pretty-xml']))
print(parser.parse_args(['--pretty-xml=5']))
结果
Namespace(xml_space=None)
Namespace(xml_space=4)
Namespace(xml_space=5)
答案 1 :(得分:0)
取消default
参数并改为使用自定义Action
:
class PrettyXMLAction(argparse.Action):
def __call__(self, parser, namespace, values, option_string=None):
if not values:
values = 4
setattr(namespace, self.dest, values)
parser.add_argument('--pretty-xml', nargs='?', type=int, dest='xml_space', action=PrettyXMLAction)
演示:
>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--pretty-xml', nargs='?', type=int, dest='xml_space', action=PrettyXMLAction)
PrettyXMLAction(option_strings=['--pretty-xml'], dest='xml_space', nargs='?', const=None, default=None, type=None, choices=None, help=None, metavar=None)
>>> parser.parse_args('--pretty-xml'.split())
Namespace(xml_space=4)
>>> parser.parse_args('--pretty-xml=5'.split())
Namespace(xml_space=5)
>>> parser.parse_args(''.split())
Namespace(xml_space=None)