Python命令行参数解析包argparse的使用详解

2022-09-14 10:18:52 作者：Se7eN_HOU

argparse 是 python 自带的命令行参数解析包，可以用来方便的服务命令行参数。本文将通过示例和大家详细讲讲argparse的使用，需要的可以参考一下

一、argparse简介

argparse 是 python 自带的命令行参数解析包，可以用来方便的服务命令行参数，使用之前需要先导入包 import argparse

二、简单案例

简单使用，创建一个名为test.py的文件

# 导入 argparse 模块
import argparse

# 创建一个argparse 的对象
parser = argparse.ArgumentParser()

# 添加一个接收参数，默认值为Se7eN
parser.add_argument("-n", "--name", default="Se7eN")

# 获取对应的参数的值
args = parser.parse_args()

# 控制台打印出来接收到的值
print("args = {}".format(args))

打印结果为：

houlei@houleideMacBook-Pro argparse_demo % python3 test.py   # 此时在test.py后面没有跟上参数，
args = Namespace(name='Se7eN')   # 所以打印出来的name为默认值 Se7eN

再次验证：

houlei@houleideMacBook-Pro argparse_demo % python3 test.py -n HOU   # 此时后面跟上了 -n 参数，并且赋值HOU
args = Namespace(name='HOU')  # 打印出来 name 的值为 HOU
houlei@houleideMacBook-Pro argparse_demo % python3 test.py --name hou  # 此时后面跟上了 --name 参数，并赋值为hou
args = Namespace(name='hou')  # 打印出来的 name 的值为 hou
houlei@houleideMacBook-Pro argparse_demo %

解释说明

在上述代码 parser.add_argument("-n", "--name", default="Se7eN") 中，有两个参数 “-n”和“--name” 分别代表什么？其中-n 和 --name 都是我们自己自定的参数名。至于n和name ,你可以随便起个变量名都可以，但是要注意，前面的一个 “-”和“--”才是关键。

一个“-”的参数，例如：-n, 他其实相当于我们在liunx中使用的指令，一个自定义的指令。

两个“--”的参数，例如：--name 他就相当于在程序中，用来接收对对应指令值的变量，例如我们在控制台上输入 python3 test.py -n hou。其中 -n 就代表使用的 -n 的指令，然后将后面的 hou 的值赋值给对应的变量， -n 对应的变量就是 --name 。所以我们使用print 打印的时候才显示的name = hou

简单理解：一个“-”的是指令，两个“--”的是接收指令内容的变量。

三、ArgumentParser参数

ArgumentParse是解析器对象，在创建一个解析器对象的时候，是有很多参数可以配置，下面就演示一下这些参数的使用

1、prog: 程序的名称（默认值：sys.argv[0]）

# 导入 argparse 模块
import argparse

# 创建一个argparse 的对象
parser = argparse.ArgumentParser()

# 添加一个接收参数，默认值为Se7eN
parser.add_argument("-n", "--name", default="Se7eN")

# 获取对应的参数的值
args = parser.parse_args()

# 控制台打印出来接收到的值
print("args = {}".format(args))

执行代码：

修改代码继续演示

# 导入 argparse 模块
import argparse

# 创建一个argparse 的对象
parser = argparse.ArgumentParser(prog="testname")    # 注意这里添加了prog=testname

# 添加一个接收参数，默认值为Se7eN
parser.add_argument("-n", "--name", default="Se7eN")

# 获取对应的参数的值
args = parser.parse_args()

# 控制台打印出来接收到的值
print("args = {}".format(args))

2、usage: 用来描述该程序的用途，并且会覆盖原来的prog的显示

修改部分程序：

# 创建一个argparse 的对象
parser = argparse.ArgumentParser(prog="testname", usage="这是功能描述模块")

展示结果

3、description:帮助文档的描述和uasge功能类似，只是显示的位置不一样

继续修改部分代码：

# 创建一个argparse 的对象
parser = argparse.ArgumentParser(prog="testname", usage="这是功能描述模块", description="这是程序的介绍，和usage的功能模块描述类似")

展示结果

4、epilog:帮助文档的额外说明，位置在最下面

继续修改部分代码：

# 创建一个argparse 的对象
parser = argparse.ArgumentParser(prog="testname", usage="这是功能描述模块", description="这是程序的介绍，和usage的功能模块描述类似", epilog="额外说明")

展示结果：

5、自定义指令的前缀

默认是"-" 和 "--"，这里如果你不喜欢使用，也可以自通过该参数自己指定前缀，个人觉得没必要。

代码演示：

# 导入 argparse 模块
import argparse

# 创建一个argparse 的对象 ，并通过prefix_chars 指定指令的前缀为 '+'
parser = argparse.ArgumentParser(prog="testname", usage="这是功能描述模块", description="这是程序的介绍，和usage的功能模块描述类似", epilog="额外说明", prefix_chars='+')

