我必须运行依赖于相对较大项目其余部分的主文件。项目结构可以这样看
main.py
opts.py
--models \
--model1.py
--model2.py
...
--schedulers
--scheduler1.py
--scheduler2.py
...
...
问题是当我必须将参数传递给每个组件时(使用argparse
)。一种简单的方法是在一个位置为每个组件定义所有参数。到目前为止,这对我有用(在opts.py
),但我想做一些更优雅的事情。在我的每个组件的解析函数中,parse_models
或者parse_scheduler
我想遍历models
和 的每个子模块,并schedulers
让它们通过调用一个函数来定义自己的参数define_arguments
,每个函数都在其中创建自己的子解析器。
总而言之,我如何遍历子模块并define_arguments
从内部调用它们的函数opts.py
?
您可以使用该glob
模块遍历所有 python 文件。您可以使用父模块的__path__
属性找到正确的路径。使用importlib.import_module
. 然后导入的模块包含一个define_arguments
函数,您可以通过该函数为每个子模块传递一个解析器来定义参数:
from glob import glob
import os, importlib
def load_submodule_parsers(parent_module, parser, help=None):
if help is None:
help = parent_module.__name__ + " modules"
modules = glob(os.path.join(parent_module.__path__, "*.py"))
subparsers = parser.add_subparsers(help=help)
for module_file in modules:
module_name = os.path.basename(module_file)[:-3]
if module == "__init__":
continue
module = importlib.import_module(module_name, package=parent_module.__name__)
if "define_arguments" not in module.__dict__:
raise ImportError(parent_module.__name__ + " submodule '" + module_name + "' does not have required 'define_arguments' function.")
parser = subparsers.add_parser(module_name)
module.define_arguments(parser)
将函数传递给父模块对象:
import argparse, models, schedulers
parser = argparse.ArgumentParser()
subparsers = parser.add_subparsers()
models_parser = subparsers.add_parser("models")
load_submodule_parsers(models, models_parser)
schedulers_parser = subparsers.add_parser("schedulers")
load_submodule_parsers(schedulers, schedulers_parser)
未经测试的代码,但我认为你可以从这里改进它
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句