python argpase模块简单使用

python2.7

手册地址：https://docs.python.org/2/howto/argparse.html#id1

实现效果：脚本程序可以带参数

python arg.py -h

一、位置参数

示例代码：

 import argparse        
 parser = argparse.ArgumentParser()      # 实例化
 parser.add_argument("square", help="display a square of a given number",
                     type=int)   #  名称  help  帮助信息   type  接受参数类型，默认为字符串
 args = parser.parse_args()    #接受的参数
 print args.square**2

使用

 $ python prog.py 4
 16
 $ python prog.py four
 usage: prog.py [-h] square
 prog.py: error: argument square: invalid int value: 'four'

二、可选参数

实例代码

 import argparse
 parser = argparse.ArgumentParser()
 parser.add_argument("-v", "--verbose", help="increase output verbosity",
                     action="store_true")   # action 为"store_true"或"store_false"
 args = parser.parse_args()
 if args.verbose:     #注意verbose,和前面名字相同
     print "verbosity turned on"

使用

 $ python prog.py -v
 verbosity turned on
 $ python prog.py --help
 usage: prog.py [-h] [-v]
 
 optional arguments:
   -h, --help     show this help message and exit
   -v, --verbose  increase output verbosity

三、混合类型

实例代码

 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

可以增加choices=[0, 1, 2]选项
 
使用

 $ python prog.py 4 -vvv
 the square of 4 equals 16
 $ python prog.py 4 -vvvv
 the square of 4 equals 16
 $ python prog.py 4
 16

 
四、增加冲突选项，多个选项只能选一个

 import argparse
 
 parser = argparse.ArgumentParser(description="calculate X to the power of Y") #描述
 group = parser.add_mutually_exclusive_group()
 group.add_argument("-v", "--verbose", action="store_true")
 group.add_argument("-q", "--quiet", action="store_true")
 parser.add_argument("x", type=int, help="the base")
 parser.add_argument("y", type=int, help="the exponent")
 args = parser.parse_args()
 answer = args.x**args.y
 
 if args.quiet:
     print answer
 elif args.verbose:
     print "{} to the power {} equals {}".format(args.x, args.y, answer)
 else:
     print "{}^{} == {}".format(args.x, args.y, answer)

使用

 python prog.py --help   #prog.py为脚本名称
 usage: prog.py [-h] [-v | -q] x y
 
 calculate X to the power of Y
 
 positional arguments:
   x              the base
   y              the exponent
 
 optional arguments:
   -h, --help     show this help message and exit
   -v, --verbose
   -q, --quiet