Как имитировать, чтобы `from x import *`работало

Я пытаюсь создать макет matplotlib, чтобы скомпилировать свои документы с помощью ReadTheDocs, но столкнулся с проблемой.

В моем коде я импортирую matplotlib, используя from matplotlib.pyplot import *.

Я использую следующий код для своих макетов (, как это предложено в FAQ ReadTheDocs):

class Mock(object):
    def __init__(self, *args, **kwargs):
        pass

    def __call__(self, *args, **kwargs):
        return Mock()

    @classmethod
    def __getattr__(cls, name):
        if name in ('__file__', '__path__'):
            return '/dev/null'
        elif name[0] == name[0].upper():
            return type(name, (), {})
        else:
            return Mock()

MOCK_MODULES = ['numpy', 'scipy', 'matplotlib', 'matplotlib.pyplot']
for mod_name in MOCK_MODULES:
    sys.modules[mod_name] = Mock()

Однако при запуске from matplotlib.pyplot import *я получаю сообщение об ошибке TypeError: 'type' object does not support indexing.

Есть ли способ изменить свой Mock, чтобы он позволял мне импортировать matplotlib, используя стиль from x import *? Мне не нужно, чтобы какие-то определенные функции были доступны, мне просто нужно, чтобы их можно было импортировать, чтобы ReadTheDocs мог правильно импортировать код.

11
задан robintw 9 August 2012 в 06:02
поделиться