Python学习教程：python设置执行选项参数

一 . sys

可以通过sys.argv得到参数列表，其中sys.argv[0]是文件名称，此后的列表值是每一个参数，这里的参数列表也支持常规的列表操作

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

二 . Argparse

Argparse模块可以让人轻松编写用户友好的命令行接口,支持自动生成帮助和使用手册，并在用户给程序传入参数时报出错误信息

操作流程

I. 导入库import argparse

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

III. 增加参数parser.add_argument()

IV. 解析参数args = parser.parse_args()

使用

参数分为位置参数和可选参数，通常会结合使用，分为以下三种情况

1. 布尔型可选参数

只要指明了--verbose关键字，则会将值设为True，并作出对应响应

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. 指定值的可选参数

指定可选参数的几个响应值，只能从指定值从指定参数值并做处理

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. 指定可选参数长度(次数)

利用count来通过次数给定参数的响应操作，这里还给定了default值是因为默认情况下未指定可选参数则它将获取None值，无法进行数值比较

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

#!/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