90 lines
2.7 KiB
Python
90 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 1
|
|
|
|
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:
|
|
print "data", row, column, row
|
|
return QtCore.QVariant(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 (not index.isValid() or
|
|
not 0 <= index.row() < self.rowCount()):
|
|
print "setData index not valid"
|
|
return False
|
|
|
|
print "setData", index.row(), index.column(), 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
|
|
else:
|
|
return False
|
|
|
|
def flags(self, index):
|
|
if index.column() == 0:
|
|
return QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsEnabled
|
|
else:
|
|
return QtCore.Qt.ItemIsSelectable | 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()):
|
|
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
|
|
|