added model

texter/texter/text_model.py

@@ -0,0 +1,78 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+from PyQt4 import QtCore, QtGui
+from operator import itemgetter
+
+class TextModel(QtCore.QAbstractTableModel):
+    def __init__(self, parent=None):
+        super(TextModel, self).__init__(parent)
+        self.text_db = list()
+        self.default_size = 32
+        self.font = QtGui.QFont("monospace", self.default_size)
+        self.font.setStyleHint(QtGui.QFont.TypeWriter)
+
+    def rowCount(self, parent=QtCore.QModelIndex()):
+        return len(self.text_db)
+
+    def columnCount(self, parent=QtCore.QModelIndex()):
+        return 2
+
+    def data(self, index, role):
+        if not index.isValid() or \
+            not 0 <= index.row() < self.rowCount():
+            return QVariant()
+
+        row = index.row()
+        column = index.column()
+        if role == QtCore.Qt.DisplayRole:
+            return self.text_db[row][column]
+            #return "foo bar"
+
+        return QtCore.QVariant()
+
+
+    def headerData(self, section, orientation=QtCore.Qt.Horizontal, role=QtCore.Qt.DisplayRole):
+        if orientation == QtCore.Qt.Horizontal:
+            if section == 0:
+                return "Preview"
+            else:
+                return "Text"
+        return QtCore.QVariant()
+
+    def setData(self, index, value, role):
+
+        if role == QtCore.Qt.EditRole:
+            self.text_db[index.row()][index.column()] = value.toString()
+
+        return True
+
+    def flags(self, index):
+        return QtCore.Qt.ItemIsSelectable |  QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsEnabled
+
+    def insertRows(self, row, count, parent=QtCore.QModelIndex()):
+        self.beginInsertRows(parent, row, row+count+1)
+        for i in range(row, row+count):
+            self.text_db.insert(parent.child(i).data())
+        self.endInsertRows()
+        return True
+
+    def removeRows(self, row, count, parent=QtCore.QModelIndex()):
+        print "removeRows", row, count
+        print map(itemgetter(0), self.text_db)
+        self.beginRemoveRows(QtCore.QModelIndex(), row, row+count+1)
+        for i in range(row, row+count):
+            print "del", i, self.text_db[row]
+            self.text_db.pop(row)
+        print "after"
+        print map(itemgetter(0), self.text_db)
+        self.endRemoveRows()
+        return True
+
+
+    def text_by_preview(self, preview):
+        for title, text in self.text_db:
+            if title == preview:
+                return title, text
+        return None
+