python-基础-命令行处理optparse

处理命令行

getopt简单
optparse强大
可以参考：https://docs.python.org/3/library/optparse.html
例子

from optparse import OptionParser 
parser = OptionParser() 
parser.add_option("-p", "--pdbk", action="store_true", 
                  dest="pdcl", 
                  default=False, 
                  help="write pdbk data to oracle db") 
parser.add_option("-z", "--zdbk", action="store_true", 
                  dest="zdcl", 
                  default=False, 
                  help="write zdbk data to oracle db") 

(options, args) = parser.parse_args() 

if options.pdcl==True: 
    print 'pdcl is true' 
if options.zdcl==True: 
    print 'zdcl is true' 

action有：store，store_true，store_false;
dest是存储的变量，default是缺省值，help是帮助提示;
执行parse_args()解析后获得选项；
可以输入如下等效命令

<yourscript> --file=outfile -q  
<yourscript> -f outfile --quiet  
<yourscript> --quiet --file outfile  
<yourscript> -q -foutfile  
<yourscript> -qfoutfile  

可以生成帮助

<yourscript> -h  
<yourscript> --help

传递参数方式

1. 命令行参数列表给parse_args()
2. 默认使用sys.argv[:1]

parse_args返回的两个值

1. options是一个对象optpars.Values，保存命令行参数，只要知道参数名，如file就可以访问对应的值：如options.file
2. args一个由positional arguments组成的列表

action指示当前命令如何处理，有一个固定值可供选择，store表示将命令行参数值保存在options对象里

parser.add_option("-f", "--file",  
                  action="store", type="string", dest="filename")  
args = ["-f", "foo.txt"]  
(options, args) = parser.parse_args(args)  
print options.filename

会打印出foo.txt
store两种形式：store_true 和 store_false用于处理不带值的情况，如：

parser.add_option("-v", action="store_true", dest="verbose")  
parser.add_option("-q", action="store_false", dest="verbose")  

当解析到 ‘-v’，options.verbose 将被赋予 True 值，反之，解析到 ‘-q’，会被赋予 False 值;
其他：store_const 、append 、count 、callback
默认值
方式1

parser.add_option("-f","--file", action="store", dest="filename", default="foo.txt")  
parser.add_option("-v", action="store_true", dest="verbose", default=True)

方式2:set_defaults()

parser.set_defaults(filename="foo.txt",verbose=True)  
parser.add_option(...)  
(options, args) = parser.parse_args() 

自动生成help帮助，需要在help中添加参数

usage = "usage: %prog [options] arg1 arg2"  
parser = OptionParser(usage=usage)  
parser.add_option("-v", "--verbose",  
                  action="store_true", dest="verbose", default=True,  
                  help="make lots of noise [default]")  
parser.add_option("-q", "--quiet",  
                  action="store_false", dest="verbose",  
                  help="be vewwy quiet (I'm hunting wabbits)")  
parser.add_option("-f", "--filename",  
                  metavar="FILE", help="write output to FILE"),  
parser.add_option("-m", "--mode",  
                  default="intermediate",  
              help="interaction mode: novice, intermediate, "  
                   "or expert [default: %default]")  

当 optparse 解析到 -h 或者 –help 命令行参数时，会调用 parser.print_help() 打印程序的帮助信息
打印完帮助信息后，optparse 将会退出，不再解析其它的命令行参数；

分组
OptonGroup

group = OptionGroup(parser, ``Dangerous Options'',  
                    ``Caution: use these options at your own risk.  ``  
                    ``It is believed that some of them bite.'')  
group.add_option(``-g'', action=''store_true'', help=''Group option.'')  
parser.add_option_group(group)

显示版本
创建 OptionParser 对象时，指定其 version 参数，用于显示当前程序的版本信息

parser = OptionParser(usage="%prog [-f] [-q]", version="%prog 1.0")  

这样就会自动解析 –version 命令行参数

$ /usr/bin/foo --version  
foo 1.0  

定义异常处理
parser.error()

(options, args) = parser.parse_args()  
[...]  
if options.a and options.b:  
    parser.error("options -a and -b are mutually exclusive")  

该例子：-a -b同时存在会报错
如果有更复杂的需求，可以继承OptionParser，重载exit() 和 erro() 方法

from optparse import OptionParser  
[...]  
def main():  
    usage = "usage: %prog [options] arg"  
    parser = OptionParser(usage)  
    parser.add_option("-f", "--file", dest="filename",  
                      help="read data from FILENAME")  
    parser.add_option("-v", "--verbose",  
                      action="store_true", dest="verbose")  
    parser.add_option("-q", "--quiet",  
                      action="store_false", dest="verbose")  
    [...]  
    (options, args) = parser.parse_args()  
    if len(args) != 1:  
        parser.error("incorrect number of arguments")  
    if options.verbose:  
        print "reading %s..." % options.filename  
    [...]  
  
if __name__ == "__main__":  
    main()