让每个子模块定义自己的子解析器

格林杰罗

我必须运行依赖于相对较大项目其余部分的主文件。项目结构可以这样看

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] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章