sloth/buttonarea.py

import sys, os
from PyQt4.QtGui import *
from PyQt4.QtCore import *

class ButtonListWidget(QWidget):
    def __init__(self, name, parent=None):
        QWidget.__init__(self, parent)
        vlayout = QVBoxLayout()
        vlayout.setSpacing(0)
        vlayout.setAlignment(Qt.AlignTop)
        vlayout.addWidget(QLabel("<center><b>" + name + "</b></center>"))
        self.button_group = QButtonGroup()
        self.setLayout(vlayout)
        
    def create_button(self, button_name):
        button = QPushButton(button_name)
        button.setFlat(True)
        button.setCheckable(True)
        button.clicked.connect(self.clickedButton)
        return button

    def add_button(self, button_name):
        button = self.create_button(button_name)
        self.layout().addWidget(button)
        self.button_group.addButton(button)
        return button
    
    def clickedButton(self):
        button = self.get_checked_button()
        label_name = str(button.text())
        print label_name

    def get_checked_button(self):
        return self.button_group.checkedButton()


class ButtonArea(QWidget):
    def __init__(self, parent=None):
        QWidget.__init__(self, parent)

        self.label_names = []
        self.label_properties = {}

        self.properties = {}
        self.property_buttons = []

        self.label_button_list = ButtonListWidget("Labels")
        self.property_button_lists = {}
        
        self.last_checked_label = None

        self.hlayout = QHBoxLayout()
        self.hlayout.setAlignment(Qt.AlignLeft)
        self.hlayout.addWidget(self.label_button_list)
        self.setLayout(self.hlayout)
        
    def init_button_lists(self):
        for label_name in self.label_names:
            button = self.label_button_list.add_button(label_name)
            button.clicked.connect(self.clickedLabelButton)

        for key, property_values in self.properties.iteritems():
            if key in ["type", "class"]:
                continue
            button_list = ButtonListWidget(key)
            for value in property_values:
                button_list.add_button(value)
            button_list.hide()
            print key
            self.property_button_lists[key] = button_list
            self.hlayout.addWidget(button_list)

    def show_only_label_properties(self, label_name):
        print "sdf", self.property_button_lists
        for name, button_list in self.property_button_lists.iteritems():
            if name in self.label_properties[label_name].keys():
                button_list.show()
            else:
                button_list.hide()
    
    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_button_list.get_checked_button()

    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 "ButtonArea:", label_name, self.last_checked_label
            self.last_checked_label = label_name
            self.show_only_label_properties(label_name)

    def load(self, config_filepath):
        execfile(config_filepath)
        self.init_button_lists()
        
def main():
    app = QApplication(sys.argv)

    ba = ButtonArea()
    ba.load("example_config.py")
    ba.show()

    return app.exec_()

if __name__ == '__main__':
    sys.exit(main())