http://www.linuxidc.com/Linux/2012-02/53764.htm(轉(zhuǎn))
本篇將介紹python中sys, getopt模塊處理命令行參數(shù)
如果想對python腳本傳參數(shù),python中對應(yīng)的argc, argv(c語言的命令行參數(shù))是什么呢博助?
需要模塊:sys
參數(shù)個(gè)數(shù):len(sys.argv)
腳本名: sys.argv[0]
參 數(shù)1: sys.argv[1]
參 數(shù)2: sys.argv[2]
test.py
1 import sys
2 print "腳本名:", sys.argv[0]
3 for i in range(1, len(sys.argv)):
4 print "參數(shù)", i, sys.argv[i]
python test.py hello world
腳本名:test.py
參數(shù) 1 hello
參數(shù) 2 world
python中使用命令行選項(xiàng):
例如我們需要一個(gè)convert.py腳本囱修。它的作用是處理一個(gè)文件,并將處理后的結(jié)果輸出到另一個(gè)文件中。
要求該腳本滿足以下條件:
1.通過-i -o選項(xiàng)來區(qū)別參數(shù)是輸入文件還是輸出文件.
python convert.py -i inputfile -o outputfile
2.當(dāng)不知道convert.py需要哪些參數(shù)時(shí)嫡锌,用-h打印出幫助信息
python convert.py -h
getopt函數(shù)原形:
getopt.getopt(args, options[, long_options])
convert.py
01 import sys, getopt
02
03 opts, args = getopt.getopt(sys.argv[1:], "hi:o:")
04 input_file=""
05 output_file=""
06
07 for op, value in opts:
08 if op == "-i":
09 input_file = value
10 elif op == "-o":
11 output_file = value
12 elif op == "-h":
13 usage()
14 sys.exit()
代碼解釋:
a) sys.argv[1:]為要處理的參數(shù)列表,sys.argv[0]為腳本名,所以用sys.argv[1:]過濾掉腳本名鹏浅。
b) "hi:o:": 當(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)吏颖。所以"hi:o:"就表示"h"是一個(gè)開關(guān)選項(xiàng);"i:"和"o:"則表示后面應(yīng)該帶一個(gè)參數(shù)恨樟。
c) 調(diào)用getopt函數(shù)半醉。函數(shù)返回兩個(gè)列表:opts和args。opts為分析出的格式信息劝术。args為不屬于格式信息的剩余的命令行參數(shù)缩多。opts是一個(gè)兩元組的列表呆奕。每個(gè)元素為:(選項(xiàng)串,附加參數(shù))。如果沒有附加參數(shù)則為空串''衬吆。
getopt函數(shù)的第三個(gè)參數(shù)[, long_options]為可選的長選項(xiàng)參數(shù)梁钾,上面例子中的都為短選項(xiàng)(如-i -o)
長選項(xiàng)格式舉例:
--version
--file=error.txt
讓一個(gè)腳本同時(shí)支持短選項(xiàng)和長選項(xiàng)
getopt.getopt(sys.argv[1:], "hi:o:", ["version", "file="])