在運(yùn)行程序時(shí)煎楣,可能需要根據(jù)不同的條件既峡,輸入不同的命令行選項(xiàng)來實(shí)現(xiàn)不同的功能守问。目前有短選項(xiàng)和長(zhǎng)選項(xiàng)兩種格式匀归。短選項(xiàng)格式為"-"加上單個(gè)字母選項(xiàng);長(zhǎng)選項(xiàng)為"--"加上一個(gè)單詞耗帕。長(zhǎng)格式是在Linux下引入的穆端。許多Linux程序都支持這兩種格式。在Python中提供了getopt模塊很好的實(shí)現(xiàn)了對(duì)這兩種用法的支持仿便,而且使用簡(jiǎn)單体啰。
取得命令行參數(shù)
在使用之前,首先要取得命令行參數(shù)嗽仪。使用sys模塊可以得到命令行參數(shù)荒勇。
import sys
print sys.argv
然后在命令行下敲入任意的參數(shù),如:
python get.py -o t --help cmd file1 file2
結(jié)果為:
['get.py', '-o', 't', '--help', 'cmd', 'file1', 'file2']
可見闻坚,所有命令行參數(shù)以空格為分隔符沽翔,都保存在了sys.argv列表中。其中第1個(gè)為腳本的文件名窿凤。
選項(xiàng)的寫法要求
對(duì)于短格式仅偎,"-"號(hào)后面要緊跟一個(gè)選項(xiàng)字母。如果還有此選項(xiàng)的附加參數(shù)卷玉,可以用空格分開哨颂,也可以不分開。長(zhǎng)度任意相种,可以用引號(hào)威恼。如以下是正確的:
-o
-oa
-obbbb
-o bbbb
-o "a b"
對(duì)于長(zhǎng)格式品姓,"--"號(hào)后面要跟一個(gè)單詞。如果還有些選項(xiàng)的附加參數(shù)箫措,后面要緊跟"="腹备,再加上參數(shù)。"="號(hào)前后不能有空格斤蔓。如以下是正確的:
--help=file1
而這些是不正確的:
--help =file1
--help = file1
--help= file1
如何用getopt進(jìn)行分析
使用getopt模塊分析命令行參數(shù)大體上分為三個(gè)步驟:
1.導(dǎo)入getopt, sys模塊
2.分析命令行參數(shù)
3.處理結(jié)果
例如模塊test1.py:
import getopt, sys
opts, args = getopt.getopt(sys.argv[1:], "ho:", ["help", "output="])#"ho:"也可以寫成'-h-o:'
print(opts)
print(args)
- 處理所使用的函數(shù)叫g(shù)etopt()植酥,因?yàn)槭侵苯邮褂胕mport導(dǎo)入的getopt模塊,所以要加上限定getopt才可以弦牡。
- 使用sys.argv[1:]過濾掉第一個(gè)參數(shù)(它是執(zhí)行腳本的名字友驮,不應(yīng)算作參數(shù)的一部分)。
- 使用短格式分析串"ho:"驾锰。當(dāng)一個(gè)選項(xiàng)只是表示開關(guān)狀態(tài)時(shí)卸留,即后面不帶附加參數(shù)時(shí),在分析串中寫入選項(xiàng)字符椭豫。當(dāng)選項(xiàng)后面是帶一個(gè)附加參數(shù)時(shí)耻瑟,在分析串中寫入選項(xiàng)字符同時(shí)后面加一個(gè)":"號(hào)。所以"ho:"就表示"h"是一個(gè)開關(guān)選項(xiàng)赏酥;"o:"則表示后面應(yīng)該帶一個(gè)參數(shù)喳整。
- 使用長(zhǎng)格式分析串列表:["help", "output="]。長(zhǎng)格式串也可以有開關(guān)狀態(tài)裸扶,即后面不跟"="號(hào)框都。如果跟一個(gè)等號(hào)則表示后面還應(yīng)有一個(gè)參數(shù)。這個(gè)長(zhǎng)格式表示"help"是一個(gè)開關(guān)選項(xiàng)姓言;"output="則表示后面應(yīng)該帶一個(gè)參數(shù)瞬项。
- 調(diào)用getopt函數(shù)。函數(shù)返回兩個(gè)列表:opts和args何荚。opts為分析出的格式信息囱淋。args為不屬于格式信息的剩余的命令行參數(shù),即不是按照getopt()里面定義的長(zhǎng)或短選項(xiàng)字符和附加參數(shù)以外的信息。opts是一個(gè)兩元組的列表餐塘。每個(gè)元素為:(選項(xiàng)串,附加參數(shù))妥衣。如果沒有附加參數(shù)則為空串''。
- 整個(gè)過程使用異常來包含戒傻,這樣當(dāng)分析出錯(cuò)時(shí)税手,就可以打印出使用信息來通知用戶如何使用這個(gè)程序。
用上面解釋的一個(gè)命令行例子運(yùn)行:
python test1.py '-h -o file --help --output=out file1 file2'
opts的輸出結(jié)果為:
[('-h', ''), ('-o', 'file'), ('--help', ''), ('--output', 'out')]
而args則為:['file1', 'file2']需纳,這就是上面不屬于格式信息的剩余的命令行參數(shù)芦倒。
對(duì)于上面例子再增加如下代碼,主要是對(duì)分析出的參數(shù)進(jìn)行判斷是否存在不翩,然后再進(jìn)一步處理兵扬。主要的處理模式為:
for o, a in opts:
if o in ("-h", "--help"):
usage()
sys.exit()
if o in ("-o", "--output"):
output = a
使用一個(gè)循環(huán)麻裳,每次從opts中取出一個(gè)兩元組,賦給兩個(gè)變量器钟。o保存選項(xiàng)參數(shù)津坑,a為附加參數(shù)。接著對(duì)取出的選項(xiàng)參數(shù)進(jìn)行處理傲霸。