【Python】Python脚本命令行解析

Python脚本命令行解析

argparse 模块
使用 argparse 模块可以方便的编写出友好的命令行接口。程序定义需要的参数，然后 argparse 将从sys.argv解析出那些参数。argparse 模块还可以自动生成帮助手册和使用手册。并在用户给程序传入无效参数时报出错误信息。

1. 一个简单的示例

prog.py 文件：

import argparseparser = argparse.ArgumentParser(description='Process some integers.')
parser.add_argument('integers', metavar='N', type=int, nargs='+',help='an integer for the accumulator')
parser.add_argument('--sum', dest='accumulate', action='store_const',const=sum, default=max,help='sum the integers (default: find the max)')args = parser.parse_args()
print(args.accumulate(args.integers))

脚本执行：

在命令行中执行该文件，返回帮助信息：

zzz@ubuntu:~/my_learning$ python3 prog.py -h
usage: prog.py [-h] [--sum] N [N ...]Process some integers.positional arguments:N           an integer for the accumulatoroptional arguments:-h, --help  show this help message and exit--sum       sum the integers (default: find the max)
zzz@ubuntu:~/my_learning$

使用适当参数运行，脚本会返回输入命令行传入整数的总和或最大值

zzz@ubuntu:~/my_learning$ python3 prog.py 1 3 2 6
6
zzz@ubuntu:~/my_learning$ python3 prog.py 1 3 2 6 --sum
12
zzz@ubuntu:~/my_learning$

脚本简析：

使用 argparse创建一个 ArgumentParse 对象。该对象将包含命令行解析成Python数据类型所需的全部信息。

>>> parser = argparse.ArgumentParser(description='Process some integers.')

给 ArgumentParse 对象添加程序参数信息，通过调用 add_argument()方法完成。指定 ArgumentParser 如何获取命令行字符串并将其转换为对象。

>>> parser.add_argument('integers', metavar='N', type=int, nargs='+',
...                     help='an integer for the accumulator')
>>> parser.add_argument('--sum', dest='accumulate', action='store_const',
...                     const=sum, default=max,
...                     help='sum the integers (default: find the max)')

这些信息在parse_args()调用时被存储和使用。 ArgumentParse 对象通过调用 parse_args() 方法解析参数。该方法将检查命令行，把每个参数转换为适当的类型然后调用相应的操作。

args = parser.parse_args()
print(args.accumulate(args.integers))

parse_args()返回一个具有两个属性（‘integers’ 和 ‘accumulate’）的对象。integers属性将是一个包含一个或多个整数的列表，而 accumulate 属性当命令行中指定了 --sum 参数时将是 sum() 函数，否则则是 max() 函数。

2. `ArgumentParser` 对象

