8

我的问题类似于如何在给定完整路径的情况下导入模块?但是我不是在导入 .py 源文件,而是在导入带有 .pyd 的包。

在运行时,我正在从一些动态生成的 c 代码创建新的包模块。我成功地生成了包 Foo ,其中包含一个__init__.py文件和一个mod.pyd

/a/temp/dir/foo/
    __init__.py
    bar.pyd

我正在使用的示例代码是

import importlib.util
spec = importlib.util.spec_from_file_location("foo.bar", "/path/to/file.py")
foo = importlib.util.module_from_spec(spec)
spec.loader.exec_module(foo)
foo.bar()

如果我尝试使用pydspec_from_file_location('foo.bar', '/a/temp/dir/foo/__init__.py')中的模块bar无法加载。

如果我尝试使用spec_from_file_location('foo.bar', '/a/temp/dir/foo/')spec_from_file_location返回None.

如果我尝试使用spec_from_file_location('foo.bar', '/a/temp/dir/foo/bar.pyd'),则会收到以下错误堆栈:

File "<frozen importlib._bootstrap>", line 577, in module_from_spec
File "<frozen importlib._bootstrap_external>", line 903, in create_module
File "<frozen importlib._bootstrap>", line 222, in _call_with_frames_removed
ImportError: DLL load failed: A dynamic link library (DLL) initialization routine failed.

推荐的 4 行 importlib 解决方案的替代方案是为MetaPathFinder和创建具体类SourceLoader。我的以下MetaPathFinder工作正常,但是,我不确定如何Loader正确实施。get_data返回值需要一些 Python 代码的 Unicode 表示。如何在加载器中返回包结构或 pyd 内容?

class MyLoader(importlib.abc.SourceLoader):
    def __init__(self, fullname, path, pyd):
        self.path = path
        self.fullname = fullname
        self.pyd = pyd

    def get_data(self, path):
        print('*** get_data', path)
        with open(os.path.join(self.path, self.pyd), 'r') as f:
            return f.read()

    def get_filename(self, fullname):
        print('*** get_filename', fullname)
        return os.path.join(self.path, '__init__.py')

    def is_package(self):
        return True

class MyFinder(MetaPathFinder):
    def __init__(self, this_module, workdir, pyd):
        self._this_module = this_module
        self._workdir = workdir
        self._pyd = pyd

    def find_spec(self, fullname, path, target=None):
        print('find_spec', fullname, path, target, self._this_module)

        if fullname != self._this_module:
            return

        filename = os.path.join(self._workdir, self._this_module)

        spec = ModuleSpec(
            name = fullname,
            loader = MyLoader(fullname, filename, self._pyd),
            origin = filename,
            loader_state = 1234,
            is_package = True,
        )
        return spec

    def is_package(self):
        return True
4

0 回答 0