我使用以下简单代码来解析一些参数;请注意,其中之一是必需的。不幸的是,当用户在不提供参数的情况下运行脚本时,显示的用法/帮助文本并不表示存在非可选参数,我觉得这很混乱。如何让 python 指示参数不是可选的?
这是代码:
import argparse
if __name__ == '__main__':
parser = argparse.ArgumentParser(
description='Foo')
parser.add_argument('-i','--input', help='Input file name', required=True)
parser.add_argument('-o','--output', help='Output file name', default="stdout")
args = parser.parse_args()
print ("Input file: %s" % args.input )
print ("Output file: %s" % args.output )
在不提供所需参数的情况下运行上述代码时,我得到以下输出:
usage: foo.py [-h] -i INPUT [-o OUTPUT]
Foo
optional arguments:
-h, --help show this help message and exit
-i INPUT, --input INPUT
Input file name
-o OUTPUT, --output OUTPUT
Output file name
-i INPUT
部分没有被方括号括起来,这微妙地表明确实是必需的。此外,您可以通过 help
参数手动解释
optional arguments
仍然具有误导性。
以 -
或 --
开头的参数通常被认为是可选的。所有其他参数都是位置参数,因此设计需要(如位置函数参数)。可能需要可选参数,但这有点违背他们的设计。由于它们仍然是非位置参数的一部分,即使它们是必需的,它们仍然会列在令人困惑的标题“可选参数”下。然而,用法部分中缺少的方括号表明它们确实是必需的。
另请参阅 documentation:
通常, argparse 模块假定 -f 和 --bar 等标志表示可选参数,在命令行中始终可以省略这些参数。注意:必需选项通常被认为是错误的形式,因为用户希望选项是可选的,因此应尽可能避免使用它们。
话虽如此,帮助中的标题“位置参数”和“可选参数”是由两个参数组生成的,其中参数自动分成。现在,您可以“破解它”并更改可选参数的名称,但更优雅的解决方案是为“必需的命名参数”(或您想调用的任何名称)创建另一个组:
parser = argparse.ArgumentParser(description='Foo')
parser.add_argument('-o', '--output', help='Output file name', default='stdout')
requiredNamed = parser.add_argument_group('required named arguments')
requiredNamed.add_argument('-i', '--input', help='Input file name', required=True)
parser.parse_args(['-h'])
usage: [-h] [-o OUTPUT] -i INPUT
Foo
optional arguments:
-h, --help show this help message and exit
-o OUTPUT, --output OUTPUT
Output file name
required named arguments:
-i INPUT, --input INPUT
Input file name
因为我更喜欢在可选参数之前列出必需的参数,所以我通过以下方式解决它:
parser = argparse.ArgumentParser()
parser._action_groups.pop()
required = parser.add_argument_group('required arguments')
optional = parser.add_argument_group('optional arguments')
required.add_argument('--required_arg', required=True)
optional.add_argument('--optional_arg')
return parser.parse_args()
这输出:
usage: main.py [-h] --required_arg REQUIRED_ARG [--optional_arg OPTIONAL_ARG]
required arguments:
--required_arg REQUIRED_ARG
optional arguments:
--optional_arg OPTIONAL_ARG
我可以不用 -h, --help
出现在可选参数组中。
以@Karl Rosaen 为基础
parser = argparse.ArgumentParser()
optional = parser._action_groups.pop() # Edited this line
required = parser.add_argument_group('required arguments')
# remove this line: optional = parser...
required.add_argument('--required_arg', required=True)
optional.add_argument('--optional_arg')
parser._action_groups.append(optional) # added this line
return parser.parse_args()
这输出:
usage: main.py [-h] [--required_arg REQUIRED_ARG]
[--optional_arg OPTIONAL_ARG]
required arguments:
--required_arg REQUIRED_ARG
optional arguments:
-h, --help show this help message and exit
--optional_arg OPTIONAL_ARG
_action_group
?就我而言,我需要向已经存在的(自定义)组添加一些参数。
再一次,以@RalphyZ 为基础
这不会破坏暴露的 API。
from argparse import ArgumentParser, SUPPRESS
# Disable default help
parser = ArgumentParser(add_help=False)
required = parser.add_argument_group('required arguments')
optional = parser.add_argument_group('optional arguments')
# Add back help
optional.add_argument(
'-h',
'--help',
action='help',
default=SUPPRESS,
help='show this help message and exit'
)
required.add_argument('--required_arg', required=True)
optional.add_argument('--optional_arg')
这将显示与上面相同并且应该在未来的版本中继续存在:
usage: main.py [-h] [--required_arg REQUIRED_ARG]
[--optional_arg OPTIONAL_ARG]
required arguments:
--required_arg REQUIRED_ARG
optional arguments:
-h, --help show this help message and exit
--optional_arg OPTIONAL_ARG
_action_groups
仅供内部使用。因此,没有跨版本的兼容性保证。
注意:根据 Christophe Vu-Brugier,以下代码从 Python 3.10 版开始不起作用。
默认情况下,parser._action_groups 中有 2 个参数组:位置参数和命名参数(标题为“可选参数”)。您可以将命名的可选参数添加到现有的“可选参数”组,并将所需的命名参数添加到新的“必需参数”组。之后,您可以重新排序组:
import argparse
parser = argparse.ArgumentParser(description='Foo')
required = parser.add_argument_group('required arguments')
required.add_argument('-i','--input', help='Input file name', required=True)
parser.add_argument('-o','--output', help='Output file name', default="stdout")
groups_order = {
'positional arguments': 0,
'required arguments': 1,
'optional arguments': 2
}
parser._action_groups.sort(key=lambda g: groups_order[g.title])
parser.parse_args(['-h'])
输出:
usage: argparse_argument_groups.py [-h] -i INPUT [-o OUTPUT]
Foo
required arguments:
-i INPUT, --input INPUT
Input file name
optional arguments:
-h, --help show this help message and exit
-o OUTPUT, --output OUTPUT
Output file name
您不需要覆盖可选组。
做就是了:
parser = argparse.ArgumentParser()
required = parser.add_argument_group('required arguments')
required.add_argument('--required_arg', required=True)
# All arguments set via parser directly will automatically go to the optional group
parser.add_argument('--optional_arg')
parser.print_help()
将打印出来
usage: [-h] --required_arg REQUIRED_ARG [--optional_arg OPTIONAL_ARG]
optional arguments:
-h, --help show this help message and exit
--optional_arg OPTIONAL_ARG
required arguments:
--required_arg REQUIRED_ARG
如果您希望在可选参数之前有必需的参数,您可以执行以下操作:
parser = argparse.ArgumentParser()
optional = parser._action_groups.pop()
required = parser.add_argument_group('required arguments')
parser._action_groups.append(optional)
required.add_argument('--required_arg', required=True)
optional.add_argument('--optional_arg')
parser.print_help()
将以正确的顺序打印组:
usage: [-h] --required_arg REQUIRED_ARG [--optional_arg OPTIONAL_ARG]
required arguments:
--required_arg REQUIRED_ARG
optional arguments:
-h, --help show this help message and exit
--optional_arg OPTIONAL_ARG
_action_groups
,这会使您面临在次要版本升级中中断的严重风险。我会坚持@Karl Rosaen 的回答。
_action_groups
之外的所有(有用的)答案。
使用 argparse 时,所需的参数通常是“位置”参数。
parser = argparse.ArgumentParser(description="Foo")
parser.add_argument("username")
parser.add_argument("password")
这将添加两个必需的位置参数。
不定期副业成功案例分享
parser.parse_args([])
相反,使用不带参数的parser.parse_args()
来捕获 sys.argv 的内容。每argparse--
和一个精确的名称来定义,以解释它们的用途。然后可以使用单个-
为常用选项定义一个短别名。在我的示例中,您可以将--output out.txt
或-o out.txt
用于完全相同的事情。-o
只是--output
的简称。一些命令行工具还允许您链接这些短别名。例如,tar -xf archive.tar
是tar --extract --file=archive.tar
的缩写。