class argparse.ArgumentParser(prog=None,    # 程序名，默认为脚本文件名。usage=None, # 根据它包含的参数来构建用法消息。description=None, # 简约描述程序的功能。epilog=None,parents=[],formatter_class=argparse.HelpFormatter,prefix_chars='-', romfile_prefix_chars=None,argument_default=None,conflict_handler='error', add_help=True, allow_abbrev=True,exit_on_error=True)

3. `ArgumentParser.add_argument()` 方法

ArgumentParser.add_argument(name or flags[, action][, nargs][, const][, default][, type][, choices][, required][, help][, metavar][, dest])

参数结束：

name or flags，选项或位置参数
- 选项创建
- 位置参数创建：以 - 前缀识别，其它识别为位置参数

>>>parser.add_argument('-f', '--foo')
>>>parser.add_argument('bar')

action，指定对应的命令行参数应当如何处理，可选项有：
- store-存储参数的值。默认动作。
```
>>> parser.add_argument('--foo')
```
- store_const-存储被const命名参数指定的值。通常用在选项中来指定一些标志。
```
>>> parser.add_argument('--foo', action='store_true')
>>> parser.add_argument('--bar', action='store_false')
>>> parser.add_argument('--baz', action='store_false')
```
- store_true and store_false-store-const的用于存储True和False的特殊用例，默认值为True和False。
- append-存储一个列表，并且将每个参数追加到列表中。
```
>>> parser.add_argument('--foo', action='append')
>>> parser.parse_args('--foo 1 --foo 2'.split())
>>> Namespace(foo=['1', '2'])
```
- append_const-存储一个列表，并将const命名参数指定的值追加到列表中。
- count-计算一个关键字参数出现的数目或次数。
```
>>> parser.add_argument('--verbose', '-v', action='count', default=0)
```
- help-打印所有当前解析器中的选项和参数都的完整帮助信息。
- version-期望有一个version命令参数在调用中，并打印版本信息。
- extend-这会存储一个列表，并将每个参数值加入到列表中
```
>>> parser.add_argument('--foo', action=argparse.BooleanOptionalAction)
>>> parser.parse_args(['--no-foo'])
Namespace(foo=False)
```

nargs，关联不同数目的命令行参数到单一动作。可选项有：

N（一个整数）。将命令行中的N个参数集到一个列表中。

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo', nargs=2)
>>> parser.add_argument('bar', nargs=1)
>>> parser.parse_args('c --foo a b'.split())
Namespace(bar=['c'], foo=['a', 'b'])

‘?’。如果可能的话，会从命令行中消耗一个参数，并产生一个单一项。如果当前没有命令行参数，则会产生 default 值。注意，对于选项，有另外的用例 - 选项字符串出现但没有跟随命令行参数，则会产生 const 值。

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo', nargs='?', const='c', default='d')
>>> parser.add_argument('bar', nargs='?', default='d')
>>> parser.parse_args(['XX', '--foo', 'YY'])
Namespace(bar='XX', foo='YY')
>>> parser.parse_args(['XX', '--foo'])
Namespace(bar='XX', foo='c')
>>> parser.parse_args([])
Namespace(bar='d', foo='d')

‘*’。将当前命令行参数集到一个列表中。和’+'类似。

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo', nargs='*')
>>> parser.add_argument('--bar', nargs='*')
>>> parser.add_argument('baz', nargs='*')
>>> parser.parse_args('a b --foo x y --bar 1 2'.split())
Namespace(bar=['1', '2'], baz=['a', 'b'], foo=['x', 'y'])

const，用于保存不从命令行中读取但被需求的常数值。常见的两种用途：
- 当add_argument()使用 action='store_const' 或 action='append_const 调用时。这些动作将 const 值添加到 parse_args() 返回的对象的属性中。
- 当 add_argument() 通过选项（例如 -f 或 --foo）调用并且 nargs='?' 时。这会创建一个可以跟随零个或一个命令行参数的选项。当解析命令行时，如果选项后没有参数，则将用 const 代替。

default，当选项或位置参数被忽略时，可以指定在命令行参数未出现时应当使用的值。

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo', default=42)
>>> parser.parse_args(['--foo', '2'])
Namespace(foo='2')
>>> parser.parse_args([])
Namespace(foo=42)

type，命令行参数默认当作字符串读入，type可以使用普通内置类型和函数或者自定义的函数作为类型转化器。

parser = argparse.ArgumentParser()
parser.add_argument('count', type=int)
parser.add_argument('distance', type=float)
parser.add_argument('street', type=ascii)
parser.add_argument('code_point', type=ord)
parser.add_argument('source_file', type=open)
parser.add_argument('dest_file', type=argparse.FileType('w', encoding='latin-1'))
parser.add_argument('datapath', type=pathlib.Path)

choices，某些参数应从一组限定值中选择，此时可以使用choices参数。

>>> parser = argparse.ArgumentParser(prog='game.py')
>>> parser.add_argument('move', choices=['rock', 'paper', 'scissors'])
>>> parser.parse_args(['rock'])
Namespace(move='rock')

required，'-'标记的选项参数一般可以被忽略，如果需要将其标志位必须，则可以使用该关键字为True来说明。

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo', required=True)
>>> parser.parse_args(['--foo', 'BAR'])
Namespace(foo='BAR')

metavar
dest

4. `ArgumentParser.parse_args()` 方法

将参数字符串转换为对象并将其设为命名空间的属性。 add_argument() 添加了对象及其赋值。

ArgumentParser.parse_args(args=None,    # 要解析的字符串列表。默认值从 sys.argv获取。namespace=None # 要获取属性的对象。默认值是一个新的空Namespace对象。)