向python腳本傳遞參數
- 2020 年 1 月 9 日
- 筆記
import sys
print sys.argv[0]
sys.argv[0]是python腳本的名字；
下面的例子是一個用來提取文本log.txt中含有匹配字符串的文本行,匹配字符串的字符串當作參數傳遞給python腳本。
#!/usr/bin/env python
import sys
import re
f = open("log.txt", "rb")
info = open("info.txt", "ab")
for line in f.readlines():
if re.search(sys.argv[1], line):
info.write(line)
info.close()
f.close()
Python文件內部接收參數用：
sys.argv
sys.argv[0]：是python腳本名
sys.arg[1]及以後各項：是接收的參數
#! /bin/python
import sys
for arg in sys.argv:
print arg
需要模塊：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])
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="])
