Python命令列引數

Python提供了一個getopt模組，用於解析命令列選項和引數。

$ python test.py arg1 arg2 arg3

Python sys模組通過sys.argv提供對任何命令列引數的存取。主要有兩個引數變數 -

• sys.argv是命令列引數的列表。
• len(sys.argv)是命令列引數的數量。

這裡sys.argv [0]是程式名稱，即指令碼的名稱。比如在上面範例程式碼中，sys.argv [0]的值就是 test.py。

範例

看看以下指令碼command_line_arguments.py的程式碼 -

#!/usr/bin/python3

import sys

print ('Number of arguments:', len(sys.argv), 'arguments.')
print ('Argument List:', str(sys.argv))

現在執行上面的指令碼，這將產生以下結果 -

F:\>python F:\worksp\python\command_line_arguments.py
Number of arguments: 1 arguments.
Argument List: ['F:\\worksp\\python\\command_line_arguments.py']

F:\>python F:\worksp\python\command_line_arguments.py arg1 arg2 arg3 arg4
Number of arguments: 5 arguments.
Argument List: ['F:\\worksp\\python\\command_line_arguments.py', 'arg1', 'arg2', 'arg3', 'arg4']

F:\>

注意 - 如上所述，第一個引數始終是指令碼名稱，它也被計入引數的數量。

解析命令列引數

Python提供了一個getopt模組，可用於解析命令列選項和引數。該模組提供了兩個功能和異常，以啟用命令列引數解析。

getopt.getopt方法

此方法解析命令列選項和引數列表。以下是此方法的簡單語法 -

getopt.getopt(args, options, [long_options])

getopt.GetoptError異常

當在引數列表中有一個無法識別的選項，或者當需要一個引數的選項不為任何引數時，會引發這個異常。
異常的引數是一個字串，指示錯誤的原因。 屬性msg和opt給出錯誤訊息和相關選項。

範例

假設想通過命令列傳遞兩個檔案名，也想給出一個選項用來顯示指令碼的用法。指令碼的用法如下 -

usage: file.py -i <inputfile> -o <outputfile>

以下是command_line_usage.py的以下指令碼 -

#!/usr/bin/python3

import sys, getopt

def main(argv):
   inputfile = ''
   outputfile = ''
   try:
      opts, args = getopt.getopt(argv,"hi:o:",["ifile=","ofile="])
   except getopt.GetoptError:
      print ('GetoptError, usage: command_line_usage.py -i <inputfile> -o <outputfile>')
      sys.exit(2)
   for opt, arg in opts:
      if opt == '-h':
         print ('usage: command_line_usage.py -i <inputfile> -o <outputfile>')
         sys.exit()
      elif opt in ("-i", "--ifile"):
         inputfile = arg
      elif opt in ("-o", "--ofile"):
         outputfile = arg
   print ('Input file is "', inputfile)
   print ('Output file is "', outputfile)

if __name__ == "__main__":
   main(sys.argv[1:])

現在，使用以下幾種方式執行來指令碼，輸出如下所示：

F:\worksp\python>python command_line_usage.py -h
usage: command_line_usage.py -i <inputfile> -o <outputfile>

F:\worksp\python>python command_line_usage.py -i inputfile.txt -o
GetoptError, usage: command_line_usage.py -i <inputfile> -o <outputfile>

F:\worksp\python>python command_line_usage.py -i inputfile.txt -o outputfile.txt
Input file is " inputfile.txt
Output file is " outputfile.txt

F:\worksp\python>