sloth/buttonarea.py

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

class ButtonListWidget(QWidget):
    selectionChanged = pyqtSignal(object)

    def __init__(self, name, parent=None):
        QWidget.__init__(self, parent)
        self.name = name
        vlayout = QVBoxLayout()
        vlayout.setSpacing(0)
        vlayout.setAlignment(Qt.AlignTop)
        vlayout.addWidget(QLabel("<center><b>" + name + "</b></center>"))
        self.button_group = QButtonGroup()
        self.button_group.setExclusive(False)
        self.buttons = {}
        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.buttons[button_name] = button
        self.layout().addWidget(button)
        self.button_group.addButton(button)
        return button

    def get_button(self, button_name):
        return self.buttons[button_name]

    def toggleChecked(self, button_name, apply=True):
        selection = None

        for button in self.button_group.buttons():
            if button.text() != button_name:
                button.setChecked(False)
            else:
                if apply:
                    button.setChecked(not button.isChecked())
                if button.isChecked():
                    selection = button_name

        self.selectionChanged.emit(selection)

    def clickedButton(self):
        button_name = str(self.sender().text())
        self.toggleChecked(button_name, False)

        #for button in self.button_group.buttons():
            #if button is not self.sender():
                #button.setChecked(False)
        #print "sender:", label_name

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


class ButtonArea(QWidget):
    stateChanged = pyqtSignal(object)

    def __init__(self, labels=None, hotkeys=None, 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.hotkeys = []

        self.hlayout = QHBoxLayout()
        self.hlayout.setAlignment(Qt.AlignLeft)
        self.hlayout.addWidget(self.label_button_list)
        self.setLayout(self.hlayout)
        self.stateChanged.connect(self.stateHasChanged)

        if labels is not None:
            for name, prop in labels:
                self.add_label(name, prop)
        if hotkeys is not None:
            for choice, name, hotkey in hotkeys:
                self.add_hotkey(choice, name, hotkey)
        self.init_button_lists()

    def stateHasChanged(self, state):
        print "stateChanged(object)", state

    def init_button_lists(self):
        for label_name in self.label_names:
            button = self.label_button_list.add_button(label_name)
        self.label_button_list.selectionChanged.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 = button_list.add_button(value)
                #button.clicked.connect(self.clickedButton)
            button_list.selectionChanged.connect(self.clickedButton)

            button_list.hide()
            print key
            self.property_button_lists[key] = button_list
            self.hlayout.addWidget(button_list)

        for choice, name, hotkey in self.hotkeys:
            if choice == "" or choice is None:
                button = self.label_button_list.get_button(name)
                shortcut = QShortcut(QKeySequence(hotkey), button, button.click)
            else:
                button = self.property_button_lists[choice].get_button(name)
                shortcut = QShortcut(QKeySequence(hotkey), button, button.click)
            button.setToolTip("[" + str(hotkey) + "]")

    def show_only_label_properties(self, label_name):
        for name, button_list in self.property_button_lists.iteritems():
            if self.label_properties.has_key(label_name) and 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 add_hotkey(self, choice, name, hotkey):
        self.hotkeys.append((choice, name, hotkey))

    def get_current_state(self):
        label_button = self.get_checked_label_button()
        if label_button != None:
            result = {}
            label = str(label_button.text())
            if self.label_properties.has_key(label):
                if self.label_properties[label].has_key("type"):
                    result["type"] = self.label_properties[label]["type"]
                if self.label_properties[label].has_key("class"):
                    result["class"] = self.label_properties[label]["class"]
                for name, button_list in self.property_button_lists.iteritems():
                    if button_list.isVisible():
                        checked_button = button_list.get_checked_button()
                        if checked_button != None:
                            result[button_list.name] = str(checked_button.text())
            return result

        return None

    def clickedButton(self, newselection):
        print "selectionChanged:", newselection
        self.stateChanged.emit(self.get_current_state())

    def clickedLabelButton(self, label_name):
        #button = self.get_checked_label_button()
        #print button
        if label_name != None:
            print "ButtonArea:", label_name
            self.show_only_label_properties(label_name)
        else:
            print "Selection Mode"
            self.show_only_label_properties("")
        self.stateChanged.emit(self.get_current_state())

def main():
    from conf import config
    config.update("example_config")

    app = QApplication(sys.argv)
    ba = ButtonArea(config.LABELS, config.HOTKEYS)
    ba.show()

    return app.exec_()

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