我正在实现一个导入钩子,它使用 pip 自动安装缺少的模块。
到目前为止,它适用于简单的模块(只有一个级别),例如unipath
. 但是,如果我尝试将它与多级导入一起使用,例如zope.interface
导入器仅获取第一部分 ( zope
)。这会导致它失败,因为zope
PyPI 中不存在。
知道如何在导入时获得完整模块的名称吗?
这里有一些代码:
class PipLoader(object):
def __init__(self):
self.module = None
def find_module(self, name, path):
print "Will install module {}".format(name)
self.module = None
sys.meta_path.remove(self)
try:
pip_install(name)
self.module = importlib.import_module(name)
finally:
sys.meta_path.append(self)
return self
def load_module(self, name):
if not self.module:
raise ImportError("Unable to load module")
module = self.module
sys.modules[name] = module
return module
def install():
sys.meta_path.append(PipLoader())