野生技术宅协会

Python argparse标准库快速入门

2020-02-04  本文已影响0人  乐百川

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类的行为。由于参数项比较多，所以官方文档建议我们使用关键字参数的方式来添加参数。

add_argument方法

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

参数名

方法的第一个参数是参数名称，可以是一个字符串(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，它指定了读取参数接下来要做什么。这部分可以看官方文档，介绍的比较全面，而且有对应的例子。我这里只简单介绍一下。

甚至如果需求更复杂的话，还可以自己实现一个新的Action类，然后添加给add_argument方法。

nargs（参数数目）

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

默认情况下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方法来处理参数了，它会返回一个命名空间对象，包含了解析之后的参数。如果要测试方法的话，可以手动给它传递一组参数，否则的话，它会自动从命令行参数读取。另外它还支持几个比较有用的特性：

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还支持一些其他特性，这里就不多做介绍了，详情请直接参考官方文档。

简单实例

扯了大半天，大家可能还是有点不会用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                 是否添加感叹号

很简单就可以学会
'''
上一篇 下一篇

猜你喜欢

热点阅读