diff --git a/quamash/__init__.py b/quamash/__init__.py index fd61525..8843081 100644 --- a/quamash/__init__.py +++ b/quamash/__init__.py @@ -25,7 +25,7 @@ QtModule = importlib.import_module(QtModuleName) if not QtModule: - for QtModuleName in ('PyQt5', 'PyQt4', 'PySide'): + for QtModuleName in ('PyQt5', 'PySide2', 'PyQt4', 'PySide'): try: QtModule = importlib.import_module(QtModuleName) except ImportError: @@ -37,10 +37,10 @@ logger.info('Using Qt Implementation: {}'.format(QtModuleName)) -QtCore = importlib.import_module(QtModuleName + '.QtCore', package=QtModuleName) -QtGui = importlib.import_module(QtModuleName + '.QtGui', package=QtModuleName) -if QtModuleName == 'PyQt5': - from PyQt5 import QtWidgets +QtCore = __import__(QtModuleName + '.QtCore', fromlist=(QtModuleName,)) +QtGui = __import__(QtModuleName + '.QtGui', fromlist=(QtModuleName,)) +if QtModuleName in ('PyQt5', 'PySide2'): + QtWidgets = __import__(QtModuleName + '.QtWidgets', fromlist=(QtModuleName,)) QApplication = QtWidgets.QApplication else: QApplication = QtGui.QApplication