python中怎么設(shè)置執(zhí)行選項(xiàng)參數(shù)

本篇文章為大家展示了python中怎么設(shè)置執(zhí)行選項(xiàng)參數(shù)，內(nèi)容簡明扼要并且容易理解，絕對能使你眼前一亮，通過這篇文章的詳細(xì)介紹希望你能有所收獲。

一 . sys

可以通過sys.argv得到參數(shù)列表，其中sys.argv[0]是文件名稱，此后的列表值是每一個參數(shù)，這里的參數(shù)列表也支持常規(guī)的列表操作

 for data in sys.argv：
 print("第{}個參數(shù)是：{}".format(sys.argv.index(data), data))
--------------------------------------------------------------------------------------
python mark_test.py first second three four -u username
第0個參數(shù)是：mark_test.py
第1個參數(shù)是：first
第2個參數(shù)是：second
第3個參數(shù)是：three
第4個參數(shù)是：four
第5個參數(shù)是：-u
第6個參數(shù)是：username

二 . Argparse

Argparse模塊可以讓人輕松編寫用戶友好的命令行接口,支持自動生成幫助和使用手冊，并在用戶給程序傳入無效參數(shù)時報出錯誤信息

操作流程

I. 導(dǎo)入庫import argparse

II. 初始化解析器parser = argparse.ArgumentParser()

III. 增加參數(shù)parser.add_argument()

IV. 解析參數(shù)args = parser.parse_args()

使用

參數(shù)分為位置參數(shù)和可選參數(shù)，通常會結(jié)合使用，分為以下三種情況

1. 布爾型可選參數(shù)

只要指明了--verbose關(guān)鍵字，則會將值設(shè)為True，并作出對應(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

2. 指定值的可選參數(shù)

指定可選參數(shù)的幾個響應(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

3. 指定可選參數(shù)長度(次數(shù))

利用count來通過次數(shù)給定參數(shù)的響應(yīng)操作，這里還給定了default值是因?yàn)槟J(rèn)情況下未指定可選參數(shù)則它將獲取None值，無法進(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

長格式和短格式是可以混合使用的，參考之前的例子，更多可參考Argparse

三 . Fire

Python Fire是一個Python庫，只需一次調(diào)用即可將任何Python組件轉(zhuǎn)換為命令行界面 Fire

1. 安裝

pip install fire

2. 使用

Fire解析過程較為簡單，在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ù)名，類也只需要指明其內(nèi)的函數(shù)名即可

• 添加多個函數(shù)而非全部調(diào)用可使用

fire.Fire({
 'newadd': newadd,
 'newprint': newprint,
})

使用類名或者構(gòu)造對象用法大都類似

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

四 . getopt

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

• args: 要解析的命令行參數(shù)列表。

• options : 以字符串的格式定義，options 后的冒號 : 表示如果設(shè)置該選項(xiàng)，必須有附加的參數(shù)，否則就不附加參數(shù)。

• long_options : 以列表的格式定義，long_options 后的等號 = 表示該選項(xiàng)必須有附加的參數(shù)，不帶冒號表示該選項(xiàng)不附加參數(shù)。

• 該方法返回值由兩個元素組成: 第一個是 (option, value) 元組的列表。 第二個是參數(shù)列表，包含那些沒有 - 或 – 的參數(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:]過濾掉腳本名。

上述內(nèi)容就是python中怎么設(shè)置執(zhí)行選項(xiàng)參數(shù)，你們學(xué)到知識或技能了嗎？如果還想學(xué)到更多技能或者豐富自己的知識儲備，歡迎關(guān)注億速云行業(yè)資訊頻道。