psychose/texter/texter/text_model.py

87 lines
2.7 KiB
Python

#!/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 role not in (1,3,4,5,6,7,8,9,10,13):
print "role", 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"
elif role == QtCore.Qt.ForegroundRole:
return QtGui.QBrush(QtCore.Qt.black)
elif role == QtCore.Qt.BackgroundRole:
return QtGui.QBrush(QtCore.Qt.white)
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:
text = value.toString()
if not text:
return False
else:
self.text_db[index.row()][index.column()] = text
return True
def flags(self, index):
return QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsDragEnabled | QtCore.Qt.ItemIsDropEnabled
def supportedDropActions(self):
return QtCore.Qt.MoveAction
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()):
self.beginRemoveRows(QtCore.QModelIndex(), row, row+count+1)
for i in range(row, row+count):
self.text_db.pop(row)
self.endRemoveRows()
return True
def text_by_preview(self, preview):
for ix, (title, text) in enumerate(self.text_db):
if title == preview:
return ix, title, text
return None