1. 程式人生 > >Python獲取命令列引數

Python獲取命令列引數

sys.argv[]

包含命令列引數的字串列表,通過下標獲取引數。

例如:

#!/usr/bin/python
# Filename: using_sys.py
  
import sys
  
print 'The command line arguments are:'
for i in sys.argv:
    print i
  
print '\n\nThe PYTHONPATH is', sys.path, '\n'
print argv[1]

argv[0]表示檔案本身路徑。 當然,agv[]也可存放多個值

getopt

用於抽出命令列選項和引數,也就是sys.argv。命令列選項使得程式的引數更加靈活。支援短選項模式和長選項模式。

import getopt
#python scriptname.py -f 'hello' --directory-prefix=/home -t --form    at 'a' 'b'
shortargs = 'f:t'
longargs = ['directory-prefix=', 'format', '--f_long=']
opts, args = getopt.getopt( sys.argv[1:], shortargs, longargs )

getopt函式的格式是getopt.getopt ( [命令列引數列表], "短選項", [長選項列表] )
短選項名後的冒號(:)表示該選項必須有附加的引數。


長選項名後的等號(=)表示該選項必須有附加的引數。
返回opts和args。

opts是一個引數選項及其value的元組( ( '-f', 'hello'), ( '-t', '' ), ( '--format', '' ), ( '--directory-prefix', '/home' ) )

args是一個除去有用引數外其他的命令列輸入 ( 'a', 'b' )

然後遍歷opts便可以獲取所有的命令列選項及其對應引數了。

遍歷opts可以獲取所有命令的選項及引數,

for opt, val in opts:
    if opt in ( '-f', '--f_long' ):
        pass
    if ....


 分析不同宣傳項引數,做不同處理。 一般,選項引數列表會被列印作為幫助選項。

python Documentation中也給出了getopt的典型使用方法:

import getopt, sys

def main():
    try:
        opts, args = getopt.getopt(sys.argv[1:], "ho:v", ["help", "output="])
    except getopt.GetoptError, err:
        # print help information and exit:
        print str(err) # will print something like "option -a not recognized"
        usage()
        sys.exit(2)
    output = None
    verbose = False
    for o, a in opts:
        if o == "-v":
            verbose = True
        elif o in ("-h", "--help"):
            usage()
            sys.exit()
        elif o in ("-o", "--output"):
            output = a
        else:
            assert False, "unhandled option"
    # ...

if __name__ == "__main__":
    main()