argparse模块入门介绍—

转载：https://blog.csdn.net/weixin_41796207/article/details/80846406

首先说明，本人是想要学习如何使用argparse模块，打造命令行程序。

但是网络上的各种教程写的只能说是惨不忍睹，反人类来形容。这样的文章写出来纯粹浪费大家时间。

找了很久，找到了python官方推荐的入门教程，看下来非常好。但是没有中文版。

本人尝试根据该教程，将argparse的核心使用方法做一介绍。

注：代码均命名为 1.py.

1. 命令行程序框架

代码如下:

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参数来对程序的主要功能做出说明。

2. 添加一个位置参数

代码如下:

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

3. 为位置参数添加说明

代码如下：

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添加的说明。

4. 指定参数类型

代码如下：

import argparse
parser = argparse.ArgumentParser()
parser.add_argument("square", type=int)
args = parser.parse_args()
print(args.square**2)

执行代码：

python 1.py 123

执行结果：

注意这里的代码指定了一个位置参数square，并且指定类型为int。

默认情况下，如果没有指定类型，argparse会将参数作为字符串存储。

也就是说，，如果没有指定类型，args.square的值是字符串'123'。

此时，在执行 args.square**2 的操作时，会报错。

5. 指定可选参数

先看代码：

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。

第二点不同，可选参数使用时，需要先声明该参数，再指定该参数的值。

正如上面代码中使用的那样。读者可以自行体会。

6. 设定无需具体值参数

比如说我们想要设定一个参数，如果指定了该参数，就执行一个功能，

如果没指定，则不执行。具体该参数的值是多少在所不问。

代码如下：

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"。

通过这个特点，就可以实现使用某一个特定功能这个目的。

7. 设定参数的简短形式

上面我们可以注意到，-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

说明，在可选参数前添加一个简短的参数，与直接使用原参数效果相同。

8. 同时添加位置参数和可选参数

代码如下：

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：

执行方式2：

python 1.py 50 -v

执行结果2：

the square of 50 equals 2500

通过多次使用add_argument方法，来添加多个命令行参数。

9. 为可选参数设置取值范围

代码为：

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属性，来设置可选参数的取值范围。

如果不在取值范围内的话，会报错。

10. 监控可选参数的执行次数

代码如下：

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' ，可以监控使用了多少次可选参数。

读者可自行根据执行结果感受该参数的作用。

11. 设置参数的默认值

代码为：

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

12 设置冲突参数

冲突参数是指，两个参数只能二选其一。

代码如下：

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

请读者根据执行结果自行思考冲突参数的设置以及特点。