# 添加一个接收参数，默认值为Se7eN，这里的指令只能使用+ 表示，否则会报错
parser.add_argument("+n", "++name", default="Se7eN")

# 获取对应的参数的值
args = parser.parse_args()

# 控制台打印出来接收到的值
print("args = {}".format(args))

执行结果

四、add_argument指令参数解释

1、action: 的作用比较多，有保存布尔类型，常量类型，数组类型的参数的功能

注意1：使用action = store_true/store_false 时修饰的指令是不能赋值的，其对应的指令的值是布尔类型的

注意2：使用action = store_const 时修饰的指令的是不能赋值，其对应的指令的值常量类型

注意3：使用action = append 是可以复制，其对应的值会将该指令对应的值放在一个数组中

1.1 先来演示

使用action，保存布尔类型的指令的值：store_true/store_false:保存布尔类型的变量：

# 导入 argparse 模块
import argparse

# 创建一个argparse 的对象
parser = argparse.ArgumentParser(prog="testname", usage="这是功能描述模块", description="这是程序的介绍，和usage的功能模块描述类似", epilog="额外说明")

# 添加一个接收参数 -n，action = store_true 表示，<strong>在使用的时候，如果填写了该指令，那么该指令对应的值就是true，如果没有使用该指令，则为空</strong>
parser.add_argument("-n", "--name", action='store_true')

# 获取对应的参数的值
args = parser.parse_args()

# 控制台打印出来接收到的值
print("args = {}".format(args))

执行结果：

1.2、使用action设置常量指令

修改代码演示：

# 添加一个接收参数 -n，action = store_const 表示，在使用的时候，如果填写了该指令，那么该指就等于10 如果没写就等于空
parser.add_argument("-n", "--name", action='store_const',const=10)

执行结果：

1.3、使用action设置数组类型的指令

修改代码演示：

# 添加一个接收参数 -n，action = append， 会把该指令的值放在数组中
parser.add_argument("-n", "--name", action='append')

执行结果

2、nargs参数：表示该指令接收值的个数

示例1:

import argparse

# 创建一个argparse 的对象
parser = argparse.ArgumentParser(prog="testname", usage="这是功能描述模块", description="这是程序的介绍，和usage的功能模块描述类似", epilog="额外说明")

# 添加一个接收参数 -n, 并且该参数还必须要接收2个值，会放在一个数组中，其中nargs等于几，就表示要接受几个参数
parser.add_argument("-n", "--name", nargs=2)

# 获取对应的参数的值
args = parser.parse_args()

# 控制台打印出来接收到的值
print("args = {}".format(args))

运行结果：

示例2：

# 导入 argparse 模块
import argparse

# 创建一个argparse 的对象
parser = argparse.ArgumentParser(prog="testname", usage="这是功能描述模块", description="这是程序的介绍，和usage的功能模块描述类似", epilog="额外说明")

# 添加一个接收参数 -n,
# nargs = "?" :  表示只接收1个参数，正则表达式的写法
# nargs = "+" :  表示接收1一个或多个参数
# nargs = "*" :  表示接收0个或者多个参数
parser.add_argument("-n", "--name", nargs="*")

# 获取对应的参数的值
args = parser.parse_args()

# 控制台打印出来接收到的值
print("args = {}".format(args))

打印结果就不演示了

3、required 该指令是必须的

# 导入 argparse 模块
import argparse

# 创建一个argparse 的对象
parser = argparse.ArgumentParser(prog="testname", usage="这是功能描述模块", description="这是程序的介绍，和usage的功能模块描述类似", epilog="额外说明")

# 添加一个接收参数 -n,
# nargs = "?" :  表示只接收1个参数，正则表达式的写法
# nargs = "+" :  表示接收1一个或多个参数
# nargs = "*" :  表示接收0个或者多个参数
parser.add_argument("-n", "--name", nargs="*", required=True)

# 获取对应的参数的值
args = parser.parse_args()

# 控制台打印出来接收到的值
print("args = {}".format(args))

演示结果：

4、choices:如有choices选项，则该指令的值只能是choices中的一个

代码演示

# 导入 argparse 模块
import argparse

# 创建一个argparse 的对象
parser = argparse.ArgumentParser(prog="testname", usage="这是功能描述模块", description="这是程序的介绍，和usage的功能模块描述类似", epilog="额外说明")

# 添加一个接收参数 -n,
# nargs = "?" :  表示只接收1个参数，正则表达式的写法
# nargs = "+" :  表示接收1一个或多个参数
# nargs = "*" :  表示接收0个或者多个参数
parser.add_argument("-n", "--name", nargs="*", required=True, choices=['zhangsan', 'lisi'])

# 获取对应的参数的值
args = parser.parse_args()

# 控制台打印出来接收到的值
print("args = {}".format(args))

运行结果：

五、vars()

vars()方法可以将parser对象变成字典，方便使用