如何更改导入库的名称?

时间:2012-09-10 13:21:03

标签: python import jython

我正在使用jython和第三方应用程序。第三方应用程序有一些内置库foo。要进行一些(单元)测试,我们希望在应用程序之外运行一些代码。由于foo绑定到应用程序,我们决定编写自己的模拟实现。

然而,有一个问题,我们在python中实现了我们的mock类,而他们的类在java中。因此,要使用他们的代码,我会做import foo,然后foo就是模拟。但是,如果我们像这样导入python模块,我们会将模块附加到名称上,因此必须编写foo.foo才能进入该类。

为方便起见,我们希望能够编写from ourlib.thirdparty import foo以将foo绑定到foo - 。但是我们希望避免直接导入ourlib.thirdparty中的所有类,因为每个文件的加载时间需要很长时间。

在python中有什么办法吗? (我没有使用导入钩子,我尝试简单地从load_module返回类或者覆盖我写给sys.modules的内容(我认为这两种方法都很难看,尤其是后面的方法))

修改

好的:这是mylib.thirdparty中的文件看起来像简化(没有魔法):

foo.py:

try:
    import foo
except ImportError:
    class foo
        ....

实际上他们看起来像这样:

foo.py:

class foo
    ....

__init__.py in ourlib.thirdparty
import sys
import os.path
import imp
#TODO: 3.0 importlib.util abstract base classes could greatly simplify this code or make it prettier.

class Importer(object):
    def __init__(self, path_entry):
        if not path_entry.startswith(os.path.join(os.path.dirname(__file__), 'thirdparty')):
            raise ImportError('Custom importer only for thirdparty objects')

        self._importTuples = {}

    def find_module(self, fullname):
        module = fullname.rpartition('.')[2]

        try:
            if fullname not in self._importTuples:
                fileObj, self._importTuples[fullname] = imp.find_module(module)

                if isinstance(fileObj, file):
                    fileObj.close()
        except:
            print 'backup'
            path = os.path.join(os.path.join(os.path.dirname(__file__), 'thirdparty'), module+'.py')
            if not os.path.isfile(path):
                return None
                raise ImportError("Could not find dummy class for %s (%s)\n(searched:%s)" % (module, fullname, path))

            self._importTuples[fullname] = path, ('.py', 'r', imp.PY_SOURCE)

        return self

    def load_module(self, fullname):
        fp = None
        python = False

        print fullname

        if self._importTuples[fullname][1][2] in (imp.PY_SOURCE, imp.PY_COMPILED, imp.PY_FROZEN):
            fp = open( self._importTuples[fullname][0], self._importTuples[fullname][1][1])
            python = True

        try:
            imp.load_module(fullname, fp, *self._importTuples[fullname])
        finally:
            if python:
                module = fullname.rpartition('.')[2]
                #setattr(sys.modules[fullname], module, getattr(sys.modules[fullname], module))
                #sys.modules[fullname] = getattr(sys.modules[fullname], module)

                if isinstance(fp, file):
                    fp.close()

                return getattr(sys.modules[fullname], module)



sys.path_hooks.append(Importer)

1 个答案:

答案 0 :(得分:0)

正如其他人所说的那样,在Python中,import语句iself的语法非常明显:

例如

from foo import foo as original_foo - 甚至import foo as module_foo

有趣的是,import statemente将名称绑定到本地上下文中导入的模块或对象 - 但是,字典sys.modules(在模块sys of course), is a live reference to all imported modules, using their names as a key. This mechanism plays a key role in avoding that Python re-reads and re-executes and already imported module , when running (that is, if various of yoru modules or sub-modules import the same foo`模块上,它只读一次 - 后续导入使用存储在sys.modules中的引用。

而且 - 除了“import ... as”语法之外,Python中的模块只是另一个对象:您可以在运行时为它们分配任何其他名称。

因此,以下代码也适用于您:

import foo
original_foo = foo
class foo(Mock):
   ...