7

我正在构建一组脚本和模块来管理我们的基础设施。为了使事情井井有条,我想尽可能多地整合工作,并尽量减少新脚本的样板。

特别是这里的问题是整合 ArgumentParser 模块。

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

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

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

脚本1

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

脚本2

#!/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.")

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

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

4

1 回答 1

11

最简单的方法是在每个模块中都有一个方法,该方法接受一个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...
于 2013-03-19T08:43:17.633 回答