argparse: 命令行选项解析

Posted on | Edited on | In | Comments:
示例文件
示例文件

准备：sys.argv 和命令行参数

1
2
3
4
5
6
7
8
9
10
# 输出命令行参数

import sys

def main(argv):
    for i in range(len(sys.argv)):
        print('sys.argv[{0}] = {1}'.format(i, sys.argv[i]))

if __name__ == "__main__":
    main(sys.argv)

命令行参数解析

Python 使用 argparse 模块实现命令行选项和参数的解析。

  1. 创建 ArgumentParser 对象
  2. 添加参数add_argument()
  3. 解析参数parse_args()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
import argparse


def main():
    # 创建参数解析器 ArgumentParser 对象
    parser = argparse.ArgumentParser()

    """
    添加参数
    - name 或 flag，指定该参数为名称参数还是选项参数
        - `foo`
        - `-f` 或 `--foo`
    - dest 
    - metavar 在使用方法 usage 中，该参数的名称
"""
    parser.add_argument('-o', dest='outfile', help='output file')
    parser.add_argument(dest='filenames', metavar='filename', nargs='*')

    # 解析参数
    args = parser.parse_args()

    print(args.filenames)
    print(args.outfile)


if __name__ == "__main__":
    main()

最终效果如上所示。argparse模块太复杂了，我并不建议使用。与之相比，Google 的 fire 更加简单灵活，我写了一个 简单的教程。