在由多个自定义模块支持的多个脚本中使用python的argparse

Question

我正在构建一组用于管理基础架构的脚本和模块.为了保持井井有条,我希望尽可能地加强努力,并尽量减少更新脚本的锅炉.

特别是这里的问题是合并ArgumentParser模块.

一个示例结构是使脚本和库组织如下:

|-- bin
    |-- script1
    |-- script2
|-- lib
    |-- logger
    |-- lib1
    |-- lib2

在这种情况下,script1可能只会使用logger和lib1,而script2使用logger和lib2.在这两种情况下,我希望记录器接受'-v'和'-d',同时script1也可以接受额外的args和lib2其他args.我知道这会导致冲突,并会手动管理.

SCRIPT1

#!/usr/bin/env python
import logger
import lib1
argp = argparse.ArgumentParser("logger", parent=[logger.argp])

SCRIPT2

#!/usr/bin/env python
import logger
import lib2

记录仪

#!/usr/bin/env python
import argparse
argp = argparse.ArgumentParser("logger")
argp.add_argument('-v', '--verbose', action="store_true", help="Verbose output")
argp.add_argument('-d', '--debug', action="store_true", help="Debug output. Assumes verbose output.")

每个脚本和lib都可能拥有它自己的参数,但是这些都必须合并到一个最终的arg_parse()中

到目前为止,我的尝试导致无法继承/扩展argp设置.如何在库文件和脚本之间完成？

Answer 1

最简单的方法是在每个模块中使用一个接受ArgumentParser实例的方法,并为其添加自己的参数.

# logger.py
def add_arguments(parser):
    parser.add_argument('-v', '--verbose', action="store_true", help="Verbose output")
    # ...

# lib1.py
def add_arguments(parser):
    parser.add_argument('-x', '--xtreme', action="store_true", help="Extremify")
    # ...

每个脚本都会创建它们ArgumentParser,将它传递给每个提供参数的模块,然后添加自己的特定参数.

# script1.py
import argparse, logger, lib1
parser = argparse.ArgumentParser("script1")
logger.add_arguments(parser)
lib1.add_arguments(parser)
# add own args...