python动态导入模块——importlib

当在写代码时，我们希望能够根据传入的选项设置，如args.model来确定要导入使用的是哪个model.py文件，而不是一股脑地导入

这种时候就需要用上python的动态导入模块

比如此时文件结构为：

├── models
│   ├── __init__.py
│   ├── cycle_gan_model.py
│   └── pix2pix_model.py
└── train.py

假设此时参数的设置中有：

import argparse

parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--model', type=str, default='cycle_gan', help='chooses which model to use. [cycle_gan | pix2pix | test | colorization]')
args = parser.parse_args()

此时如果想要根据传入的参数args.model的值决定导入哪个模块，则：

import importlib
model_filename = "models." + args.model + "_model"
# 根据指定的args.model来导入本地models文件夹中的相应args.model + '_model.py'模块
# 假设如果args.model = cycle_gan，则导入模块models.cycle_gan_model
modellib = importlib.import_module(model_filename)

然后就能够使用modellib来调用cycle_gan_model模块

等价于：

frommodels import cycle_gan_model as modellib


总结来说：倒入包的‘路径’是拼接出来的。类似于路径拼接一个道理。

在wapiti中的应用代码实例：

for mod_name in attack.modules:
    mod = import_module("wapitiCore.attack." + mod_name)
    mod_instance = getattr(mod, mod_name)(self.crawler, self.persister, logger, self.attack_options)   # 出发一个个攻击模块，建立一个个对象
    if hasattr(mod_instance, "set_timeout"):
        mod_instance.set_timeout(self.crawler.timeout)
    self.attacks.append(mod_instance)

    self.attacks.sort(key=attrgetter("PRIORITY"))