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

简介

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模块

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 201,468评论 5 473
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 84,620评论 2 377
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 148,427评论 0 334
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,160评论 1 272
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,197评论 5 363
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,334评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,775评论 3 393
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,444评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,628评论 1 295
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,459评论 2 317
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,508评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,210评论 3 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,767评论 3 303
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,850评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,076评论 1 258
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,627评论 2 348
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,196评论 2 341

推荐阅读更多精彩内容