mirror of
https://github.com/wassname/sloth.git
synced 2026-07-05 22:37:19 +08:00
Tobias Gehrig
120 lines
3.9 KiB
Python
120 lines
3.9 KiB
Python
import sys, os
|
|
from PyQt4.QtGui import *
|
|
from PyQt4.QtCore import *
|
|
|
|
class ButtonArea(QWidget):
|
|
def __init__(self, parent=None):
|
|
QWidget.__init__(self, parent)
|
|
|
|
self.label_names = []
|
|
self.label_properties = {}
|
|
|
|
self.properties = {}
|
|
|
|
self.last_checked_label = None
|
|
|
|
def create_button(self, button_name):
|
|
button = QPushButton(button_name)
|
|
button.setFlat(True)
|
|
button.setCheckable(True)
|
|
button.clicked.connect(self.clickedButton)
|
|
return button
|
|
|
|
def update_label_buttons(self):
|
|
label_vlayout = self.create_choice_list("Labels")
|
|
self.label_buttongroup = QButtonGroup()
|
|
for label_name in self.label_names:
|
|
button = QPushButton(label_name)
|
|
button.setFlat(True)
|
|
button.setCheckable(True)
|
|
button.clicked.connect(self.clickedLabelButton)
|
|
label_vlayout.addWidget(button)
|
|
self.label_buttongroup.addButton(button)
|
|
if label_name == self.last_checked_label:
|
|
button.setChecked(True)
|
|
|
|
self.hlayout = QHBoxLayout()
|
|
self.hlayout.addLayout(label_vlayout)
|
|
self.setLayout(self.hlayout)
|
|
|
|
def update_buttons(self):
|
|
layout = self.hlayout.takeAt(1)
|
|
|
|
dynamic_hlayout = QHBoxLayout()
|
|
|
|
if self.last_checked_label != None:
|
|
for key in self.label_properties[self.last_checked_label].keys():
|
|
if key in ["type", "class"]:
|
|
continue
|
|
vlayout = self.create_choice_list(key)
|
|
buttongroup = QButtonGroup()
|
|
for value in self.properties[key]:
|
|
button = self.create_button(value)
|
|
vlayout.addWidget(button)
|
|
buttongroup.addButton(button)
|
|
dynamic_hlayout.addLayout(vlayout)
|
|
|
|
self.hlayout.addLayout(dynamic_hlayout)
|
|
self.update()
|
|
self.updateGeometry()
|
|
|
|
def add_label(self, label_name, properties = {}):
|
|
self.label_names.append(label_name)
|
|
self.label_properties[label_name] = properties
|
|
for key, value in properties.iteritems():
|
|
if self.properties.has_key(key):
|
|
self.properties[key] |= set(value)
|
|
else:
|
|
self.properties[key] = set(value)
|
|
|
|
def get_checked_label_button(self):
|
|
return self.label_buttongroup.checkedButton()
|
|
|
|
def create_choice_list(self, list_name, elements = []):
|
|
vlayout = QVBoxLayout()
|
|
vlayout.setSpacing(0)
|
|
#vlayout.setMargin(0)
|
|
#vlayout.setContentsMargins(0, 0, 0, 0)
|
|
vlayout.addWidget(QLabel("<center><b>" + list_name + "</b></center>"))
|
|
if len(elements) > 0:
|
|
buttongroup = QButtonGroup()
|
|
for element in elements:
|
|
self.create_button(element)
|
|
return vlayout
|
|
|
|
def clickedButton(self):
|
|
button = self.get_checked_label_button()
|
|
label_name = str(button.text())
|
|
print label_name
|
|
|
|
def clickedLabelButton(self):
|
|
button = self.get_checked_label_button()
|
|
label_name = str(button.text())
|
|
if label_name != self.last_checked_label:
|
|
print label_name, self.last_checked_label
|
|
self.last_checked_label = label_name
|
|
self.update_buttons()
|
|
|
|
def load(self, config_filepath):
|
|
execfile(config_filepath)
|
|
#if self.get_checked_label_button() == None:
|
|
# if len(self.label_buttongroup.buttons()) != 0:
|
|
# self.label_buttongroup.buttons()[0].setChecked(True)
|
|
#
|
|
#self.last_checked_label = str(self.get_checked_label_button().text())
|
|
self.update_label_buttons()
|
|
self.update_buttons()
|
|
|
|
def main():
|
|
app = QApplication(sys.argv)
|
|
|
|
ba = ButtonArea()
|
|
ba.load("example_config.py")
|
|
ba.show()
|
|
|
|
return app.exec_()
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|
|
|