import argparse模塊

argparse 是 Python 內(nèi)置的一個(gè)用于命令項(xiàng)選項(xiàng)與參數(shù)解析的模塊结蟋,通過在程序中定義好我們需要的參數(shù)藐俺,argparse 將會從 sys.argv 中解析出這些參數(shù)硫戈,并自動生成幫助和使用信息倍谜。當(dāng)然瓢省,Python 也有第三方的庫可用于命令行解析窟坐,而且功能也更加強(qiáng)大海渊,比如 docoptClick哲鸳。

argparse 使用

簡單示例

我們先來看一個(gè)簡單示例臣疑。主要有三個(gè)步驟:

  • 創(chuàng)建 ArgumentParser() 對象
  • 調(diào)用 add_argument() 方法添加參數(shù)
  • 使用 parse_args() 解析添加的參數(shù)
import argparse

parser = argparse.ArgumentParser()
parser.add_argument('integer', type=int, help='display an integer')
args = parser.parse_args()

print args.integer

將上面的代碼保存為文件 argparse_usage.py,在終端運(yùn)行徙菠,結(jié)果如下:

$ python argparse_usage.py
usage: argparse_usage.py [-h] integer
argparse_usage.py: error: too few arguments

$ python argparse_usage.py abcd
usage: argparse_usage.py [-h] integer
argparse_usage.py: error: argument integer: invalid int value: 'abcd'

$ python argparse_usage.py -h
usage: argparse_usage.py [-h] integer

positional arguments:
  integer     display an integer

optional arguments:
  -h, --help  show this help message and exit

$ python argparse_usage.py 10
10

定位參數(shù)
上面的示例讯沈,其實(shí)就展示了定位參數(shù)的使用,我們再來看一個(gè)例子 - 計(jì)算一個(gè)數(shù)的平方:

import argparse

parser = argparse.ArgumentParser()
parser.add_argument("square", help="display a square of a given number", type=int)
args = parser.parse_args()
print args.square**2

將上面的代碼保存為文件 argparse_usage.py婿奔,在終端運(yùn)行缺狠,結(jié)果如下:

$ python argparse_usage.py 9
81

可選參數(shù)
現(xiàn)在看下可選參數(shù)的用法问慎,所謂可選參數(shù),也就是命令行參數(shù)是可選的挤茄,廢話少說如叼,看下面例子:

import argparse

parser = argparse.ArgumentParser()

parser.add_argument("--square", help="display a square of a given number", type=int)
parser.add_argument("--cubic", help="display a cubic of a given number", type=int)

args = parser.parse_args()

if args.square:
    print args.square**2

if args.cubic:
    print args.cubic**3

將上面的代碼保存為文件 argparse_usage.py,在終端運(yùn)行穷劈,結(jié)果如下:

$ python argparse_usage.py --h
usage: argparse_usage.py [-h] [--square SQUARE] [--cubic CUBIC]

optional arguments:
  -h, --help       show this help message and exit
  --square SQUARE  display a square of a given number
  --cubic CUBIC    display a cubic of a given number

$ python argparse_usage.py --square 8
64

$ python argparse_usage.py --cubic 8
512

$ python argparse_usage.py 8
usage: argparse_usage.py [-h] [--square SQUARE] [--cubic CUBIC]
argparse_usage.py: error: unrecognized arguments: 8

$ python argparse_usage.py  # 沒有輸出

混合使用
定位參數(shù)和選項(xiàng)參數(shù)可以混合使用笼恰,看下面一個(gè)例子,給一個(gè)整數(shù)序列歇终,輸出它們的和或最大值(默認(rèn)):

import argparse

parser = argparse.ArgumentParser(description='Process some integers.')
parser.add_argument('integers', metavar='N', type=int, nargs='+',
                   help='an integer for the accumulator')
parser.add_argument('--sum', dest='accumulate', action='store_const',
                   const=sum, default=max,
                   help='sum the integers (default: find the max)')

args = parser.parse_args()
print args.accumulate(args.integers)

結(jié)果:

$ python argparse_usage.py
usage: argparse_usage.py [-h] [--sum] N [N ...]
argparse_usage.py: error: too few arguments
$ python argparse_usage.py 1 2 3 4
4
$ python argparse_usage.py 1 2 3 4 --sum
10

add_argument() 方法
add_argument() 方法定義如何解析命令行參數(shù):

ArgumentParser.add_argument(name or flags...[, action][, nargs][, const][, default][, type][, choices][, required][, help][, metavar][, dest])

每個(gè)參數(shù)解釋如下:

