本文是从我另一个博客转载过来的,欢迎大家点击进去看一下,帮我增加点人气^_^
选择模块
根据python参考手册的提示,optparse 已经废弃,应使用 argparse
教程
概念
argparse 模块使用 add_argument 来添加可选的命令行参数,原型如下:
ArgumentParser.add_argument(name or flags...[, action][, nargs][, const][, default][, type][, choices][, required][, help][, metavar][, dest])
Define how a single command-line argument should be parsed. Each parameter has its own more detailed description below, but in short they are:
name or flags - Either a name or a list of option strings, e.g. foo or -f, --foo.
action - The basic type of action to be taken when this argument is encountered at the command line.
nargs - The number of command-line arguments that should be consumed.
const - A constant value required by some action and nargs selections.
default - The value produced if the argument is absent from the command line.
type - The type to which the command-line argument should be converted.
choices - A container of the allowable values for the argument.
required - Whether or not the command-line option may be omitted (optionals only).
help - A brief description of what the argument does.
metavar - A name for the argument in usage messages.
dest - The name of the attribute to be added to the object returned by parse_args().
上面的说明其实不用看,直接看示例好了:
仅仅想展示一些信息
# -*- coding: utf-8 -*-
"""
argparse tester
"""
import argparse
parser = argparse.ArgumentParser(description='argparse tester')
parser.add_argument("-v", "--verbose", help="increase output verbosity",
action="store_true")
args = parser.parse_args()
if args.verbose:
print "hello world"
它会输出如下:
$ python t.py
$ python t.py -v
hello world
$ python t.py -h
usage: t.py [-h] [-v]
argparse tester
optional arguments:
-h, --help show this help message and exit
-v, --verbose increase output verbosity
这里 -v 是这个选项的简写,--verbose 是完整拼法,都是可以的
help 是帮助信息,不解释了
args = parse_args() 会返回一个命名空间,只要你添加了一个可选项,比如 verbose,它就会把 verbose 加到 args 里去,就可以直接通过 args.verbose 访问。
如果你想给它起个别名,就需要在 add_argument 里加多一个参数 dest='vb'
这样你就可以通过 args.vb 来访问它了。
action="store_true" 表示该选项不需要接收参数,直接设定 args.verbose = True,
当然如果你不指定 -v,那么 args.verbose 就是 False
但如果你把 action="store_true" 去掉,你就必须给 -v 指定一个值,比如 -v 1
做个求和程序
# -*- coding: utf-8 -*-
"""
argparse tester
"""
import argparse
parser = argparse.ArgumentParser(description='argparse tester')
parser.add_argument("-v", "--verbose", help="increase output verbosity", action="store_true")
parser.add_argument('numbers', type=int, help="numbers to calculate", nargs='+')
parser.add_argument('-s', '--sum', help="sum all numbers", action='store_true', default=True)
args = parser.parse_args()
print "Input:", args.numbers
print "Result:"
results = args.numbers
if args.verbose:
print "hello world"
if args.sum:
results = sum(args.numbers)
print "tSum:tt%s" % results
输出如下:
[pan@pan-pc test]$ python t2.py -h
usage: t2.py [-h] [-v] [-s] numbers [numbers ...]
argparse tester
positional arguments:
numbers numbers to calculate
optional arguments:
-h, --help show this help message and exit
-v, --verbose increase output verbosity
-s, --sum sum all numbers
[pan@pan-pc test]$ python t2.py 1 2 3 -s
Input: [1, 2, 3]
Result:
Sum: 6
注意到这此可选项 numbers 不再加上 “-” 前缀了,因为这个是位置选项
如果把 nargs="+" 去掉,则只能输入一个数字。因为它指定了number 选项的值个数
如果把 type=int 去掉,则 args.numbers 就是一个字符串,而不会自动转换为整数
注意到 --sum 选项的最后还加上了 default=True,意思是即使你不在命令行中指定 -s,它也会默认被设置为 True
只能2选1的可选项
# -*- coding: utf-8 -*-
"""
argparse tester
"""
import argparse
parser = argparse.ArgumentParser(description='argparse tester')
#group = parser.add_mutually_exclusive_group()
parser.add_argument("-v", "--verbose", help="increase output verbosity", action="store_true")
parser.add_argument('numbers', type=int, help="numbers to calculate", nargs='+')
parser.add_argument('-s', '--sum', help="sum all numbers", action='store_true', default=True)
parser.add_argument('-f', '--format', choices=['int', 'float'], help='choose result format', default='int')
args = parser.parse_args()
print "Input:", args.numbers
print "Result:"
results = args.numbers
if args.verbose:
print "hello world"
if args.format == 'int':
format = '%d'
else:
format = '%f'
if args.sum:
results = sum(args.numbers)
print 'tsum:tt%s' % (format % results)
输出如下:
[pan@pan-pc test]$ python t2.py 1 2 3 -f float
Input: [1, 2, 3]
Result:
sum: 6.000000
[pan@pan-pc test]$ python t2.py 1 2 3 -f double
usage: t2.py [-h] [-v] [-s] [-f {int,float}] numbers [numbers ...]
t2.py: error: argument -f/--format: invalid choice: 'double' (choose from 'int', 'float')
在添加选项 -f 时,传入了 choices=['int', 'float'] 参数,表示该选项只能从 int 或 float 中2选1