diff options
-rw-r--r-- | pyside2uic/Compiler/qobjectcreator.py | 16 | ||||
-rw-r--r-- | pyside2uic/objcreator.py | 3 |
2 files changed, 17 insertions, 2 deletions
diff --git a/pyside2uic/Compiler/qobjectcreator.py b/pyside2uic/Compiler/qobjectcreator.py index fd1b685..296ac36 100644 --- a/pyside2uic/Compiler/qobjectcreator.py +++ b/pyside2uic/Compiler/qobjectcreator.py @@ -29,13 +29,24 @@ except NameError: from sets import Set as set from pyside2uic.Compiler.indenter import write_code -from pyside2uic.Compiler.qtproxies import QtWidgets, Literal, strict_getattr +from pyside2uic.Compiler.qtproxies import (QtWidgets, QtGui, Literal, + strict_getattr) logger = logging.getLogger(__name__) DEBUG = logger.debug +class _QtGuiWrapper(object): + def search(clsname): + try: + return strict_getattr(QtGui, clsname) + except AttributeError: + return None + + search = staticmethod(search) + + class _QtWidgetsWrapper(object): def search(clsname): try: @@ -124,6 +135,9 @@ class CompilerCreatorPolicy(object): def __init__(self): self._modules = [] + def createQtGuiWrapper(self): + return _QtGuiWrapper + def createQtWidgetsWrapper(self): return _QtWidgetsWrapper diff --git a/pyside2uic/objcreator.py b/pyside2uic/objcreator.py index deed44f..1721c8f 100644 --- a/pyside2uic/objcreator.py +++ b/pyside2uic/objcreator.py @@ -47,7 +47,8 @@ class QObjectCreator(object): self._cpolicy = creatorPolicy self._cwFilters = [] - self._modules = [self._cpolicy.createQtWidgetsWrapper()] + self._modules = [self._cpolicy.createQtWidgetsWrapper(), + self._cpolicy.createQtGuiWrapper()] # Get the optional plugins. for plugindir in widgetPluginPath: |