Question

根据文档，六个支持adding custom renames到six.moves：

six.add_move(item)

将项目添加到six.moves映射。 item应为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'

内置动作工作正常。我如何让它工作？

Answer 1

您无法从移动中导入名称。使用：

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)

如何在六个中添加自定义重命名？

1 个答案: