-
Notifications
You must be signed in to change notification settings - Fork 0
/
DevicePalette.py
90 lines (71 loc) · 2.85 KB
/
DevicePalette.py
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
import os
import types
import logging
from PySide.QtCore import *
from PySide.QtGui import *
import devices
import generated.guiPalette
lg = logging.getLogger('autoaudit')
class DevicePalette(QWidget):
"""
a palette which shows all supported devices and allows
them to be selected. Usually this will be shown inside
a QDockWidget. Its primary use is on the MainWin to
show what units the tool knows about and to pre-filter
the search.
"""
def __init__(self, parent=None):
super(DevicePalette, self).__init__()
self.parent = parent
self.drivers = {}
self.ui = generated.guiPalette.Ui_wPalette()
self.ui.setupUi(self)
self.ui.pbSelAll.clicked.connect(self.doSelectAll)
self.ui.pbSelNone.clicked.connect(self.doSelectNone)
self.deviceMap = {}
# DEBUG
# self.loadCruft()
self.loadDrivers()
def loadCruft(self):
"test func to provide some sample data"
for item in ['toaster', 'parasol', 'racecar', 'marigold']:
qlwi = QListWidgetItem(item)
qlwi.setFlags(Qt.ItemIsUserCheckable|Qt.ItemIsEnabled|Qt.ItemIsSelectable)
qlwi.setCheckState(Qt.Checked)
self.ui.lwDrivers.addItem(qlwi)
def loadDrivers(self):
"load all drivers in the devices module"
self.deviceMap = dict([(mod.description, mod)
for mod
in devices.__dict__.values()
if isinstance(mod, types.ModuleType)
and hasattr(mod, 'Device')])
for key in sorted(self.deviceMap):
qlwi = QListWidgetItem(key)
qlwi.setFlags(Qt.ItemIsUserCheckable|Qt.ItemIsEnabled|Qt.ItemIsSelectable)
qlwi.setCheckState(Qt.Checked)
self.ui.lwDrivers.addItem(qlwi)
# TODO resize widget to fit
# self.ui.lwDrivers.setMinimumWidth(self.ui.lwDrivers.sizeHintForColumn(0))
# self.ui.lwDrivers.resize(self.ui.lwDrivers.contentsHeight())
def getSelectedDevices(self):
"return list of all checked drivers"
rl = []
for i in range(0, self.ui.lwDrivers.count()):
item = self.ui.lwDrivers.item(i)
if item.checkState():
dev = self.deviceMap.get(item.text(), None)
if dev is not None:
rl.append(dev)
return rl
def doSelectAll(self):
"modify selection to check all"
for i in range(self.ui.lwDrivers.count()):
self.ui.lwDrivers.item(i).setCheckState(Qt.Checked)
def doSelectNone(self):
"modify selection to check none"
for i in range(self.ui.lwDrivers.count()):
self.ui.lwDrivers.item(i).setCheckState(Qt.Unchecked)
if __name__ == '__main__':
dp = DevicePalette()
print dp.loadDrivers()