💡 如果想阅读最新的文章,或者有技术问题需要交流和沟通,可搜关注微信公众号号号“希望睿智”。
概述在上一节,我们介绍了Python的datetime模块,包括:datetime模块中一些常用的属性和函数。在这一节,我们将介绍Python的argparse模块。argparse模块是Python的一个标准库,用于编写命令行界面。它可以处理命令行参数和选项,并生成帮助和使用信息。
该模块主要包含的功能如下:
帮助和使用信息生成:自动生成帮助和使用信息,包括参数和选项的说明、用法示例等。
命令行参数解析:解析命令行参数和选项,包括位置参数、可选参数和标志。
参数类型检查:支持对参数类型进行检查,包括整数、浮点数、字符串等。
子命令支持:支持定义子命令,每个子命令可以有自己的参数和选项。
参数组:支持将参数分组,以方便管理和使用。
下面,我们将逐一介绍argparse模块中一些常用的类和函数。
argparse.ArgumentParser类argparse.ArgumentParser是argparse模块中的一个类,用于创建解析器对象,帮助处理命令行参数和选项,并生成帮助和使用信息。使用argparse.ArgumentParser类可以简化命令行参数和选项的解析过程,以下是具体的使用步骤:
1、导入argparse模块:import argparse。
2、实例化一个对象:parser = argparse.ArgumentParser()。
3、加入参数或选项:parser.add_argument(),每一个add_argument方法对应一个要关注的参数或选项。
4、获取参数:args = parser.parse_args(),解析成功之后即可使用。
在创建argparse.ArgumentParser对象时,可以指定一些可选参数,比如:description,用于描述程序的用途和参数的说明。
add_argument基本选项通过调用add_argument函数,可以指定命令行参数的名称、类型、默认值等属性,为参数提供帮助文本和元数据。
# test.pyimport argparse# 创建解析器对象parser = argparse.ArgumentParser(description = '这是一个示例程序')# 添加参数parser.add_argument('param_name', type = int, help = '这是一个整数参数')# 解析参数args = parser.parse_args()# 使用参数print(args.param_name)
在上面的示例代码中,我们使用add_argument函数添加了一个名为param_name的参数,指定了它的类型为整数,并提供了帮助文本。然后,通过调用parse_args函数解析命令行参数,并将解析后的参数值存储在args对象中。最后,我们可以使用args.param_name访问该参数的值。如果我们在命令行输入python test.py 66命令,则会输出:66。
add_argument更多选项add_argument函数还有许多其他选项,如下。
metavar:指定参数的元变量名称,用于帮助文本中。
nargs:指定参数接收的值的数量,可以是固定的数量,也可以是可变数量。其中,'+'表示至少需要一个参数,'*'表示可以接受任意数量的参数,数字表示只接受固定数量的参数。
default:指定参数的默认值。
choices:限制参数的取值范围。
required:指定参数是否必需。
help:提供参数的帮助文本。
下面是一个稍微复杂一点的示例代码,用到了更多的选项。
# test.pyimport argparse# 创建解析器对象parser = argparse.ArgumentParser(description = '这是一个示例程序')# 添加参数parser.add_argument('integers', metavar='N', type=int, nargs='+', help = '用于运算的整数的个数')parser.add_argument('--sum', dest='compute_sum', action='store_true', help = '计算输入的整数的和')parser.add_argument('--max', dest='find_max', action='store_true', help = '找到输入的最大整数')parser.add_argument('--average', dest='compute_average', action='store_true', help = '计算输入的平均值')# 解析参数args = parser.parse_args()# 使用参数if args.compute_sum: print(sum(args.integers))elif args.find_max: print(max(args.integers))elif args.compute_average: print(sum(args.integers) / len(args.integers))else: print('未选择任何操作')
在上面的示例代码中,我们添加了如下几个参数和选项。
integers:一个位置参数,接受一个或多个整数作为输入。
--sum:一个可选的选项,用于计算输入整数的总和。
--max:一个可选的选项,用于找出输入整数的最大值。
--average:一个可选的选项,用于计算输入整数的平均值。
通过解析命令行参数,我们可以根据用户选择的选项执行相应的操作,并打印输出相应的结果。上述代码的执行结果如下。
python test.py -husage: test.py [-h] [--sum] [--max] [--average] N [N ...]这是一个示例程序positional arguments: N 用于运算的整数的个数options: -h, --help show this help message and exit --sum 计算输入的整数的和 --max 找到输入的最大整数 --average 计算输入的平均值 python test.py --sum 5 6 718python test.py --max 5 6 77python test.py --average 5 6 76.0python test.py 5 6 7未选择任何操作
add_argument的action选项action选项用来定义当命令行选项被触发时的行为。有多种可用的行为,如下。
'store':默认值。当命令行选项被触发时,将选项后面的参数值存储下来。
'store_const':当命令行选项被触发时,将一个常量值存储下来。比如:add_argument('--foo', action='store_const', const=42)会将42存储为args.foo。
'store_true' 和 'store_false':当命令行选项被触发时,将一个布尔值存储下来。比如:add_argument('--foo', action='store_true')会在args.foo为真(即选项被触发)时存储一个True值。
'append':当命令行选项被触发时,将一个值添加到列表中。比如:add_argument('--foo', action='append', default=[])会将任何遇到的--foo选项的值添加到args.foo列表中。
'append_const':当命令行选项被触发时,将一个常量值添加到列表中。比如:add_argument('--foo', action='append_const', const=42)会将42添加到args.foo列表中。
# test.pyimport argparseparser = argparse.ArgumentParser()parser.add_argument('--param1', action = 'store_true')parser.add_argument('--param2', action = 'append', default = [])parser.add_argument('numbers', nargs = '+')args = parser.parse_args()print(args.param1)print(args.param2)print(args.numbers)
在上面的示例代码中,--param1选项被触发时,将存储一个True值。--param2选项可以接受任意数量的参数,并将它们添加到一个列表中。numbers选项接受一个或多个参数,并将它们添加到一个列表中。上述代码的执行结果如下。
python test.py 99 100 --param1 --param2 6 --param2 9True['6', '9']['99', '100']