以前
optparse
比较火,但是在python2.7版本后,官方不再维护这个模块,转而大力支持argparse
。
argparse
模块可以让人轻松编写用户友好的命令行接口。她可以从 sys.argv
中解析出参数,并自动生成帮助和使用手册,还能在传入无效参数时报出错误信息。
基础用法
由于是标准库,所以不需要安装,直接使用:
import argparse
parser = argparse.ArgumentParser()
# parser.add_argument("-i", "--info", action = "store", type = "string", dest = "sample_info", help = "the sample information")
args = parser.parse_args()
# print(args.sample_info)
运行以上代码,如下:
- 不输入参数,也没有输出结果:
$ python3 test.py
- 输入默认参数
--help
(也可缩写为-h
),会输出自动生成的帮助信息:$ python3 test.py --help # 输出: usage: test.py [-h] optional arguments: -h, --help show this help message and exit
- 输入错误参数,会输出报错信息:
$ python3 test.py -verbose # 输出: usage: test.py [-h] test.py: error: unrecognized arguments: --verbose
创建解析器
使用 argparse
的第一步是创建一个 ArgumentParser
对象,如下:
import argparse
parser = argparse.ArgumentParser(prog=None, usage=None,description=None,
epilog=None, parents=[], formatter_class=argparse.HelpFormatter, prefix_chars='-', fromfile_prefix_chars=None,
argument_default=None, conflict_handler='error', add_help=True, allow_abbrev=True)
创建一个新的 ArgumentParser
对象。所有的参数都应当作为关键字参数传入。部分参数在下面都有它更详细的描述,但简而言之,它们是:
-
prog
- 程序的名称(默认:sys.argv[0]) -
usage
- 描述程序用途的字符串(默认值:从添加到解析器的参数生成) -
description
- 在参数帮助文档之前显示的文本(默认值:无) -
epilog
- 在参数帮助文档之后显示的文本(默认值:无) -
parents
- 一个 ArgumentParser 对象的列表,它们的参数也应包含在内 -
formatter_class
- 用于自定义帮助文档输出格式的类 -
prefix_chars
- 可选参数的前缀字符集合(默认值:'-') -
fromfile_prefix_chars
- 当需要从文件中读取其他参数时,用于标识文件名的前缀字符集合(默认值:None) -
argument_default
- 参数的全局默认值(默认值: None) -
conflict_handler
- 解决冲突选项的策略(通常是不必要的) -
add_help
- 为解析器添加一个 -h/--help 选项(默认值: True) -
allow_abbrev
- 如果缩写是无歧义的,则允许缩写长选项 (默认值:True)
prog
默认情况下, ArgumentParser
对象在帮助消息中显示的程序名称是 sys.argv[0]
。若要自定义程序名,则使用 prog
,如下:
>>> parser = argparse.ArgumentParser(prog='myprogram')
>>> parser.print_help()
usage: myprogram [-h]
optional arguments:
-h, --help show this help message and exit
无论是从 sys.argv[0]
或是从 prog=
参数确定的程序名称,都可以在帮助消息里通过 %(prog)s
格式串来引用,如下:
>>> parser = argparse.ArgumentParser(prog='myprogram')
>>> parser.add_argument('--foo', help='foo of the %(prog)s program')
>>> parser.print_help()
usage: myprogram [-h] [--foo FOO]
optional arguments:
-h, --help show this help message and exit
--foo FOO foo of the myprogram program
usage
构建用法消息,如下:
>>> parser = argparse.ArgumentParser(prog='PROG', usage='%(prog)s [options]')
>>> parser.add_argument('--foo', nargs='?', help='foo help')
>>> parser.add_argument('bar', nargs='+', help='bar help')
>>> parser.print_help()
usage: PROG [options]
positional arguments:
bar bar help
optional arguments:
-h, --help show this help message and exit
--foo [FOO] foo help
description
简要描述这个程度做什么以及怎么做。在帮助消息中,这个描述会显示在命令行用法字符串和各种参数的帮助消息之间,如下:
>>> parser = argparse.ArgumentParser(description='A foo that bars')
>>> parser.print_help()
usage: argparse.py [-h]
A foo that bars
optional arguments:
-h, --help show this help message and exit
epilog
一些程序喜欢在 description 参数后显示额外的对程序的描述,如下:
>>> parser = argparse.ArgumentParser(
... description='A foo that bars',
... epilog="And that's how you'd foo a bar")
>>> parser.print_help()
usage: argparse.py [-h]
A foo that bars
optional arguments:
-h, --help show this help message and exit
And that's how you'd foo a bar
add_help
是否关闭自动生成的帮助信息,如下:
>>> parser = argparse.ArgumentParser(prog='PROG', add_help=False)
>>> parser.add_argument('--foo', help='foo help')
>>> parser.print_help()
usage: PROG [--foo FOO]
optional arguments:
--foo FOO foo help
添加参数
使用 add_argument()
方法添加参数。其中,部分参数在下面都有它更详细的描述,但简而言之,它们是:
-
name or flags
- 一个命名或者一个选项字符串的列表,例如 foo 或 -f, --foo。 -
action
- 当参数在命令行中出现时使用的动作基本类型。 -
nargs
- 命令行参数应当消耗的数目。 -
const
- 被一些 action 和 nargs 选择所需求的常数。 -
default
- 当参数未在命令行中出现时使用的值。 -
type
- 命令行参数应当被转换成的类型。 -
choices
- 可用的参数的容器。 -
required
- 此命令行选项是否可省略 (仅选项可用)。 -
help
- 一个此选项作用的简单描述。 -
metavar
- 在使用方法消息中使用的参数值示例。 -
dest
- 被添加到 parse_args() 所返回对象上的属性名。
name or flags
第一个传递给 add_argument()
的参数必须是一系列 flags 或者是一个简单的参数名,如下:
>>> parser.add_argument('-f', '--foo')
or
>>> parser.add_argument('bar')
action
-
store
- 存储参数的值。这是默认的动作。如下:
>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo')
>>> parser.parse_args('--foo 1'.split())
Namespace(foo='1')
-
store_const
- 存储被 const 命名参数指定的值。store_const
动作通常用在选项中来指定一些标志。如下:
>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo', action='store_const', const=42)
>>> parser.parse_args(['--foo'])
Namespace(foo=42)
-
store_true
和store_false
- 这些是store_const
分别用作存储True
和False
值的特殊用例。另外,它们的默认值分别为False
和True
。如下:
>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo', action='store_true')
>>> parser.add_argument('--bar', action='store_false')
>>> parser.add_argument('--baz', action='store_false')
>>> parser.parse_args('--foo --bar'.split())
Namespace(foo=True, bar=False, baz=True)
-
append
- 存储一个列表,并且将每个参数值追加到列表中。在允许多次使用选项时很有用。如下:
>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo', action='append')
>>> parser.parse_args('--foo 1 --foo 2'.split())
Namespace(foo=['1', '2'])
-
count
- 计算一个关键字参数出现的数目或次数。例如,对于一个增长的详情等级来说有用。如下:
>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--verbose', '-v', action='count')
>>> parser.parse_args(['-vvv'])
Namespace(verbose=3)
-
version
- 期望有一个version=
命名参数在add_argument()
调用中,并打印版本信息并在调用后退出。如下:
>>> import argparse
>>> parser = argparse.ArgumentParser(prog='PROG')
>>> parser.add_argument('--version', action='version', version='%(prog)s 2.0')
>>> parser.parse_args(['--version'])
PROG 2.0