python3 argparse模組
argparse模組為python3中處理命令列引數的庫 命令列引數分為位置引數和可選引數, 例: ls home #位置引數 ls -l #可選引數
簡單示例, 主要有三個步驟
- 建立 ArgumentParser() 物件
- 呼叫 add_argument() 方法新增引數
- 使用 parse_args() 解析新增的引數
位置引數
test.py
檔案
import argparse
parse = argparse.ArgumentParser()
parse.add_argument('test') #新增引數 test
args = parse.parse_args()
print (args)
執行命令 python test.py -h
usage: test.py [-h] test
positional arguments:
test
optional arguments:
-h, --help show this help message and exit
# 可以看到多了一個位置引數test
執行命令 python test.py 123
Namespace(test='123')
# 已經成功賦值, 預設為字串
可選引數
test.py
檔案
import argparse
parse = argparse.ArgumentParser( )
parse.add_argument('-l') #新增可選引數 -l
args = parse.parse_args()
print(args)
執行命令 python test.py -h
usage: test.py [-h] [-l L]
optional arguments:
-h, --help show this help message and exit
-l L
# 可以看到多了一個位置引數 -l
執行命令 python test.py -l 123
Namespace(l='123')
# 已經成功賦值, 預設為字串
若 parse.add_argument('-l', '--list')
python test.py -l 123
Namespace(list='123')
# 可選引數兩種呼叫方法, l和list都能用, 最後引數為list
add_argument 方法常用引數
ArgumentParser.add_argument(name or flags…[, action][, nargs][, const][, default][, type][, choices][, required][, help][, metavar][, dest])
help – 引數的幫助資訊
parse.add_argument('-l', '--list', help='help~help~')
# python test.py -h
usage: test.py [-h] [-l LIST]
optional arguments:
-h, --help show this help message and exit
-l LIST, --list LIST help~help~
type – 命令列引數應該被轉換成的型別
parse.add_argument('-l', '--list', type=int)
# python test.py -l 123
Namespace(list=123)
# python test.py -l test
usage: test.py [-h] [-l LIST]
test.py: error: argument -l/--list: invalid int value: 'test'
(錯誤, 提示必須輸入整數)
choices – 取值範圍
parse.add_argument('-l', '--list', type=int, choices=[0,1,2])
# python test.py -l 1
Namespace(list=1)
# python test.py -l 5
usage: test.py [-h] [-l {0,1,2}]
test.py: error: argument -l/--list: invalid choice: 5 (choose from 0, 1, 2)
dest – 解析後的引數名稱
parse.add_argument('-l', '--list', dest='test')
# python test.py -l 123
Namespace(test='123')
default – 預設值
parse.add_argument('-l', '--list', type=int, default=0)
# python test.py -l 123
Namespace(list=123)
# python test.py
Namespace(list=0)
required – 可選引數是否可以省略, 僅針對可選引數
parse.add_argument('-l', '--list', type=int, required=True)
# python test.py -l 123
Namespace(list=123)
# python test.py
usage: test.py [-h] -l LIST
test.py: error: the following arguments are required: -l/--list
nargs – 應該讀取的命令列引數個數, 可以是具體的數字, 或者是 ? 號(0或者1個引數
),或者是 * 號(0或多個引數
), 或者是 + 號(1或多個引數
)
parse.add_argument('-l', '--list', type=int, nargs='?')
# python test.py -l 1
Namespace(list=1)
# python test.py -l 1 2
usage: test.py [-h] [-l [LIST]]
test.py: error: unrecognized arguments: 2
action=‘store_true’ / action=‘store_false’ – 儲存相應的布林值, 這兩個動作被用於實現布林開關
parse.add_argument('-l', '--list', action='store_true')
# python test.py
Namespace(list=False)
# python test.py -l
Namespace(list=True)
互斥引數
import argparse
parse = argparse.ArgumentParser()
group = parse.add_mutually_exclusive_group()
group.add_argument('-x')
group.add_argument('-y')
args = parse.parse_args()
print(args)
# python test.py -x 2
Namespace(x='2', y=None)
# python test.py -y 3
Namespace(x='None', y='3')
#python test.py -x 2 -y 3
usage: test.py [-h] [-x X | -y Y]
test.py: error: argument -y: not allowed with argument -x