使用argparse模組新增命令列引數

RolandHe發表於2024-11-08
命令列

有時，我們希望我們的程式執行時能夠透過引數來控制其行為，一種就是透過一個前端互動的介面讓使用者來配置引數，另一種就是直接透過命令列的方式啟動程式的同時設定引數。

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')

相關文章