Импорт крюков для PyQt4.QtCore

Я пытаюсь настроить некоторые крючки импорта через sys.meta_path, в несколько схожих подходах к this SO вопрос. Для этого мне нужно определить две функции find_module и load_module, как описано в ссылке выше. Вот моя функция load_module,

import imp

def load_module(name, path):
    fp, pathname, description = imp.find_module(name, path)

    try:
        module = imp.load_module(name, fp, pathname, description)
    finally:
        if fp:
             fp.close()
    return module

который отлично работает для большинства модулей, но не работает для PyQt4.QtCore при использовании Python 2.7:

name = "QtCore"
path = ['/usr/lib64/python2.7/site-packages/PyQt4']

mod = load_module(name, path)

который возвращает,

Traceback (most recent call last):
   File "test.py", line 19, in <module>
   mod = load_module(name, path)
   File "test.py", line 13, in load_module
   module = imp.load_module(name, fp, pathname, description)
SystemError: dynamic module not initialized properly

Тот же код отлично работает с Python 3.4 (хотя imp становится устаревшим и importlib в идеале следует использовать вместо него).

Я полагаю, что это имеет какое-то отношение к инициализации динамического модуля SIP. Есть ли что-нибудь еще, что я должен попробовать с Python 2.7?

Примечание: это относится как к PyQt4, так и к PyQt5.

Изменить: это может быть связано с этим вопросом, как действительно,

cd /usr/lib64/python2.7/site-packages/PyQt4
python2 -c 'import QtCore'

не работает с той же ошибкой. Тем не менее я не уверен, что бы это было вокруг...

Edit2: после запроса @Nikitastrong > для конкретного примера использования, я пытаюсь перенаправить импорт, поэтому, когда вы делаете import A, что происходит, это import B. Можно было бы подумать, что для этого достаточно переименовать модуль в find_spec/find_module, а затем использовать по умолчанию load_module. Однако неясно, где найти реализацию по умолчанию load_module в Python 2. Ближайшая реализация, которую я нашел, похожа на future.standard_library.RenameImport. Не похоже, что имеется полный путь реализации importlib от Python 3 до 2.

В этом gist можно найти минимальный рабочий пример для крючков импорта, воспроизводящих эту проблему.

Ответ 1

UPD: эта часть не очень актуальна после ответов на ответы, поэтому см. UPD ниже.

Почему бы просто не использовать importlib.import_module, который доступен как в Python 2.7, так и в Python 3:

#test.py

import importlib

mod = importlib.import_module('PyQt4.QtCore')
print(mod.__file__)

на Ubuntu 14.04:

$ python2 test.py 
/usr/lib/python2.7/dist-packages/PyQt4/QtCore.so

Так как это динамический модуль, как сказано в ошибке (и фактический файл QtCore.so), можно также взглянуть на imp.load_dynamic.

Другим решением может быть принудительное выполнение кода инициализации модуля, но IMO это слишком много хлопот, так почему бы просто не использовать importlib.

UPD. В pkgutil есть вещи, которые могут помочь. О чем я говорил в своем комментарии, попробуйте изменить свой поиск следующим образом:

import pkgutil

class RenameImportFinder(object):

    def find_module(self, fullname, path=None):
        """ This is the finder function that renames all imports like
             PyQt4.module or PySide.module into PyQt4.module """
        for backend_name in valid_backends:
            if fullname.startswith(backend_name):
                # just rename the import (That what i thought about)
                name_new = fullname.replace(backend_name, redirect_to_backend)
                print('Renaming import:', fullname, '->', name_new, )
                print('   Path:', path)


                # (And here, don't create a custom loader, get one from the
                # system, either by using 'pkgutil.get_loader' as suggested
                # in PEP302, or instantiate 'pkgutil.ImpLoader').

                return pkgutil.get_loader(name_new) 

                #(Original return statement, probably 'pkgutil.ImpLoader'
                #instantiation should be inside 'RenameImportLoader' after
                #'find_module()' call.)
                #return RenameImportLoader(name_orig=fullname, path=path,
                #       name_new=name_new)

    return None

Не удается проверить код выше, поэтому, пожалуйста, попробуйте сами.

P.S. Обратите внимание: imp.load_module(), который работал на вас в Python 3, устарел с Python 3.3.

Другое решение - не использовать крючки вообще, а вместо этого обернуть __import__:

print(__import__)

valid_backends = ['shelve']
redirect_to_backend = 'pickle'

# Using closure with parameters 
def import_wrapper(valid_backends, redirect_to_backend):
    def wrapper(import_orig):
        def import_mod(*args, **kwargs):
            fullname = args[0]
            for backend_name in valid_backends:
                if fullname.startswith(backend_name):
                    fullname = fullname.replace(backend_name, redirect_to_backend)
                    args = (fullname,) + args[1:]
            return import_orig(*args, **kwargs)
        return import_mod
    return wrapper

# Here it important to assign to __import__ in __builtin__ and not
# local __import__, or it won't affect the import statement.
import __builtin__
__builtin__.__import__ = import_wrapper(valid_backends, 
                                        redirect_to_backend)(__builtin__.__import__)

print(__import__)

import shutil
import shelve
import re
import glob

print shutil.__file__
print shelve.__file__
print re.__file__
print glob.__file__

выход:

<built-in function __import__>
<function import_mod at 0x02BBCAF0>
C:\Python27\lib\shutil.pyc
C:\Python27\lib\pickle.pyc
C:\Python27\lib\re.pyc
C:\Python27\lib\glob.pyc

shelve переименован в pickle, а pickle импортируется по умолчанию машиной с именем переменной shelve.

Ответ 2

При поиске модуля, который является частью пакета, такого как PyQt4.QtCore, вы должны рекурсивно находить каждую часть имени без .. И imp.load_module требует, чтобы параметр name был полным именем модуля с ., разделяющим пакет и имя модуля.

Потому что QtCore является частью пакета, вместо этого вы делаете python -c 'import PyQt4.QtCore'. Здесь код для загрузки модуля.

import imp

def load_module(name):
    def _load_module(name, pkg=None, path=None):
        rest = None
        if '.' in name:
            name, rest = name.split('.', 1)
        find = imp.find_module(name, path)
        if pkg is not None:
            name = '{}.{}'.format(pkg, name)
        try:
            mod = imp.load_module(name, *find)
        finally:
            if find[0]:
                find[0].close()
        if rest is None:
            return mod 
        return _load_module(rest, name, mod.__path__)
    return _load_module(name)

Тест;

print(load_module('PyQt4.QtCore').qVersion())  
4.8.6