python基礎之 sys.argv[]用法


sys.argv[]是用來獲取命令行參數的,sys.argv[0]表示代碼本身文件路徑,所以參數從1開始.

 

arg[1]表示第一個命令行參數

arg[1][2:] 表示取第一個命令行參數,但是去掉前兩個字節

比如命令行參數為   “--help” ,就去掉“--”,執行"help"參數。

以下代碼來自簡明Python教程。

這里如果有sys.arg[0]則表示cat.py
[python]  view plain copy 在CODE上查看代碼片 派生到我的代碼片
  1. #!/usr/bin/python  
  2. # Filename: cat.py  
  3.   
  4. import sys  
  5.   
  6. def readfile(filename):  
  7.     '''''Print a file to the standard output.'''  
  8.     f = file(filename)  
  9.     while True:  
  10.         line = f.readline()  
  11.         if len(line) == 0:  
  12.             break  
  13.         print line, # notice comma  
  14.     f.close()  
  15.   
  16. # Script starts from here  
  17. if len(sys.argv) < 2:  
  18.     print 'No action specified.'  
  19.     sys.exit()  
  20.   
  21. if sys.argv[1].startswith('--'):  
  22.     option = sys.argv[1][2:]  
  23.     # 取 sys.argv[1] but without the first two characters,這里去掉“--”  
  24.     if option == 'version':  
  25.         print 'Version 1.2'  
  26.     elif option == 'help':  
  27.         print '''''\ 
  28. This program prints files to the standard output. 
  29. Any number of files can be specified. 
  30. Options include: 
  31.   --version : Prints the version number 
  32.   --help    : Display this help'''  
  33.     else:  
  34.         print 'Unknown option.'  
  35.     sys.exit()  
  36. else:  
  37.     for filename in sys.argv[1:]:  
  38.         readfile(filename)  

    在運行程序時,可能需要根據不同的條件,輸入不同的命令行選項來實現不同的功能。目前有短選項長選項兩種格式。短選項格式為"-"加上單個字母選項;長選項為"--"加上一個單詞。長格式是在Linux下引入的。許多Linux程序都支持這兩種格式。在Python中提供了getopt模塊很好的實現了對這兩種用法的支持,而且使用簡單。


取得命令行參數
  在使用之前,首先要取得命令行參數。使用sys模塊可以得到命令行參數。
import sys
print sys.argv

  然后在命令行下敲入任意的參數,如:
python get.py -o t --help cmd file1 file2

  結果為:
['get.py', '-o', 't', '--help', 'cmd', 'file1', 'file2']

  可見,所有命令行參數以空格為分隔符,都保存在了sys.argv列表中。其中第1個為腳本的文件名。

選項的寫法要求
  對於短格式,"-"號后面要緊跟一個選項字母。如果還有此選項的附加參數,可以用空格分開,也可以不分開。長度任意,可以用引號。如以下是正確的:
-o
-oa
-obbbb
-o bbbb
-o "a b"
  對於長格式,"--"號后面要跟一個單詞。如果還有些選項的附加參數,后面要緊跟"=",再加上參數。"="號前后不能有空格。如以下是正確的:

--help=file1

  而這些是不正確的:
-- help=file1
--help =file1
--help = file1
--help= file1

如何用getopt進行分析
  使用getopt模塊分析命令行參數大體上分為三個步驟:

1.導入getopt, sys模塊
2.分析命令行參數
3.處理結果

  第一步很簡單,只需要:
import getopt, sys

  第二步處理方法如下(以Python手冊上的例子為例):
try:
    opts, args = getopt.getopt(sys.argv[1:], "ho:", ["help", "output="])
except getopt.GetoptError:
    # print help information and exit:

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

  使用一個循環,每次從opts中取出一個兩元組,賦給兩個變量。o保存選項參數,a為附加參數。接着對取出的選項參數進行處理。(例子也采用手冊的例子)


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM