本篇將介紹python中sys, getopt模塊處理命令行參數(shù)

如果想對python腳本傳參數(shù)，python中對應(yīng)的argc, argv(c語言的命令行參數(shù))是什么呢？

需要模塊：sys

參數(shù)個數(shù)：len(sys.argv)

腳本名：    sys.argv[0]
參數(shù)1：     sys.argv[1]
參數(shù)2：     sys.argv[2]

test.py

import sys
print "腳本名：", sys.argv[0]
for i in range(1, len(sys.argv)):
 print "參數(shù)", i, sys.argv[i]
>>>python test.py hello world

腳本名：test.py

參數(shù) 1 hello
參數(shù) 2 world

python中使用命令行選項：

例如我們需要一個convert.py腳本。它的作用是處理一個文件,并將處理后的結(jié)果輸出到另一個文件中。

要求該腳本滿足以下條件：

1.通過-i -o選項來區(qū)別參數(shù)是輸入文件還是輸出文件.

>>> python convert.py -i inputfile -o outputfile

2.當不知道convert.py需要哪些參數(shù)時，用-h打印出幫助信息

>>> python convert.py -h

getopt函數(shù)原形:

getopt.getopt(args, options[, long_options])
convert.py
python test.py -i inputfile -o outputfile

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:]為要處理的參數(shù)列表，sys.argv[0]為腳本名，所以用sys.argv[1:]過濾掉腳本名。

b) "hi:o:": 當一個選項只是表示開關(guān)狀態(tài)時，即后面不帶附加參數(shù)時，在分析串中寫入選項字符。當選項后面是帶一個附加參數(shù)時，在分析串中寫入選項字符同時后面加一個":"號。

　　所以"hi:o:"就表示"h"是一個開關(guān)選項；

  "i:"和"o:"則表示后面應(yīng)該帶一個參數(shù)。

c) 調(diào)用getopt函數(shù)。函數(shù)返回兩個列表：opts和args。

  opts為分析出的格式信息。opts是一個兩元組的列表。每個元素為：(選項串,附加參數(shù))。如果沒有附加參數(shù)則為空串''。

　　上面例子中的ops為：[('h',''),('-i','inputfile'),('-o','outputfile')]

  args為不屬于格式信息的剩余的命令行參數(shù)。

getopt函數(shù)的第三個參數(shù)[, long_options]為可選的長選項參數(shù)，上面例子中的都為短選項(如-i -o)

長選項格式舉例:

--version
--file=error.txt

讓一個腳本同時支持短選項和長選項

getopt.getopt(sys.argv[1:], "hi:o:", ["version", "file="])

總結(jié)

以上所述是小編給大家介紹的Python 獲得命令行參數(shù)的方法，希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復(fù)大家的。在此也非常感謝大家對腳本之家網(wǎng)站的支持！

最新評論