python - imp.find_module() 支持压缩鸡蛋

Question

imp.find_module()没有从压缩的鸡蛋中找到模块。

如何找到可以来自两个地方的模块:目录或压缩的鸡蛋？在我的例子中，重要的是我可以提供一个 path 参数，比如 imp.find_module() 支持它。

背景

不知何故，包在我们的环境中安装了两次。作为压缩鸡蛋和普通文件。我想写一个检查，告诉我一个模块是否安装了两次。参见 https://stackoverflow.com/a/23990989/633961

Answer 1

假设使用 Python 2，我认为您需要的信息在 PEP 302 - New Import Hooks 中(PEP 对于 Python 3 已经过时，在这方面完全不同)。

从 ZIP 文件中查找和导入模块在 zipimport 中实现，如 PEP 所述，它被“ Hook ”到进口机器中。当 PEP 302 和从 ZIP 导入被添加到 Python 时，imp 模块没有被改编，即 imp 完全不知道 PEP 302 钩子(Hook)。

一个“通用”find_module 函数可以找到像 imp 这样的模块并且尊重 PEP 302 钩子(Hook)，大致如下所示:

import imp
import sys

def find_module(fullname, path=None):
    try:
        # 1. Try imp.find_module(), which searches sys.path, but does
        # not respect PEP 302 import hooks.
        result = imp.find_module(fullname, path)
        if result:
            return result
    except ImportError:
        pass
    if path is None:
        path = sys.path
    for item in path:
        # 2. Scan path for import hooks. sys.path_importer_cache maps
        # path items to optional "importer" objects, that implement
        # find_module() etc.  Note that path must be a subset of
        # sys.path for this to work.
        importer = sys.path_importer_cache.get(item)
        if importer:
            try:
                result = importer.find_module(fullname, [item])
                if result:
                    return result
            except ImportError:
                pass
    raise ImportError("%s not found" % fullname)

if __name__ == "__main__":
    # Provide a simple CLI for `find_module` above.
    import argparse
    parser = argparse.ArgumentParser()
    parser.add_argument("-p", "--path", action="append")
    parser.add_argument("modname", nargs='+')
    args = parser.parse_args()
    for name in args.modname:
        print find_module(name, args.path)

但请注意，在 ZIP 存档中查找模块的结果看起来与 imp.find_module 返回的结果完全不同:您将得到一个 zipimport.zipimporter特定 ZIP 的对象。上面的小程序在被要求查找常规模块、内置模块和压缩鸡蛋中的模块时打印以下内容:

$ python find_module.py grin os sys
<zipimporter object "<my venv>/lib/python2.7/site-packages/grin-1.2.1-py2.7.egg">
(<open file '<my venv>/lib/python2.7/os.py', mode 'U' at 0x10a0bbf60>, '<my venv>/lib/python2.7/os.py', ('.py', 'U', 1))
(None, 'sys', ('', '', 6))