使用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')
posted @ 2024-11-08 20:08  RolandHe  阅读(12)  评论(0编辑  收藏  举报