Python argparse标准库快速入门

Python是一门很是好用的脚本语言，天然使用它来开发命令行程序也比较方便。并且Python的标准库中有一个名为argparse的库，能够很是方便的让咱们把命令行参数转换成所需的数据格式。下面就让咱们来看看如何使用argparse标准库吧。

若是想详细了解argparse的话，能够查看Python官方文档，目前已经出了中文版文档，极大的方便了咱们中国开发者，虽然有些地方没有完整翻译，可是已经彻底够用了。

初见argparse

首先来看看argparse的基本用法，下面是最简单的一个例子。能够看到argparse标准库其实用起来很简单，分红三个步骤：

1. 建立ArgumentParser对象
2. 使用add_argument方法添加参数
3. 使用parse_args方法接受并解析对象

下面咱们就依次来看看这些步骤吧。

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-from', type=str)
parser.add_argument('-to', type=str, default='everyone')

args = parser.parse_args('-from yitian'.split(' '))
print(args)

# 运行结果
# Namespace(from='yitian', to='everyone')

ArgumentParser

ArgumentParser是最重要的一个类，咱们要使用argparse标准库就必然须要建立这个类的实例。若是有须要的话，能够经过在构造函数中设置各类参数的方式来修改ArgumentParser类的行为。因为参数项比较多，因此官方文档建议咱们使用关键字参数的方式来添加参数。

• prog - 程序的名称（默认：sys.argv[0]），默认是Python程序的文件名
• usage - 描述程序用途的字符串（默认值：从添加到解析器的参数生成）
• description - 在参数帮助文档以前显示的文本（默认值：无）
• epilog - 在参数帮助文档以后显示的文本（默认值：无）
• parents - 一个 ArgumentParser 对象的列表，它们的参数也应包含在内。假如多个parser有一些能够共享的参数，能够经过设置子parser的方式来共享
• formatter_class - 用于自定义帮助文档输出格式的类
• prefix_chars - 可选参数的前缀字符集合（默认值：'-'）
• fromfile_prefix_chars - 当须要从文件中读取其余参数时，用于标识文件名的前缀字符集合（默认值：None）
• argument_default - 参数的全局默认值（默认值： None），假如须要给全部参数指定一个相通的默认值，能够修改这个，若是但愿全局禁用默认值，可使用argparse.SUPRESS
• conflict_handler - 解决冲突选项的策略（一般是没必要要的）
• add_help - 为解析器添加一个 -h/--help 选项（默认值： True）
• allow_abbrev - 若是缩写是无歧义的，则容许缩写长选项 （默认值：True）

add_argument方法

有了Parser实例，就能够调用它的add_argument方法来添加程序能够接受的参数了。这个参数比较复杂，功能也十分强大。

• name or flags - 一个命名或者一个选项字符串的列表，例如 foo 或 -f, --foo。
• action - 当参数在命令行中出现时使用的动做基本类型。
• nargs - 命令行参数应当消耗的数目。
• const - 被一些 action 和 nargs 选择所需求的常数。
• default - 当参数未在命令行中出现时使用的值。
• type - 命令行参数应当被转换成的类型。
• choices - 可用的参数的容器。
• required - 此命令行选项是否可省略 （仅选项可用）。
• help - 一个此选项做用的简单描述。
• metavar - 在使用方法消息中使用的参数值示例。
• dest - 被添加到 parse_args() 所返回对象上的属性名。

参数名

方法的第一个参数是参数名称，能够是一个字符串(name)或者是-开头的一组字符串(flags)，前者是位置参数，会按照添加的顺序被读取；后者是关键字参数，能够以任意顺序指定。若是指定的是关键字参数（flags），能够同时添加缩写和完整名，它们分别须要用-和--来作前缀。

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('foo')
parser.add_argument('bar')
parser.add_argument('-f', '--ff', action='store_true')

args = parser.parse_args('FOO BAR -f'.split(' '))
print(args)
args = parser.parse_args('--ff BAR FOO'.split(' '))
print(args)

'''运行结果，注意指定的参数和获取到的结果的不一样
Namespace(bar='BAR', ff=True, foo='FOO')
Namespace(bar='FOO', ff=True, foo='BAR')
'''

action（行为）

接下来介绍的参数都是关键字参数，能够按需制定。这里比较重要的一个是action，它指定了读取参数接下来要作什么。这部分能够看官方文档，介绍的比较全面，并且有对应的例子。我这里只简单介绍一下。

• store和store_const用来保存具体的值，他们之间有一些区别，一会再介绍。
• store_true和store_false比较方便，能够用来设置一些开关参数。例如我想指定-f的时候开启某个功能，忽略这个参数的时候不执行，就能够把它的行为指定成store_true，而后在程序中就能够获得f名字的参数真值，而后简单的条件判断就能够了。
• append和append_const会将参数存储成一个列表。
• count会存储参数出现的次数。常见用法是指定日志输出级别，例若有的程序-v会显示简单输出，-vv会显示复杂输出。

甚至若是需求更复杂的话，还能够本身实现一个新的Action类，而后添加给add_argument方法。

nargs（参数数目）

这个参数指定你的程序能够接受的参数个数，可使用如下几个值：

• N（一个正整数），表示后面的N个值会被读取为参数，注意指定为1的话会变成一个单元素列表。
• ?，和正则表达式里的概念差很少，后面的一个值会被读取，若是没有的话会从default读入。
• +，后面的多个值会被读取，若是没有会抛出异常。
• ?，后面的多个值会被读取，没有值的话也能够。
• argparse.REMAINDER，它会将后面全部值读取为一个参数，一般用做向其余命令行传递参数用。

