1. sys
????可以通過(guò)
sys.argv
得到參數(shù)列表谦絮,其中sys.argv[0]
是文件名稱恋技,此后的列表值是每一個(gè)參數(shù),這里的參數(shù)列表也支持常規(guī)的列表操作
for data in sys.argv:
print("第{}個(gè)參數(shù)是:{}".format(sys.argv.index(data), data))
--------------------------------------------------------------------------------------
python mark_test.py first second three four -u username
第0個(gè)參數(shù)是:mark_test.py
第1個(gè)參數(shù)是:first
第2個(gè)參數(shù)是:second
第3個(gè)參數(shù)是:three
第4個(gè)參數(shù)是:four
第5個(gè)參數(shù)是:-u
第6個(gè)參數(shù)是:username
2. Argparse
????
Argparse
模塊可以讓人輕松編寫(xiě)用戶友好的命令行接口,支持自動(dòng)生成幫助和使用手冊(cè)邮旷,并在用戶給程序傳入無(wú)效參數(shù)時(shí)報(bào)出錯(cuò)誤信息
操作流程
I. 導(dǎo)入庫(kù)import argparse
II. 初始化解析器parser = argparse.ArgumentParser()
III. 增加參數(shù)parser.add_argument()
IV. 解析參數(shù)args = parser.parse_args()
使用
參數(shù)分為位置參數(shù)和可選參數(shù)累舷,通常會(huì)結(jié)合使用,分為以下三種情況
I. 布爾型可選參數(shù)
只要指明了
--verbose
關(guān)鍵字壤圃,則會(huì)將值設(shè)為True
陵霉,并作出對(duì)應(yīng)響應(yīng)
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("--verbose", help="increase output verbosity",
action="store_true")
args = parser.parse_args()
if args.verbose:
print("verbosity turned on")
-------------------------------------------------------------------------------------------
python mark_test.py 4 --verbose
#verbosity turned on
--------------------------------------------------------------------------------------------
python mark_test.py --ajschahc 2
#usage: mark_test.py [-h] [--verbose]
#mark_test.py: error: unrecognized arguments: --ashjca 1
II. 指定值的可選參數(shù)
指定可選參數(shù)的幾個(gè)響應(yīng)值,只能從指定值從指定參數(shù)值并做處理
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("square", type=int,
help="display a square of a given number")
parser.add_argument("-v", "--verbosity", type=int, choices=[0, 1, 2],
help="increase output verbosity")
args = parser.parse_args()
answer = args.square**2
if args.verbosity == 2:
print("the square of {} equals {}".format(args.square, answer))
elif args.verbosity == 1:
print("{}^2 == {}".format(args.square, answer))
else:
print(answer)
-----------------------------------------------------------------------------------------------
python mark_test.py 4 -v 3
#usage: mark_test.py [-h] [-v {0,1,2}] square
#mark_test.py: error: argument -v/--verbosity: invalid choice: 3 (choose from 0, 1, 2)
-----------------------------------------------------------------------------------------------
python mark_test.py 4 -v 2
#the square of 4 equals 16
III. 指定可選參數(shù)長(zhǎng)度(次數(shù))
????利用
count
來(lái)通過(guò)次數(shù)給定參數(shù)的響應(yīng)操作埃唯,這里還給定了default值是因?yàn)槟J(rèn)情況下未指定可選參數(shù)則它將獲取None
值,無(wú)法進(jìn)行數(shù)值比較
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("square", type=int,
help="display a square of a given number")
parser.add_argument("-v", "--verbosity", action="count", default=0,
help="increase output verbosity")
args = parser.parse_args()
answer = args.square**2
if args.verbosity >= 2:
print("the square of {} equals {}".format(args.square, answer))
elif args.verbosity >= 1:
print("{}^2 == {}".format(args.square, answer))
else:
print(answer)
--------------------------------------------------------------------------------
python mark_test.py 4 -v
#the square of 4 equals 16
--------------------------------------------------------------------------------
python mark_test.py 4 -vv
#4^2 == 16
- 長(zhǎng)格式和短格式是可以混合使用的鹰晨,參考之前的例子墨叛,更多可參考Argparse
3. Fire
????Python Fire是一個(gè)Python庫(kù),只需一次調(diào)用即可將任何Python組件轉(zhuǎn)換為命令行界面 Fire
I. 安裝
pip install fire
II. 使用
????Fire解析過(guò)程較為簡(jiǎn)單模蜡,在cmd中直接調(diào)用py文件中的函數(shù)漠趁、變量、類忍疾、實(shí)例等等
import fire
def newprint(text):
print('my '+text)
def newadd(a,b):
return a + b
fire.Fire() # 只要這一條命令
----------------------------------------------------------------
python cmd.py newprint notebook
#my notebook
----------------------------------------------------------------
python cmd.py newadd 2 3
#python cmd.py newadd --a 2 --b 3
- 在函數(shù)中將
fire.Fire
改為固定的方法或者類名闯传,則僅執(zhí)行該實(shí)例,如果是函數(shù)卤妒,調(diào)用時(shí)則不再需要指定函數(shù)名甥绿,類也只需要指明其內(nèi)的函數(shù)名即可 - 添加多個(gè)函數(shù)而非全部調(diào)用可使用
fire.Fire({
'newadd': newadd,
'newprint': newprint,
})
使用類名或者構(gòu)造對(duì)象用法大都類似
import fire
class Myclass:
def __init__(self, name):
self.name = name
def nameprint(self, parm):
print(parm+ ', I am ' + self.name)
fire.Fire(Myclass)
-----------------------------------------------------------------
python cmd.py nameprint Yes --name June
#Yes, I am June
4. getopt
getopt.getopt(args, options[, long_options])
- args:要解析的命令行參數(shù)列表。
-
options : 以字符串的格式定義则披,
options
后的冒號(hào) : 表示如果設(shè)置該選項(xiàng)共缕,必須有附加的參數(shù),否則就不附加參數(shù)士复。 -
long_options : 以列表的格式定義图谷,
long_options
后的等號(hào) = 表示該選項(xiàng)必須有附加的參數(shù)翩活,不帶冒號(hào)表示該選項(xiàng)不附加參數(shù)。 - 該方法返回值由兩個(gè)元素組成: 第一個(gè)是
(option, value)
元組的列表便贵。 第二個(gè)是參數(shù)列表菠镇,包含那些沒(méi)有 - 或 -- 的參數(shù)。
#!/usr/bin/python
# -*- coding: UTF-8 -*-
import sys, getopt
def main(argv):
inputfile = ''
outputfile = ''
try:
opts, args = getopt.getopt(argv,"hi:o:",["ifile=","ofile="])
except getopt.GetoptError:
print 'test.py -i <inputfile> -o <outputfile>'
sys.exit(2)
for opt, arg in opts:
if opt == '-h':
print 'test.py -i <inputfile> -o <outputfile>'
sys.exit()
elif opt in ("-i", "--ifile"):
inputfile = arg
elif opt in ("-o", "--ofile"):
outputfile = arg
print '輸入的文件為:', inputfile
print '輸出的文件為:', outputfile
if __name__ == "__main__":
main(sys.argv[1:])
---------------------------------------------------------------------------
$ python test.py -h
#usage: test.py -i <inputfile> -o <outputfile>
---------------------------------------------------------------------------
$ python test.py -i inputfile -o outputfile
#輸入的文件為: inputfile
#輸出的文件為: outputfile
-
sys.argv[1:]
為要處理的參數(shù)列表承璃,sys.argv[0]
為腳本名利耍,所以用sys.argv[1:]
過(guò)濾掉腳本名。