psychose/texter/main.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

import PyQt4.uic
from PyQt4 import QtCore, QtGui
from PyKDE4.kdeui import KActionCollection, KRichTextWidget

MainWindowForm, MainWindowBase = PyQt4.uic.loadUiType('texter.ui')

from operator import itemgetter
import cPickle

class MainWindow(MainWindowBase, MainWindowForm):
    def __init__(self, parent = None):
        super(MainWindow, self).__init__(parent)


        # setup the ui
        self.setupUi(self)
        #self.show_public_text.setRichTextSupport(KRichTextWidget.RichTextSupport(0xffffffff))
        self.edit_private_text.setRichTextSupport(KRichTextWidget.RichTextSupport(0xffffffff))
        self.edit_action_collection = KActionCollection(self)
        self.edit_private_text.createActions(self.edit_action_collection)
        for action in self.edit_action_collection.actions():
            self.toolBar.addAction(action)

        self.current_text = None
        self.current_title = None
        self.current_index = None
        self.is_published = False

        self.tabWidget.currentChanged.connect(self.slot_toggleToolbox)
        self.add_button.clicked.connect(self.slot_addText)
        self.save_button.clicked.connect(self.slot_save)
        self.publish_button.toggled.connect(self.slot_toggle_publish)
        self.remove_item_button.clicked.connect(self.slot_removeItem)
        self.edit_item_selection.activated.connect(self.slot_editLoadItem)
        self.item_list.currentRowChanged.connect(self.slot_showLoadItem)
        self.auto_publish_checkbox.toggled.connect(self.slot_toggle_publish)

        self.items = dict()
        self.slot_load()
        self.next_button.setShortcut(QtGui.QKeySequence(QtCore.Qt.Key_Space))
        self.next_button.clicked.connect(self.slot_next_item)

        self.previous_button.setShortcut(QtGui.QKeySequence(QtCore.Qt.Key_Backspace))
        self.previous_button.clicked.connect(self.slot_previous_item)


    def slot_next_item(self):
        print "slot_next_item"
        print "current_title", self.current_title, self.current_text
        index = (self.item_list.currentRow() + 1) % self.item_list.count()
        self.item_list.setCurrentRow(index)
        self.edit_item_selection.setCurrentIndex(index)
        self.slot_editLoadItem(index)
        self.slot_showLoadItem(index)

    def slot_previous_item(self):
        print "slot_previous_item"
        print "current_title", self.current_title, self.current_text
        index = (self.item_list.currentRow() - 1) % self.item_list.count()
        self.item_list.setCurrentRow(index)
        self.edit_item_selection.setCurrentIndex(index)
        self.slot_editLoadItem(index)
        self.slot_showLoadItem(index)

    def slot_toggleToolbox(self, index):
        print "slot_toggleToolbox"
        print "current_title", self.current_title, self.current_text
        if index == 0:
            self.toolBar.setEnabled(True)
        else:
            self.toolBar.setEnabled(False)

    def slot_publish(self):
        #QPropertyAnimation animation(self.public_text.palette(), "geometry");
        #animation.setDuration(10000);
        #animation.setStartValue(QRect(0, 0, 100, 30));
        #animation.setEndValue(QRect(250, 250, 100, 30));
        #animation.start();
        print "slot_publish"
        print "current_title", self.current_title, self.current_text
        self.show_public_text.setTextOrHtml(self.current_text)
        self.is_published = True

    def slot_toggle_publish(self, state=None):
        #QPropertyAnimation animation(self.public_text.palette(), "geometry");
        #animation.setDuration(10000);
        #animation.setStartValue(QRect(0, 0, 100, 30));
        #animation.setEndValue(QRect(250, 250, 100, 30));
        #animation.start();
        print "slot_toggle_publish", state
        print "current_title", self.current_title, self.current_text

        if state:
            self.slot_publish()
        else:
            self.slot_clear()

    def slot_clear(self):
        print "slot_clear"
        print "current_title", self.current_title, self.current_text

        self.show_public_text.clear()
        self.is_published = False

    def slot_removeItem(self):
        text = self.edit_item_selection.currentText()
        index = self.edit_item_selection.currentIndex()
        title = text.split(": ")[1]
        del self.items[title]
        self.edit_item_selection.removeItem(index)
        self.item_list.removeItemWidget(self.item_list.item(index))
        new_index = self.edit_item_selection.currentIndex()
        if new_index != -1:
            self.slot_editLoadItem()
        else:
            self.item_title.clear()
            self.item_position_input.setValue(0)


    def slot_editLoadItem(self, index):
        print "slot_editLoadItem", index
        itemText = self.edit_item_selection.itemText(index)
        position, title = itemText.split(": ", 1)
        text, position = self.items[title]
        self.edit_private_text.setTextOrHtml(text)
        self.item_title.setText(title)
        self.item_position_input.setValue(position)

    def slot_showLoadItem(self, index):
        public_rect = self.show_public_text.geometry()
        global_rect = QtCore.QRect(self.mapToGlobal(public_rect.topLeft()), self.mapToGlobal(public_rect.bottomRight()))
        print "slot_showLoadItem", global_rect, index
        print "current_title", self.current_title, self.current_text
        item = self.item_list.item(index)
        print "item", item
        if item is None:
            return
        self.current_title = item.text()
        self.current_text, self.current_index = self.items[self.current_title]
        print "current_title", self.current_title, self.current_text
        if self.auto_publish_checkbox.isChecked():
            self.show_public_text.setHtml(self.current_text)

    def title_by_index(self, ix):
        for title, (text, index) in self.items.iteritems():
            if index == ix:
                return title
        return None

    def slot_changeItem(self, old_title):
        print "slot_changeItem"
        text, index = self.items.pop(old_title)
        new_text = self.edit_private_text.textOrHtml()
        new_title = self.item_title.text()
        self.items[new_title] = (new_text, index)
        self.show_public_text.setTextOrHtml(new_text)
        self.item_title.setText(new_title)
        self.edit_item_selection.setItemText(index, "%d: %s" % (index, new_title))
        self.item_list.item(index).setText(new_title)

    def slot_addText(self):
        print "slot add"
        index = self.item_position_input.value()
        if index - self.item_list.count() > 1:
            old_title = self.edit_item_selection.currentText().split(": ")[1]
            self.item_title.setText(old_title)
            text, index = self.items[old_title]
            self.edit_private_text.setTextOrHtml(text)
            self.item_position_input.setValue(index)
            return
        old_title = self.title_by_index(index)
        if old_title is not None:
            self.slot_changeItem(old_title)
            return

        title = self.item_title.text()
        text = self.edit_private_text.textOrHtml()
        self.items[title] = (text, index)
        self.edit_item_selection.insertItem(index, "%d: %s" % (index, title))
        self.item_list.insertItem(index, title)
        self.edit_item_selection.setCurrentIndex(index)
        self.item_list.setCurrentRow(index)

    def slot_save(self):
        cPickle.dump(self.items, open("448_texter.db", "w"), cPickle.HIGHEST_PROTOCOL)

    def slot_load(self):
        try:
            self.items = cPickle.load(open("448_texter.db"))
        except Exception, e:
            print e

        data = list()
        for title, (text, index) in self.items.iteritems():
            data.append((title, text, index))

        data = sorted(data, key=itemgetter(2))
        for title, text, index in data:
            self.edit_item_selection.addItem("%d: %s" % (index, title))
            self.item_list.addItem(title)

        self.edit_item_selection.setCurrentIndex(0)
        self.item_list.setCurrentRow(0)
        self.current_title, self.current_text, self.current_index = data[0]

        self.edit_private_text.setTextOrHtml(self.current_text)
        self.item_position_input.setValue(self.current_index)
        self.item_title.setText(self.current_title)



if ( __name__ == '__main__' ):
    app = None
    if ( not app ):
        app = QtGui.QApplication([])

    window = MainWindow()
    window.show()

    if ( app ):
        app.exec_()