imp.find_module()支持拉链蛋

时间:2015-03-10 11:19:38

标签: python python-import

imp.find_module()找不到拉链蛋的模块。

如何找到模块可以来自这两个地方:目录或压缩蛋?在我的情况下,重要的是我可以像imp.find_module()一样提供path参数。

背景

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

1 个答案:

答案 0 :(得分:5)

假设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对象。当被要求从压缩蛋中找到常规模块,内置模块和模块时,上面的litte程序会打印以下内容:

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