1. 程式人生 > >Python 獲得命令列引數的方法

Python 獲得命令列引數的方法

如果想對python指令碼傳引數,python中對應的argc, argv(c語言的命令列引數)是什麼呢?
需要模組:sys
引數個數:len(sys.argv)
指令碼名:    sys.argv[0]
引數1:     sys.argv[1]
引數2:     sys.argv[2]


test.py

import sys
print "指令碼名:", sys.argv[0]
for i in range(1, len(sys.argv)):
    print "引數", i, sys.argv[i]

>>>python test.py hello world

指令碼名:test.py
引數 1 hello
引數 2 world

python中使用命令列選項:
例如我們需要一個convert.py指令碼。它的作用是處理一個檔案,並將處理後的結果輸出到另一個檔案中。
要求該指令碼滿足以下條件:
1.通過-i -o選項來區別引數是輸入檔案還是輸出檔案.
>>> python convert.py -i inputfile -o outputfile
2.當不知道convert.py需要哪些引數時,用-h打印出幫助資訊
>>> python convert.py -h

getopt函式原形:
getopt.getopt(args, options[, long_options])

convert.py

複製程式碼
import sys, getopt

opts, args = getopt.getopt(sys.argv[1:], "hi:o:")
input_file=""
output_file=""
for op, value in opts:
    if op == "-i":
        input_file = value
    elif op == "-o":
        output_file = value
    elif op == "-h":
        usage()
        sys.exit()
複製程式碼

程式碼解釋:
a) sys.argv[1:]為要處理的引數列表,sys.argv[0]為指令碼名,所以用sys.argv[1:]過濾掉指令碼名。
b) "hi:o:": 當一個選項只是表示開關狀態時,即後面不帶附加引數時,在分析串中寫入選項字元。當選項後面是帶一個附加引數時,在分析串中寫入選項字元同時後面加一個":"號。所以"hi:o:"就表示"h"是一個開關選項;"i:"和"o:"則表示後面應該帶一個引數。
c) 呼叫getopt函式。函式返回兩個列表:opts和args。opts為分析出的格式資訊。args為不屬於格式資訊的剩餘的命令列引數。opts是一個兩元組的列表。每個元素為:(選項串,附加引數)。如果沒有附加引數則為空串''。

getopt函式的第三個引數[, long_options]為可選的長選項引數,上面例子中的都為短選項(如-i -o)
長選項格式舉例:
--version
--file=error.txt
讓一個指令碼同時支援短選項和長選項
getopt.getopt(sys.argv[1:], "hi:o:", ["version", "file="])

相關推薦

no