码迷,mamicode.com
首页 > 编程语言 > 详细

Python命令行选项参数解析策略

时间:2016-03-18 10:54:34      阅读:302      评论:0      收藏:0      [点我收藏+]

标签:

概述

在Python的项目开发过程中,我们有时需要为程序提供一些可以通过命令行进行调用的接口。不过,并不是直接使用 command + 当前文件 就ok的,我们需要对其设置可选的各种各样的操作类型。所以,这种情况下我们就有必要对传入的参数进行解析操作。下面就此问题提出几种不同的解决策略,希望于你有益。


版权说明

著作权归作者所有。
商业转载请联系作者获得授权,非商业转载请注明出处。
作者:Coding-Naga
发表日期: 2016年3月18日
链接:http://blog.csdn.net/lemon_tree12138/article/details/50912898
来源:CSDN
更多内容:分类 >> Thinking In Python


目录


朴素字符串匹配方案

分析

其实此方法可以很直接地表达程序猿们在逻辑这条路走了多远。当然,这并没有包含任何的不敬。
这的确是一种方案,甚至可以说是一种算法。因为其直截了当,所以在程序开发初期,本人也是如此乐此不疲。除了完全的一个一个的命令之外,我们还可以对参数进行json封装,这样就更加体贴了。
关于,这一点我想也不用多说什么的了。还是留一些时间给后面的两种解决方案比较妥当。不过还是可以说说此方案的优点与缺点。

优点

  1. 针对不同的参数进行不同的处理,针对性强

缺点

  1. 因为其针对性太强,所以其复用性太差

getopt模块

简介

此模块是python内置的一个模块。该模块是专门用来处理命令行参数的。
其基本使用格式如下:

opts, args = getopt.getopt(args, shortopts, longopts = [])

对于getopt()方法的第一个参数,就是我们通过命令行传入的参数。不过这里也有一个值得注意的地方,我需要对参数列表进行切片处理。因为我们获得的第一个(args[0])命令行参数是当前文件名称,这并不是我们需要的。
对于getopt()的第二个参数,是shortopts;第三个参数,是longopts
shortopts比如:-h
longopts比如:–help
shortopts是以’-‘为前缀的,longopts是以”- -“为前缀的.

我们也可以单独使用短参数。基本使用格式如下:

opts, args = getopt.getopt(sys.argv[1:], "ld:")

实战用例

from __init__ import *

def usage():
    print ‘prama_config.py usage:‘
    print ‘-h, --help:      Print help message.‘
    print ‘-v, --version:   Print script version‘
    print ‘-o, --output:    Input an output verb‘
    print ‘-m, --message:   Send a message to someone.‘
    print ‘--foo:           Test option ‘
    print ‘--fre:           Another test option‘

def version():
    print ‘prama_config.py 1.0.1‘

def output(args):
    print ‘Hello, %s‘ % args

def message(sender, receiver, msg):
    print("{0} Send a message to {1}, content is \‘{2}\‘.".format(sender, receiver, msg))

def main(argv):
    try:
        opts, args = getopt.getopt(argv[1:], ‘hvom:‘, [‘help=‘, ‘message=‘, ‘foo=‘, ‘fre=‘])
    except getopt.GetoptError, err:
        print str(err)
        usage()
        sys.exit(2)
    for o, a in opts:
        if o in (‘-h‘, ‘--help‘):
            usage()
            sys.exit(1)
        elif o in (‘-v‘, ‘--version‘):
            version()
            sys.exit(0)
        elif o in (‘-o‘, ‘--output‘):
            output(a)
            sys.exit(0)
        elif o in (‘-m‘, ‘--message‘):
            message(a, args[0], args[1])
            sys.exit(0)
        else:
            print ‘unhandled option‘
            sys.exit(3)

if __name__ == ‘__main__‘:
    main(sys.argv)

OptionParser模块

简介

前面说到getopt,不过getopt太小了,而且从代码的角度看,面向过程的嫌疑很重。相对getopt,OptionParser就显得比较专业级了。OptionParser通过parser.add_option()添加选项参数,再通过parser.parse_args()进行解析参数选项。整个过程很面向对象
对于OptionParser还有一个优点在于,我们不需要为OptionParser设置help的选项,help选项已经被内置到模块当中去了。

parser.add_option()参数说明

  1. action: action是parse_args() 方法的参数之一,它指示 optparse 当解析到一个命令行参数时该如何处理。actions 有一组固定的值可供选择,默认是’store ‘,表示将命令行参数值保存在 options 对象里。action的取值有store, store_true, store_false三个;
  2. dest: dest是存储的变量,命令行的执行命令将会保存到dest指定的值当中。比如,下面代码中的-p命令,就会被保存到dest=”pdcl”指定的options的pdcl变量中;
  3. default: 用于设置上面dest中保存变量的默认值。比如下面的代码中,我们就将默认值打成了False。那么,我们通过options.pdclp这个变量访问到的值就是False;
  4. type: 用于指定dest中保存变量值的数据类型。默认的数据类型为string
  5. help: 用于指定当前命令的提示信息。

实战用例

from optparse import OptionParser

parser = OptionParser()

parser.add_option(
    "-p", "--pdbk",
    action="store_true",  # 指示 optparse 当解析到一个命令行参数时该如何处理
    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"
)

parser.add_option(
    "-f", "--file",  # 操作指令
    action="store",
    dest="filename",  # 存储的变量
    type="string",  # 变量类型
    help="write report to FILE",  # 显示的帮助信息
    metavar="FILE"  # 存储变量的值
)

parser.add_option(
    "-q", "--quiet",
    action="store_false",
    dest="verbose",
    default=True,
    help="don‘t print status messages to stdout"
)

(options, args) = parser.parse_args()

if options.pdcl is True:
    print ‘pdcl is true‘
if options.zdcl is True:
    print ‘zdcl is true‘
if options.filename is not None:
    print("filename={0}".format(options.filename))
    print(args)

通过对上面三种参数解析策略的说明,可以看到这里使用OptionParser模块进行解析是最佳方式。


Ref


Python命令行选项参数解析策略

标签:

原文地址:http://blog.csdn.net/lemon_tree12138/article/details/50912898

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!