Toolbox/PyQt4ImportHook.py

# -*- coding: utf-8 -*-

# Copyright (c) 2014 - 2015 Detlev Offenbach <detlev@die-offenbachs.de>
#

"""
Module implementing an import hook converting PyQt5 imports to PyQt4 imports.
"""

from __future__ import unicode_literals

import sys
try:
    if "--pyqt4" in sys.argv:
        sys.argv.remove("--pyqt4")
        # fake a failed PyQt5 import
        raise ImportError
    import PyQt5    # __IGNORE_WARNING__
except ImportError:
    import importlib

    class PyQt4Importer(object):
        """
        Class implementing an importer converting PyQt5 imports to PyQt4
        imports.
        """
        def __init__(self):
            """
            Constructor
            """
            self.__path = None
        
        def find_module(self, fullname, path=None):
            """
            Public method returning the module loader.
            
            @param fullname name of the module to be loaded (string)
            @param path path to resolve the module name (string)
            @return module loader object
            """
            if fullname.startswith("PyQt5"):
                self.__path = path
                return self
            
            return None
        
        def load_module(self, fullname):
            """
            Public method to load a module.
            
            @param fullname name of the module to be loaded (string)
            @return reference to the loaded module (module)
            """
            if fullname in ["PyQt5.QtWidgets", "PyQt5.QtPrintSupport"]:
                newname = "PyQt4.QtGui"
            elif fullname in ["PyQt5.QtWebKitWidgets"]:
                newname = "PyQt4.QtWebKit"
            else:
                newname = fullname.replace("PyQt5", "PyQt4")
            
            module = importlib.import_module(newname)
            sys.modules[fullname] = module
            if fullname == "PyQt5.QtCore":
                import PyQt4.QtGui
                module.qInstallMessageHandler = module.qInstallMsgHandler
                module.QItemSelectionModel = PyQt4.QtGui.QItemSelectionModel
                module.QItemSelection = PyQt4.QtGui.QItemSelection
                module.QSortFilterProxyModel = \
                    PyQt4.QtGui.QSortFilterProxyModel
                module.QAbstractProxyModel = PyQt4.QtGui.QAbstractProxyModel
                module.QStringListModel = PyQt4.QtGui.QStringListModel
            return module

    sys.meta_path.insert(0, PyQt4Importer())
    
    if sys.version_info[0] == 2:
        import sip
        sip.setapi('QString', 2)
        sip.setapi('QVariant', 2)
        sip.setapi('QTextStream', 2)