有時候我們需要寫一些腳本處理一些任務,這時候往往需要提供一些命令行參數,根據不同參數進行不同的處理,在Python里,命令行的參數和C語言很類似(因為標准Python是用C語言實現的)。在C語言里,main函數的原型為int main(int argc, char **argv),這里主要指linux平台, argc指的是命令行傳入的參數個數(程序的name為第一個參數),而argv則是一個指針數組,每一個元素為指向一個命令行參數的指針。在Python里的命令行參數是存儲在sys.argv里,argv是一個列表,第一個元素也為程序名稱。
看下面的例子:
test.py
# -*- coding:utf-8 -*-
import sys if __name__ == '__main__': for item in sys.argv: print i
在命令行運行./test.py 1 2 3,結果如下:
test.py
1
2
3
用這種方式可以進行命令行參數的判斷從而進行不同的動作。
而Python的標准庫里實際上有專門處理命令行參數的getopt模塊,里面的提供了2個函數和一個類,我們主要使用getopt函數,先看下函數原型:
def getopt(args, shortopts, longopts = []):
先看一個例子,這樣會便於理解。
try:
opts, args = getopt.getopt(sys.argv[1:], "ho:", ["help", "output="]) except getopt.GetoptError: # print help information and exit: for name, value in opts: print name, value for item in args: print item
1. 處理所使用的函數叫getopt() ,因為是直接使用import 導入的getopt 模塊,所以要加上限定getopt 才可以。
2. 使用sys.argv[1:] 過濾掉第一個參數(它是執行腳本的名字,不應算作參數的一部分)。
3. 使用短格式分析串"ho:" 。當一個選項只是表示開關狀態時,即后面不帶附加參數時,在分析串中寫入選項字符。當選項后面是帶一個附加參數時,在分析串中寫入選項字符同時后面加一個":" 號 。所以"ho:" 就表示"h" 是一個開關選項;"o:" 則表示后面應該帶一個參數。
4. 使用長格式分析串列表:["help", "output="] 。長格式串也可以有開關狀態,即后面不跟"=" 號。如果跟一個等號則表示后面還應有一個參數 。這個長格式表示"help" 是一個開關選項;"output=" 則表示后面應該帶一個參數。
5. 調用getopt 函數。函數返回兩個列表:opts 和args 。opts 為分析出的格式信息。args 為不屬於格式信息的剩余的命令行參數。opts 是一個兩元組的列表。每個元素為:( 選項串, 附加參數) 。如果沒有附加參數則為空串'' 。
6. 整個過程使用異常來包含,這樣當分析出錯時,就可以打印出使用信息來通知用戶如何使用這個程序。
如上面解釋的一個命令行例子為:
'-h -o file --help --output=out file1 file2'
在分析完成后,opts 應該是:
[('-h', ''), ('-o', 'file'), ('--help', ''), ('--output', 'out')]
而args 則為:
['file1', 'file2']
下面就是根據不同參數處理:
for o, a in opts: if o in ("-h", "--help"): usage() sys.exit() if o in ("-o", "--output"): output = a