1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
import logging
try:
set()
except NameError:
from sets import Set as set
from PyQt4.uic.Compiler.indenter import write_code
from PyQt4.uic.Compiler.qtproxies import 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 _ModuleWrapper(object):
def __init__(self, name, classes):
if "." in name:
idx = name.rfind(".")
self._package = name[:idx]
self._module = name[idx + 1:]
else:
self._package = None
self._module = name
self._classes = set(classes)
self._used = False
def search(self, cls):
if cls in self._classes:
self._used = True
return type(cls, (QtGui.QWidget,), {"module": self._module})
else:
return None
def _writeImportCode(self):
if self._used:
if self._package is None:
write_code("import %s" % self._module)
else:
write_code("from %s import %s" % (self._package, self._module))
class _CustomWidgetLoader(object):
def __init__(self):
self._widgets = {}
self._usedWidgets = set()
def addCustomWidget(self, widgetClass, baseClass, module):
assert widgetClass not in self._widgets
self._widgets[widgetClass] = (baseClass, module)
def _resolveBaseclass(self, baseClass):
try:
for x in range(0, 10):
try: return strict_getattr(QtGui, baseClass)
except AttributeError: pass
baseClass = self._widgets[baseClass][0]
else:
raise ValueError("baseclass resolve took too long, check custom widgets")
except KeyError:
raise ValueError("unknown baseclass %s" % baseClass)
def search(self, cls):
try:
self._usedWidgets.add(cls)
baseClass = self._resolveBaseclass(self._widgets[cls][0])
DEBUG("resolved baseclass of %s: %s" % (cls, baseClass))
return type(cls, (baseClass,),
{"module" : ""})
except KeyError:
return None
def _writeImportCode(self):
imports = {}
for widget in self._usedWidgets:
_, module = self._widgets[widget]
imports.setdefault(module, []).append(widget)
for module, classes in imports.items():
write_code("from %s import %s" % (module, ", ".join(classes)))
class CompilerCreatorPolicy(object):
def __init__(self):
self._modules = []
def createQtGuiWrapper(self):
return _QtGuiWrapper
def createModuleWrapper(self, name, classes):
mw = _ModuleWrapper(name, classes)
self._modules.append(mw)
return mw
def createCustomWidgetLoader(self):
cw = _CustomWidgetLoader()
self._modules.append(cw)
return cw
def instantiate(self, clsObject, objectname, ctor_args, is_attribute=True, no_instantiation=False):
return clsObject(objectname, is_attribute, ctor_args, no_instantiation)
def invoke(self, rname, method, args):
return method(rname, *args)
def getSlot(self, object, slotname):
return Literal("%s.%s" % (object, slotname))
def _writeOutImports(self):
for module in self._modules:
module._writeImportCode()
|
