使用argparse模块添加命令行参数
有时,我们希望我们的程序运行时能够通过参数来控制其行为,一种就是通过一个前端交互的界面让用户来配置参数,另一种就是直接通过命令行的方式启动程序的同时设置参数。
argparse模块就实现了命令行参数的功能,参数名如果是 --parm 或 -p 这种加了 - 或 -- 前缀的,默认为可选参数,但也可以指定其为必填参数。 不带前缀的,就是位置参数,也是必填参数。使用了这些参数后,无论是在开发环境运行,即python main.py --parm
, 或是将程序打包后再运行,即 main.exe --parm
都是生效的。
简单用法如下:
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("step", help="1: 执行金额及客户匹配, 2: 执行与发票的一对多金额匹配", choices=['1','2'])
args = parser.parse_args()
if args.step == '1':
print('开始执行step 1')
else:
print('开始执行step 2')