我正在使用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)
答案 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):
...