美文网首页
python学习笔记之--命令行参数处理三(argparse)

python学习笔记之--命令行参数处理三(argparse)

作者: itsenlin | 来源:发表于2022-02-26 15:02 被阅读0次

    简介

    argparse相当于optparse的升级版,optparse模块已经不再维护,新代码最好使用argparse模块
    argparse对比optparse模块有以下优势

    • 处理位置参数。
    • 支持子命令。
    • 允许替代选项前缀例如 + 和 /。
    • 处理零个或多个以及一个或多个风格的参数。
    • 生成更具信息量的用法消息。
    • 提供用于定制 type 和 action 的更为简单的接口。

    用法步骤与optparse模块类似,先通过一个实例来整体看下argparse模块的用法

    一个实例

    import argparse
    
    parser = argparse.ArgumentParser(description='simple test for argparse')
    
    parser.add_argument('-o', '--output', dest='filename', help='write to file')
    parser.add_argument('-v', '--verbose', action='store_true', default=False, help='print status messages to stdout')
    parser.add_argument('-n', dest='num', metavar='NUMBER', type=int, default=0, help='print status messages to stdout')
    parser.add_argument('--version', action='version', version='1.0')
    
    subparser = parser.add_subparsers(help='sub command help')
    parser_a = subparser.add_parser('cal', help='help for sub commd "a"')
    parser_a.add_argument('--sum', dest='accumulate', action='store_const', const=sum, default=max, help='sum the integers (default: find the max)')
    parser_a.add_argument('integers', metavar='N', type=int, nargs='+', help='an integer for the accumulator')
    parser_a.add_argument('--version', action='version', version='1.0')
    
    args = parser.parse_args()
    
    print(args)
    print(args.accumulate(args.integers))
    

    从实例源码来看与optparse模块很像,也有一些新的特性,像子命令(subparser),并且帮助文档也比optparse模块更合理,像位置参数与选项参数分开显示,如下

    PS D:\scan\python> python .\argparseTest.py -h       
    usage: argparseTest.py [-h] [-o FILENAME] [-v] [-n NUMBER] [--version] {cal} ...
    
    simple test for argparse
    
    positional arguments:
      {cal}                 sub command help                       # 带大括号的表示子命令
        cal                 help for sub commd "a"
    
    optional arguments:
      -h, --help            show this help message and exit
      -o FILENAME, --output FILENAME
                            write to file
      -v, --verbose         print status messages to stdout
      -n NUMBER             print status messages to stdout
      --version             show program's version number and exit
    PS D:\scan\python>
    

    模块用法

    模块用法与optparse模块类似,也是分为三步:创建解析器对象、添加参数、解析参数

    创建解析器对象

    • 此模块无需显示的添加usage信息,可以根据后面添加的参数自动解析
    • 版本信息作为命令选项参数添加,而不再在这里添加
    • 可以使用prog参数指定usage信息中程序的名称,例如上面例子中默认显示的是argparseTest.py,可以通过prog='argparseTest'将后缀名去掉,也可以指定自定义的名称
    • description参数可以描述本程序的简介信息,是最常用的一个参数

    其他的参数可以参考官方文档:argumentparser-objects

    添加参数

    添加选项的接口原型

    ArgumentParser.add_argument(name or flags...[, action][, nargs][, const][, default][, type][, choices][, required][, help][, metavar][, dest])
    
    • name or flags表示选项名称,可以是短选项、也可以是长选项,也可以两者并存

    • action中增加了一个version,用于处理程序版本信息

    • nargs表示命令行参数应该消耗的数目,也即一个参数对应于命令行中非选项参数的个数,有以下几种取值

      1. nargs=[number]通过具体数值限制一个参数要对应到命令行中number个非选项参数,聚成一个列表,不管命令行中对应参数是多还是少,都会报错
        import argparse
        
        parser = argparse.ArgumentParser()
        parser.add_argument('a', nargs=1)
        parser.add_argument('-b', nargs=2)
        
        args = parser.parse_args()
        
        print(args)
        
        使用及输出如下:
        PS D:\scan\python> python .\testarg.py -h   
        usage: testarg.py [-h] [-b B B] a
        
        positional arguments:
          a
        
        optional arguments:
          -h, --help  show this help message and exit
          -b B B
        PS D:\scan\python> python .\testarg.py 1 
        Namespace(a=['1'], b=None)
        PS D:\scan\python>
        PS D:\scan\python> python .\testarg.py 1 -b 2
        usage: testarg.py [-h] [-b B B] a
        testarg.py: error: argument -b: expected 2 arguments
        PS D:\scan\python> python .\testarg.py 1 -b 2 3
        Namespace(a=['1'], b=['2', '3'])
        PS D:\scan\python> python .\testarg.py 0 1 -b 2 3
        usage: testarg.py [-h] [-b B B] a
        testarg.py: error: unrecognized arguments: 1
        PS D:\scan\python>
        
      2. nargs=?的一个更普遍用法是允许可选的输入或输出文件
        >>> parser = argparse.ArgumentParser()
        >>> parser.add_argument('infile', nargs='?', type=argparse.FileType('r'),
        ...                     default=sys.stdin)
        >>> parser.add_argument('outfile', nargs='?', type=argparse.FileType('w'),
        ...                     default=sys.stdout)
        >>> parser.parse_args(['input.txt', 'output.txt'])
        Namespace(infile=<_io.TextIOWrapper name='input.txt' encoding='UTF-8'>,
                  outfile=<_io.TextIOWrapper name='output.txt' encoding='UTF-8'>)
        >>> parser.parse_args([])
        Namespace(infile=<_io.TextIOWrapper name='<stdin>' encoding='UTF-8'>,
                  outfile=<_io.TextIOWrapper name='<stdout>' encoding='UTF-8'>)
        
      3. nargs=*,所有当前命令行参数被聚集到一个列表中。可以有0个或者多个
      4. nargs=+,所有当前命令行参数被聚集到一个列表中。可以有1个或者多个
    • 其他参数说明参考官方文档:the-add-argument-method

    另外一个比较常用的是子命令接口

    ArgumentParser.add_subparsers([title][, description][, prog][, parser_class][, action][, option_string][, dest][, required][, help][, metavar])
    

    参数说明参考官方文档:sub-commands

    解析参数

    接口原型

    ArgumentParser.parse_args(args=None, namespace=None)
    

    返回一个列表,包含添加的所有的参数,可以通过.来访问

    参考

    本章学习了argparse模块最常用的一些用法,大部分场景已经可以满足要求
    还有更多好用的特性,可以参考官方文档说明:argparse模块

    相关文章

      网友评论

          本文标题:python学习笔记之--命令行参数处理三(argparse)

          本文链接:https://www.haomeiwen.com/subject/xduqrrtx.html