$命令行參數解析模塊argparse的用法

argparse是python內置的命令行參數解析模塊，可以用來為程序配置功能豐富的命令行參數，方便使用，本文總結一下其基本用法。

測試腳本

把以下腳本存在argtest.py文件中：

# coding:utf-8
# 測試argparse模塊的基本用法
import argparse

# 創建參數解析對象，並添加腳本用法幫助
parser = argparse.ArgumentParser(description = 'test the base usage of argparse.')

# 添加位置參數
# 所謂位置參數，就是指直接添加的參數而不用使用'-'、'--'等符號
# 添加了位置參數，它就是必選參數
parser.add_argument('arg0')

# 添加可選參數，但如果執行命令時帶有該參數，后面必須跟參數值
# '-'后面跟短參數，'--'后面跟長參數
# help參數為該參數的幫助信息
parser.add_argument('-a1','--arg1',help = 'this is arg1')

# 添加可選參數，但后面不能跟參數值
parser.add_argument('-a2','--arg2',help = 'this is arg2',action = 'store_true')

# 添加可選參數並指定參數值數據類型為整型，且數據范圍為[0,1,2]，且指定默認值為0,如果輸入的值不是整型或值不在要求的范圍內，則會報錯
parser.add_argument('-a3','--arg3',type = int,choices = [0,1,2],default = 0,help = 'this is arg3')

# 添加一組可選的互斥參數
# a4和a5參數不能同時出現，否則會報錯
group = parser.add_mutually_exclusive_group()
group.add_argument('-a4','--arg4',action = 'store_true')
group.add_argument('-a5','--arg5',action = 'store_true')

#####################################

# 執行解析參數
args = parser.parse_args()

# 打印出位置參數'arg0'
print 'arg0 is: ',args.arg0

# 打印出其他位置參數，注意這里要用參數的'--'名稱（長參數）
if args.arg1:
    print 'arg1 is: ',args.arg1

# 因為arg2后面沒有跟參數值，所以打印出來是True
if args.arg2:
    print 'arg2 is: ',args.arg2
    
if args.arg3:
    print 'arg3 is: ',args.arg3
    
if args.arg4:
    print 'arg4 is: ',args.arg4
    
if args.arg5:
    print 'arg5 is: ',args.arg5
    

執行腳本進行測試

(.env) e:\code\python-basic\libs\argparse>argtest.py -h
usage: argtest.py [-h] [-a1 ARG1] [-a2] [-a3 {0,1,2}] [-a4 | -a5] arg0

test the base usage of argparse.

positional arguments:
  arg0

optional arguments:
  -h, --help            show this help message and exit
  -a1 ARG1, --arg1 ARG1
                        this is arg1
  -a2, --arg2           this is arg2
  -a3 {0,1,2}, --arg3 {0,1,2}
                        this is arg3
  -a4, --arg4
  -a5, --arg5

(.env) e:\code\python-basic\libs\argparse>python argtest.py 000 -a1 111 -a2 --ar
g3 2 -a4
arg0 is:  000
arg1 is:  111
arg2 is:  True
arg3 is:  2
arg4 is:  True

(.env) e:\code\python-basic\libs\argparse>python argtest.py
usage: argtest.py [-h] [-a1 ARG1] [-a2] [-a3 {0,1,2}] [-a4 | -a5] arg0
argtest.py: error: too few arguments

(.env) e:\code\python-basic\libs\argparse>python argtest.py 000
arg0 is:  000

(.env) e:\code\python-basic\libs\argparse>python argtest.py 000 -a1
usage: argtest.py [-h] [-a1 ARG1] [-a2] [-a3 {0,1,2}] [-a4 | -a5] arg0
argtest.py: error: argument -a1/--arg1: expected one argument

(.env) e:\code\python-basic\libs\argparse>python argtest.py 000 -a1 111
arg0 is:  000
arg1 is:  111

(.env) e:\code\python-basic\libs\argparse>python argtest.py 000 -a1 111 -a2
arg0 is:  000
arg1 is:  111
arg2 is:  True

(.env) e:\code\python-basic\libs\argparse>python argtest.py 000 -a1 111 -a2 222
usage: argtest.py [-h] [-a1 ARG1] [-a2] [-a3 {0,1,2}] [-a4 | -a5] arg0
argtest.py: error: unrecognized arguments: 222

(.env) e:\code\python-basic\libs\argparse>python argtest.py 000 -a3 333         
usage: argtest.py [-h] [-a1 ARG1] [-a2] [-a3 {0,1,2}] [-a4 | -a5] arg0          
argtest.py: error: argument -a3/--arg3: invalid choice: 333 (choose from 0, 1, 2)                                                                             

(.env) e:\code\python-basic\libs\argparse>python argtest.py 000 -a3 2
arg0 is:  000
arg3 is:  2

(.env) e:\code\python-basic\libs\argparse>python argtest.py 000 -a4 -a5
usage: argtest.py [-h] [-a1 ARG1] [-a2] [-a3 {0,1,2}] [-a4 | -a5] arg0
argtest.py: error: argument -a5/--arg5: not allowed with argument -a4/--arg4

(.env) e:\code\python-basic\libs\argparse>python argtest.py 000 -a4
arg0 is:  000
arg4 is:  True