默认状况下nargs会按照action的类型来判断参数个数，store和store_const会读取后面的一个值做为参数。

const

这个参数须要和带有const的action来配合使用。

default

指定参数的默认值。这里有一个颇有趣的点，若是你看官方文档比较仔细的话，可能会产生一个和我同样的疑问：store_const和const配合使用能够指定默认值，而store和default也能够指定默认值，那么它们之间有什么区别呢？其实区别仍是蛮大的，看看下面这个例子就明白了。

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-foo', action='store', nargs='?', default=10, const=20)

print(parser.parse_args('-foo 0'.split(' ')))
print(parser.parse_args('-foo'.split(' ')))
print(parser.parse_args(''))

'''
Namespace(foo='0')
Namespace(foo=20)
Namespace(foo=10)
'''

1. 当foo参数彻底忽略的时候，会使用default的值。
2. 当指定了foo参数，但没有指定后面的值时，会使用const的值。
3. 当指定了foo参数和后面的值时，会使用咱们指定的值。

type

这个是指定参数类型的，int、float、str那些。固然比较特殊的一个是open，它会将参数做为文件来打开。

若是默认的open还不能知足，还可使用argparse.FileType，它提供了读写模式、文件编码、缓冲区大小等详细设置。

parser.add_argument('bar', type=argparse.FileType('w'))

甚至有需求的话，这里还可使用咱们本身的函数，只要它的参数是一个字符串，返回值是转换之后的值就能够。

choices

若是确认参数范围限定是几个定值，可使用choices参数来指定，可接受的值包括字面值列表以及range函数。

required

指定参数是不是必须的。

metavar和dest

metavar参数用来指定参数的显示名称，而dest用来指定参数底层使用的属性名。

注意下面的程序输出，foo参数只修改了metavar，因此在帮助信息输出中发生了变化，可是在Namespace底层仍然使用foo保存值。而bar参数修改了dest，因此底层属性名发生了变化，可是在帮助信息中并无什么改变。

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-foo', metavar='foooo')
parser.add_argument('-bar', dest='barrrr')

parser.print_help()
print(parser.parse_args(''))

'''
usage: argparse_sample.py [-h] [-foo foooo] [-bar bar]

optional arguments:
  -h, --help  show this help message and exit
  -foo foooo
  -bar bar
Namespace(barrrr=None, foo=None)
'''

help

有条件的话最好给每一个参数添加帮助信息，这样使用者在用-h命令的时候就能够看到参数的帮助信息了。

parse_args方法

编辑好了参数，就能够调用parse_args方法来处理参数了，它会返回一个命名空间对象，包含了解析以后的参数。若是要测试方法的话，能够手动给它传递一组参数，不然的话，它会自动从命令行参数读取。另外它还支持几个比较有用的特性：

• 若是参数比较长，可使用等号来链接参数与值，例如-foo=bar。
• 若是参数是单字母长度的，能够将参数和值直接写在一块儿。

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-foo')
parser.add_argument('-b')

print(parser.parse_args('-foo=bar'.split()))
print(parser.parse_args('-bX'.split()))

'''
Namespace(b=None, foo='bar')
Namespace(b='X', foo=None)
'''

另外还支持默认无歧义的参数缩写。

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-f1aaaaaaaaaa')
parser.add_argument('-f2aaaaaaaaaa')

print(parser.parse_args('-f1 a -f2 b'.split(' ')))

命名空间对象

前面也看到了，解析完参数返回的值是命名空间对象，它用起来很是简单，直接访问属性值就能够了。

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-a')
parser.add_argument('-b')

args = parser.parse_args('-a a -b b'.split(' '))
print(args.a)

# a

其余特性

argparse还支持一些其余特性，这里就很少作介绍了，详情请直接参考官方文档。

• 子parser。有些程序支持子命令，这时候可使用子parser建立更复杂的parser。
• FileType。更详细的设置文件参数的方法。
• 参数组。若是参数比较多，可使用参数组的方式将功能相近的参数进行分组，这样不论是开发人员仍是使用人员都能更清晰的使用参数。
• 互斥参数组。若是某些参数不能同时使用，能够将它们加到互斥参数组中。
• 部分解析。默认状况下parse_argument在遇到未知参数的时候会报错，若是须要保存这些参数传递给其余命令行的时候，可使用parse_known_args方法。它不会由于未知参数报错， 并且会将全部不认识的参数存储成一个列表做为第二个返回值。

简单实例

扯了大半天，你们可能仍是有点不会用argparse，不过其实只要看一个简单的例子就能够了。保存下面的文件，而后用命令行调用，看看不一样的参数会有什么输出。

import argparse

parser = argparse.ArgumentParser(prog='ParserSample',
                                 description='简单实例程序，学习如何解析命令行参数',
                                 epilog='很简单就能够学会')

parser.add_argument('greeting', type=str, help='问候信息，必需')
parser.add_argument('-fromm', default='yitian', type=str, help='发送人，默认是易天')
parser.add_argument('-to', default='everyone', type=str, nargs='*', help='接收人，默认是全部人')
parser.add_argument('-p', action='store_true', help='是否添加感叹号')

args = parser.parse_args()

output = f'{args.fromm} say {args.greeting} to {args.to}'
if args.p:
    output = output + '!'

print(output)

'''
usage: ParserSample [-h] [-fromm FROMM] [-to [TO [TO ...]]] [-p] greeting

简单实例程序，学习如何解析命令行参数

positional arguments:
  greeting           问候信息，必需

optional arguments:
  -h, --help         show this help message and exit
  -fromm FROMM       发送人，默认是易天
  -to [TO [TO ...]]  接收人，默认是全部人
  -p                 是否添加感叹号

很简单就能够学会
'''