imp.find_module() 支持压缩鸡蛋

Question

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

如何找到可以来自两个地方的模块：目录或压缩鸡蛋？在我的情况下，重要的是我可以提供像 imp.find_module() 这样的 path 参数支持它。

背景

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

Answer 1

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

从 ZIP 档案中查找和导入模块在 zipimport 中实现，它被“挂钩”到 PEP 所描述的导入机制中。当 PEP 302 和从 ZIP 导入被添加到 Python 时，imp 模块没有被适配，即imp 完全不知道 PEP 302 钩子。

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

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 返回的结果完全不同：您将获得特定 ZIP 的 zipimport.zipimporter 对象。当要求查找常规模块、内置模块和压缩鸡蛋中的模块时，上面的小程序会打印以下内容：

$ 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))