Posted by:
努力记

努力记,通过记录,计划,让自己的努力可以触碰彩虹!

10,732

Python 命令行参数解析

  • 2014-12-19
  • 暂无评论
Python用于编写脚本的场景非常多,如何处理脚本(命令行)的参数当然非常关键
首先导入sys,参数列表在sys.arg中
import sys

if __name__ == '__main__':
    for arg in sys.argv:
        print arg

 

可以发现sys.arg[0],也就是第一个参数是执行脚本的名字,所以一般获取参数是sys.arg[1:]
如果参数比较简单,或者做给自己用用,那就自己判断下吧
但如果想让脚本看起来专业点:比如支持长短格式,长格式添加=,帮助信息等完善的参数操作,通过相应的类库可以较快实现
1.getopt
import sys
import getopt


def main():
    try:
        opts, args = getopt.getopt(sys.argv[1:],'d:f:h',['days=','files=','help']) # 当一个选项只是表示开关状态时,即后面不带附加参数时,在分析串中写入选项字符。当选项后面是带一个附加参数时,在分析串中写入选项字符同时后面加一个":"号
    except getopt.GetoptError:
        usage()
        sys.exit()
    print(opts)
    print(args)
    
    for o, a in opts:
        if o in ("-h", "--help"):
            usage()
            sys.exit()
        elif o in ("-d","--days"):
            day = a
        elif o in ("-f", "--files"):
            files = a
    print(day)
    print(files)


def usage():
    print """
    usage:xxxxxxx
    """

if __name__ == '__main__':
    main()

 

如果要实现完整的功能,以上应该算一个基本的结构了,对于小功能的设置应该能满足了
但!整个过程还是相对零散,对于更加复杂的应用,编写的代码简直无法直视
下面介绍2款完善的类库,是官方推荐的
功能强大,易于使用,方便地生成标准的、符合Unix/posix规范的命令行说明
2.optparse
import optparse


def main():
    parser = optparse.OptionParser(usage="%prog -f server.list -u root", version="%prog 1")
    parser.add_option("-f", "--file", dest="File", action="store", help="Host list")
    parser.add_option("-u", "--user",action="store", dest="User",help="Username")
    parser.add_option('-v', '–version', action='store_true', dest='version', default=False, help='show the version of this command')
    #store_true这个action的意思是只有在使用该参数的时候存储, default则可以给一个默认值。也就是你加了-v 显示版本信息,否则不显示。
    (options, args) = parser.parse_args()
    """
    1. options , 这是一个对象(optpars.Values),保存命令行参数值。只要知道命令行参数名,如file,就可以访问其对应的值:options.file。
    2. args , 一个由 positional arguments 组成的列表
    """


if __name__ == '__main__':
    main()

 

已经能非常方便的进行参数解析,并且还提供自动help、参数分组、版本提示等
3.argparse
应该是对于optparse的一个升级,官方在2.7版本提议用argparse代替optparse
代码更易理解,功能基本和optparse一致
import argparse


def main():
    parser = argparse.ArgumentParser(usage="%prog -f server.list -u root", version="%prog 1")
    parser.add_argument("-f", "--file", dest="File", action="store", help="Host list")
    parser.add_argument("-u", "--user",action="store", dest="User",help="Username")
    parser.add_argument('-v', '–version', action='store_true', dest='version', default=False, help='show the version of this command') 

    options = parser.parse_args()

    # 如果运行程序带了一个不认识的参数,就会报错,可以使用下面的接口做不同处理
    options, args = parser.parse_known_args()
    

if __name__ == '__main__':
    main()

 

以上2者使用基本相同,代码上基本使用基本只要替换接口名称即可
但argparse的自动help及usage能显示的更加专业,或许还有更深层次的不同


back up ↑

无觅相关文章插件,快速提升流量