根据文档,six 支持向six.moves
添加自定义重命名:
six.add_move(item)
将项添加到
six.moves
映射中。 项应为MovedAttribute
或MovedModule
实例。
和:
class six.MovedModule(name, old_mod, new_mod)
为引用不同名称
six.moves
创建映射 Python 2 和 3 中的模块。old_mod
是 Python 2 模块的名称。new_mod
是 Python 3 模块的名称。
但是,这段代码为我带来了一个ImportError
:
from six import add_move, MovedModule
add_move(MovedModule('mock', 'mock', 'unittest.mock'))
from six.moves.mock import MagicMock
当我使用六个 1.9.0 在 Python 3.4.2 上运行它时,我收到此错误:
Traceback (most recent call last):
File "test_six_moves.py", line 2, in <module>
from six.moves.mock import MagicMock
ImportError: No module named 'six.moves.mock'
内置动作工作正常。我如何让它工作?
您无法从移动中导入名称。用:
from __future__ import print_function
from six import add_move, MovedModule
add_move(MovedModule('mock', 'mock', 'unittest.mock'))
from six.moves import mock
print(mock.MagicMock)
这将为您提供:
# Python 2
<class 'mock.MagicMock'>
# Python 3
<class 'unittest.mock.MagicMock'>
请注意,从移动中导入适用于附带six
的移动。例如:from six.moves.configparser import ConfigParser
有效。
这段代码(来自six.py
)是为什么:
for attr in _moved_attributes:
setattr(_MovedItems, attr.name, attr)
if isinstance(attr, MovedModule):
_importer._add_module(attr, "moves." + attr.name)
事实上,如果您运行以下操作(当然不建议干预私有属性),则导入将起作用:
import six
mod = six.MovedModule('mock', 'mock', 'unittest.mock')
six.add_move(mod)
six._importer._add_module(mod, "moves." + mod.name)