name or flags - 選項(xiàng)字符串的名字或者列表社证,例如 foo 或者 -f, --foo。
action - 命令行遇到參數(shù)時(shí)的動作练湿,默認(rèn)值是 store猴仑。
store_const,表示賦值為const肥哎;
append辽俗,將遇到的值存儲成列表,也就是如果參數(shù)重復(fù)則會保存多個(gè)值;
append_const篡诽,將參數(shù)規(guī)范中定義的一個(gè)值保存到一個(gè)列表崖飘;
count,存儲遇到的次數(shù)杈女;此外朱浴,也可以繼承 argparse.Action 自定義參數(shù)解析;
nargs - 應(yīng)該讀取的命令行參數(shù)個(gè)數(shù)达椰,可以是具體的數(shù)字翰蠢,或者是?號,當(dāng)不指定值時(shí)對于 Positional argument 使用 default啰劲,對于 Optional argument 使用 const梁沧;或者是 * 號,表示 0 或多個(gè)參數(shù)蝇裤;或者是 + 號表示 1 或多個(gè)參數(shù)廷支。
const - action 和 nargs 所需要的常量值。
default - 不指定參數(shù)時(shí)的默認(rèn)值栓辜。
type - 命令行參數(shù)應(yīng)該被轉(zhuǎn)換成的類型恋拍。
choices - 參數(shù)可允許的值的一個(gè)容器。
required - 可選參數(shù)是否可以省略 (僅針對可選參數(shù))藕甩。
help - 參數(shù)的幫助信息施敢,當(dāng)指定為 argparse.SUPPRESS 時(shí)表示不顯示該參數(shù)的幫助信息.
metavar - 在 usage 說明中的參數(shù)名稱,對于必選參數(shù)默認(rèn)就是參數(shù)名稱,對于可選參數(shù)默認(rèn)是全大寫的參數(shù)名稱.
dest - 解析后的參數(shù)名稱僵娃,默認(rèn)情況下羡藐,對于可選參數(shù)選取最長的名稱,中劃線轉(zhuǎn)換為下劃線.
注意:dest參數(shù)悯许,例如:

parser.add_argument('-b', action="store", dest="b")

那么即可用 args.b 來訪問該值。

具體參見:http://blog.xiayf.cn/2013/03/30/argparse/

參考資料

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末辉阶,一起剝皮案震驚了整個(gè)濱河市先壕,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌谆甜,老刑警劉巖垃僚,帶你破解...
    沈念sama閱讀 222,183評論 6 516
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異规辱,居然都是意外死亡谆棺,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 94,850評論 3 399
  • 文/潘曉璐 我一進(jìn)店門罕袋,熙熙樓的掌柜王于貴愁眉苦臉地迎上來改淑,“玉大人,你說我怎么就攤上這事浴讯《湎模” “怎么了?”我有些...
    開封第一講書人閱讀 168,766評論 0 361
  • 文/不壞的土叔 我叫張陵榆纽,是天一觀的道長仰猖。 經(jīng)常有香客問我,道長奈籽,這世上最難降的妖魔是什么饥侵? 我笑而不...
    開封第一講書人閱讀 59,854評論 1 299
  • 正文 為了忘掉前任,我火速辦了婚禮衣屏,結(jié)果婚禮上躏升,老公的妹妹穿的比我還像新娘。我一直安慰自己勾拉,他們只是感情好煮甥,可當(dāng)我...
    茶點(diǎn)故事閱讀 68,871評論 6 398
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著藕赞,像睡著了一般成肘。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上斧蜕,一...
    開封第一講書人閱讀 52,457評論 1 311
  • 那天双霍,我揣著相機(jī)與錄音,去河邊找鬼。 笑死洒闸,一個(gè)胖子當(dāng)著我的面吹牛染坯,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播丘逸,決...
    沈念sama閱讀 40,999評論 3 422
  • 文/蒼蘭香墨 我猛地睜開眼单鹿,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了深纲?” 一聲冷哼從身側(cè)響起仲锄,我...
    開封第一講書人閱讀 39,914評論 0 277
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎湃鹊,沒想到半個(gè)月后儒喊,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 46,465評論 1 319
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡币呵,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 38,543評論 3 342
  • 正文 我和宋清朗相戀三年怀愧,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片余赢。...
    茶點(diǎn)故事閱讀 40,675評論 1 353
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡芯义,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出妻柒,到底是詐尸還是另有隱情毕贼,我是刑警寧澤,帶...
    沈念sama閱讀 36,354評論 5 351
  • 正文 年R本政府宣布蛤奢,位于F島的核電站鬼癣,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏啤贩。R本人自食惡果不足惜待秃,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 42,029評論 3 335
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望痹屹。 院中可真熱鬧章郁,春花似錦、人聲如沸志衍。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,514評論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽楼肪。三九已至培廓,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間春叫,已是汗流浹背肩钠。 一陣腳步聲響...
    開封第一講書人閱讀 33,616評論 1 274
  • 我被黑心中介騙來泰國打工泣港, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人价匠。 一個(gè)月前我還...
    沈念sama閱讀 49,091評論 3 378
  • 正文 我出身青樓当纱,卻偏偏與公主長得像,于是被迫代替她去往敵國和親踩窖。 傳聞我的和親對象是個(gè)殘疾皇子坡氯,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,685評論 2 360