首先说明,本人是想要学习如何使用argparse模块,打造命令行程序。
但是网络上的各种教程写的只能说是惨不忍睹,反人类来形容。这样的文章写出来纯粹浪费大家时间。
找了很久,找到了python官方推荐的入门教程,看下来非常好。但是没有中文版。
本人尝试根据该教程,将argparse的核心使用方法做一介绍。
注:代码均命名为 1.py.
代码如下:
import argparse
parser = argparse.ArgumentParser(description="程序的主要功能是...")
parser.parse_args()
执行代码:
python 1.py --help
执行结果:
usage: 1.py [-h]
程序的主要功能是...
optional arguments:
-h, --help show this help message and exit
--help选项就是将命令行参数说一个说明。
-h 是 --help的缩写形式,二者效果相同。
如程序中展示的,可以通过description参数来对程序的主要功能做出说明。
代码如下:
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("echo")
args = parser.parse_args()
print('参数echo的值是{}'.format(args.echo))
执行代码:
python 1.py 'hello'
执行结果:
参数echo的值是hello
以上代码就是添加一个参数,存储在名字为'echo'的变量里面。
注意,要获取变量的值,需要先通过parse_args()方法获取args对象。
所以命令行信息都可以通过args对象取值获得。
注意,由于这里指定了位置参数,也就是说,如果执行程序时没有携带参数,则会报错。
执行代码:
python 1.py 'hello'
usage: 1.py [-h] echo
1.py: error: the following arguments are required: echo
代码如下:
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("echo", help="我是关于echo的说明")
args = parser.parse_args()
print('参数echo的值是{}'.format(args.echo))
执行-h选项:
python 1.py -h
执行结果:
positional arguments:
echo 我是关于echo的说明
optional arguments:
-h, --help show this help message and exit
可以看见,-h的执行结果中,已经打印了代码中为位置参数echo添加的说明。
代码如下:
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("square", type=int)
args = parser.parse_args()
print(args.square**2)
执行代码:
python 1.py 123
执行结果:
15129
注意这里的代码指定了一个位置参数square,并且指定类型为int。
默认情况下,如果没有指定类型,argparse会将参数作为字符串存储。
也就是说,,如果没有指定类型,args.square的值是字符串'123'。
此时,在执行 args.square**2 的操作时,会报错。
先看代码:
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("--verbosity")
args = parser.parse_args()
print('可选参数verbosity的值是{}'.format(args.verbosity))
执行代码:
python 1.py --verbosity 123
执行结果:
可选参数verbosity的值是123
注意,在参数前加上前缀--,即意味着这个参数是可选参数。
可选参数与位置参数有两点不同:
第一,可选参数可以不指定,如果这样执行代码:
python 1.py
则执行结果为:
可选参数verbosity的值是None
注意此时的,verbosity的值是None。
第二点不同,可选参数使用时,需要先声明该参数,再指定该参数的值。
正如上面代码中使用的那样。读者可以自行体会。
比如说我们想要设定一个参数,如果指定了该参数,就执行一个功能,
如果没指定,则不执行。具体该参数的值是多少在所不问。
代码如下:
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("--verbose", action="store_true")
args = parser.parse_args()
print("参数verbose的值是{}".format(args.verbose))
执行代码:
python 1.py --verbose
执行结果:
参数verbose的值是True
可以看到,在执行代码时,指定了可选参数--verbose,但是没有指定verbose的值。
此时verbose的值就默认为布尔值True。
实现这一目的的方法是将action参数的值指定为 "store_true"。
通过这个特点,就可以实现使用某一个特定功能这个目的。
上面我们可以注意到,-h 是 --help 和简单形式,二者功能相同。
我们也可以为自己设定的参数指定简短的形式。
代码为:
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("-v","--verbose", action="store_true")
args = parser.parse_args()
print("参数verbose的值是{}".format(args.verbose))
执行代码:
python 1.py -v
执行结果:
参数verbose的值是True
说明,在可选参数前添加一个简短的参数,与直接使用原参数效果相同。
代码如下:
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("square", type=int)
parser.add_argument("-v", "--verbose", action="store_true")
args = parser.parse_args()
if args.verbose:
print("the square of {} equals {}".format(args.square, args.square**2))
else:
print(args.square**2)
执行方式1:
python 1.py 50
执行结果1:
2500
执行方式2:
python 1.py 50 -v
执行结果2:
the square of 50 equals 2500
通过多次使用add_argument方法,来添加多个命令行参数。
代码为:
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("-v", "--verbosity", type=int, choices=[0, 1, 2])
args = parser.parse_args()
print("可选参数verbosity的值为{}".format(args.verbosity))
多次不同方式执行代码:
MacBook-Pro:Desktop yao$ python 1.py
可选参数verbosity的值为None
MacBook-Pro:Desktop yao$ python 1.py -v 1
可选参数verbosity的值为1
MacBook-Pro:Desktop yao$ python 1.py -v 2
可选参数verbosity的值为2
MacBook-Pro:Desktop yao$ python 1.py -v 3
usage: 1.py [-h] [-v {0,1,2}]
1.py: error: argument -v/--verbosity: invalid choice: 3 (choose from 0, 1, 2)
说明,通过设置choices属性,来设置可选参数的取值范围。
如果不在取值范围内的话,会报错。
代码如下:
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("-v", "--verbosity", action="count")
args = parser.parse_args()
print("可选参数verbosity的值为{}".format(args.verbosity))
以不同方式多次执行:
MacBook-Pro:Desktop yao$ python 1.py -v
可选参数verbosity的值为1
MacBook-Pro:Desktop yao$ python 1.py -vv
可选参数verbosity的值为2
MacBook-Pro:Desktop yao$ python 1.py -vvv
可选参数verbosity的值为3
MacBook-Pro:Desktop yao$ python 1.py -vvvv
可选参数verbosity的值为4
MacBook-Pro:Desktop yao$ python 1.py -vvvvv
可选参数verbosity的值为5
MacBook-Pro:Desktop yao$ python 1.py
可选参数verbosity的值为None
说明,通过设置action属性值为 'count' ,可以监控使用了多少次可选参数。
读者可自行根据执行结果感受该参数的作用。
代码为:
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("-v", "--verbosity", default=0)
args = parser.parse_args()
print("可选参数verbosity的值为{}".format(args.verbosity))
不同方式多次执行:
MacBook-Pro:Desktop yao$ python 1.py
可选参数verbosity的值为0
MacBook-Pro:Desktop yao$ python 1.py -v 100
可选参数verbosity的值为100
注意此时的默认值为0,而不是None
冲突参数是指,两个参数只能二选其一。
代码如下:
import argparse
parser = argparse.ArgumentParser()
group = parser.add_mutually_exclusive_group()
group.add_argument("-v", "--verbose", action="store_true")
group.add_argument("-q", "--quiet", action="store_true")
args = parser.parse_args()
print("可选参数verbosity的值为{}".format(args.verbose))
print("可选参数quiet的值为{}".format(args.quiet))
不同方式多次执行:
MacBook-Pro:Desktop yao$ python 1.py
可选参数verbosity的值为False
可选参数quiet的值为False
MacBook-Pro:Desktop yao$ python 1.py -v
可选参数verbosity的值为True
可选参数quiet的值为False
MacBook-Pro:Desktop yao$ python 1.py -q
可选参数verbosity的值为False
可选参数quiet的值为True
MacBook-Pro:Desktop yao$ python 1.py -v -q
usage: 1.py [-h] [-v | -q]
1.py: error: argument -q/--quiet: not allowed with argument -v/--verbose
MacBook-Pro:Desktop yao$ python 1.py -h
usage: 1.py [-h] [-v | -q]
optional arguments:
-h, --help show this help message and exit
-v, --verbose
-q, --quiet
请读者根据执行结果自行思考冲突参数的设置以及特点。