2015-06-07 02:21:08 +12:00
|
|
|
#!/usr/bin/env python
|
|
|
|
#--!-- coding: utf8 --!--
|
|
|
|
|
|
|
|
from qt import *
|
|
|
|
from enums import *
|
|
|
|
from ui.editors.t2tHighlighter import *
|
2015-06-15 22:18:42 +12:00
|
|
|
from functions import *
|
|
|
|
|
2015-06-07 02:21:08 +12:00
|
|
|
try:
|
|
|
|
import enchant
|
|
|
|
except ImportError:
|
|
|
|
enchant = None
|
|
|
|
|
2015-06-15 22:18:42 +12:00
|
|
|
class textEditView(QTextEdit):
|
2015-06-07 02:21:08 +12:00
|
|
|
|
2015-06-07 06:29:02 +12:00
|
|
|
def __init__(self, parent=None, index=None, html=None, spellcheck=True, dict="", autoResize=False):
|
2015-06-07 02:21:08 +12:00
|
|
|
QTextEdit.__init__(self, parent)
|
2015-06-07 06:29:02 +12:00
|
|
|
|
2015-06-15 22:18:42 +12:00
|
|
|
self._column = Outline.text.value
|
|
|
|
self._index = None
|
|
|
|
self._indexes = None
|
|
|
|
self._placeholderText = None
|
|
|
|
self._updating = False
|
|
|
|
self._item = None
|
|
|
|
self._highlighting = True
|
|
|
|
|
2015-06-07 05:10:44 +12:00
|
|
|
self.spellcheck = spellcheck
|
|
|
|
self.currentDict = dict
|
2015-06-08 12:07:08 +12:00
|
|
|
self.highlighter = None
|
2015-06-15 22:18:42 +12:00
|
|
|
self._autoResize = autoResize
|
2015-06-07 05:10:44 +12:00
|
|
|
|
|
|
|
if index:
|
|
|
|
self.setCurrentModelIndex(index)
|
|
|
|
|
|
|
|
elif html:
|
|
|
|
self.document().setHtml(html)
|
|
|
|
self.setReadOnly(True)
|
2015-06-15 22:18:42 +12:00
|
|
|
|
|
|
|
self.setAutoResize(self._autoResize)
|
|
|
|
|
|
|
|
def setModel(self, model):
|
|
|
|
self._model = model
|
|
|
|
self._model.dataChanged.connect(self.update)
|
2015-06-07 08:39:43 +12:00
|
|
|
|
2015-06-15 22:18:42 +12:00
|
|
|
def setColumn(self, col):
|
|
|
|
self._column = col
|
|
|
|
|
|
|
|
def setHighlighting(self, val):
|
|
|
|
self._highlighting = val
|
2015-06-07 05:10:44 +12:00
|
|
|
|
|
|
|
def setCurrentModelIndex(self, index):
|
2015-06-15 22:18:42 +12:00
|
|
|
self._indexes = None
|
2015-06-07 05:10:44 +12:00
|
|
|
if index.isValid():
|
2015-06-15 22:18:42 +12:00
|
|
|
if index.column() != self._column:
|
|
|
|
index = index.sibling(index.row(), self._column)
|
|
|
|
self._index = index
|
|
|
|
self._item = index.internalPointer()
|
|
|
|
if self._placeholderText != None:
|
|
|
|
self.setPlaceholderText(self._placeholderText)
|
|
|
|
|
2015-06-07 05:10:44 +12:00
|
|
|
self._model = index.model()
|
|
|
|
self.document().contentsChanged.connect(self.submit)
|
|
|
|
self._model.dataChanged.connect(self.update)
|
|
|
|
self.updateText()
|
|
|
|
|
|
|
|
self.defaultFontPointSize = qApp.font().pointSize()
|
|
|
|
self.highlightWord = ""
|
|
|
|
self.highligtCS = False
|
|
|
|
|
2015-06-15 22:18:42 +12:00
|
|
|
if self._highlighting and not self.highlighter:
|
2015-06-08 12:07:08 +12:00
|
|
|
self.highlighter = t2tHighlighter(self)
|
2015-06-07 05:10:44 +12:00
|
|
|
|
|
|
|
# Spellchecking
|
|
|
|
if enchant and self.spellcheck:
|
|
|
|
self.dict = enchant.Dict(self.currentDict if self.currentDict else enchant.get_default_language())
|
2015-06-08 08:06:57 +12:00
|
|
|
else:
|
|
|
|
self.spellcheck = False
|
2015-06-07 05:10:44 +12:00
|
|
|
|
2015-06-15 22:18:42 +12:00
|
|
|
def setCurrentModelIndexes(self, indexes):
|
|
|
|
self._index = None
|
|
|
|
self._indexes = []
|
|
|
|
|
|
|
|
for i in indexes:
|
|
|
|
if i.isValid():
|
|
|
|
if i.column() != self._column:
|
|
|
|
i = i.sibling(i.row(), self._column)
|
|
|
|
self._indexes.append(i)
|
|
|
|
|
|
|
|
self.document().contentsChanged.connect(self.submit)
|
|
|
|
self.updateText()
|
2015-06-07 02:21:08 +12:00
|
|
|
|
2015-06-07 05:10:44 +12:00
|
|
|
def update(self, topLeft, bottomRight):
|
2015-06-15 22:38:38 +12:00
|
|
|
if self._updating:
|
2015-06-15 22:18:42 +12:00
|
|
|
return
|
|
|
|
|
|
|
|
elif self._index:
|
|
|
|
if topLeft.row() <= self._index.row() <= bottomRight.row():
|
|
|
|
self.updateText()
|
|
|
|
|
|
|
|
elif self._indexes:
|
|
|
|
update = False
|
|
|
|
for i in self._indexes:
|
|
|
|
if topLeft.row() <= i.row() <= bottomRight.row():
|
|
|
|
update = True
|
|
|
|
if update:
|
|
|
|
self.updateText()
|
2015-06-07 02:21:08 +12:00
|
|
|
|
2015-06-07 05:10:44 +12:00
|
|
|
def updateText(self):
|
2015-06-15 22:38:38 +12:00
|
|
|
self._updating = True
|
2015-06-15 22:18:42 +12:00
|
|
|
if self._index:
|
|
|
|
if self.toPlainText() != toString(self._model.data(self._index)):
|
|
|
|
self.document().setPlainText(toString(self._model.data(self._index)))
|
|
|
|
|
|
|
|
elif self._indexes:
|
|
|
|
t = []
|
|
|
|
same = True
|
|
|
|
for i in self._indexes:
|
|
|
|
item = i.internalPointer()
|
|
|
|
t.append(toString(item.data(self._column)))
|
|
|
|
|
|
|
|
for t2 in t[1:]:
|
|
|
|
if t2 != t[0]:
|
|
|
|
same = False
|
|
|
|
break
|
|
|
|
|
|
|
|
if same:
|
|
|
|
self.document().setPlainText(t[0])
|
|
|
|
else:
|
|
|
|
self.document().setPlainText("")
|
|
|
|
|
|
|
|
if not self._placeholderText:
|
|
|
|
self._placeholderText = self.placeholderText()
|
|
|
|
|
|
|
|
self.setPlaceholderText(self.tr("Various"))
|
2015-06-15 22:38:38 +12:00
|
|
|
self._updating = False
|
2015-06-07 05:10:44 +12:00
|
|
|
|
2015-06-15 22:18:42 +12:00
|
|
|
def submit(self):
|
2015-06-15 22:38:38 +12:00
|
|
|
|
|
|
|
if self._updating:
|
|
|
|
return
|
|
|
|
|
2015-06-15 22:18:42 +12:00
|
|
|
if self._index:
|
|
|
|
#item = self._index.internalPointer()
|
|
|
|
if self.toPlainText() != self._model.data(self._index):
|
2015-06-15 22:38:38 +12:00
|
|
|
self._updating = True
|
2015-06-15 22:18:42 +12:00
|
|
|
self._model.setData(self._index, self.toPlainText())
|
2015-06-15 22:38:38 +12:00
|
|
|
self._updating = False
|
2015-06-15 22:18:42 +12:00
|
|
|
|
|
|
|
elif self._indexes:
|
|
|
|
self._updating = True
|
|
|
|
for i in self._indexes:
|
|
|
|
item = i.internalPointer()
|
|
|
|
if self.toPlainText() != toString(item.data(self._column)):
|
|
|
|
self._model.setData(i, self.toPlainText())
|
|
|
|
self._updating = False
|
|
|
|
|
|
|
|
# -----------------------------------------------------------------------------------------------------
|
|
|
|
# Resize stuff
|
|
|
|
|
2015-06-07 05:10:44 +12:00
|
|
|
def resizeEvent(self, e):
|
|
|
|
QTextEdit.resizeEvent(self, e)
|
2015-06-15 22:18:42 +12:00
|
|
|
if self._autoResize:
|
2015-06-07 06:29:02 +12:00
|
|
|
self.sizeChange()
|
2015-06-07 05:10:44 +12:00
|
|
|
|
|
|
|
def sizeChange(self):
|
|
|
|
docHeight = self.document().size().height()
|
|
|
|
if self.heightMin <= docHeight <= self.heightMax:
|
|
|
|
self.setMinimumHeight(docHeight)
|
|
|
|
|
2015-06-15 22:18:42 +12:00
|
|
|
def setAutoResize(self, val):
|
|
|
|
self._autoResize = val
|
|
|
|
if self._autoResize:
|
|
|
|
self.document().contentsChanged.connect(self.sizeChange)
|
|
|
|
self.heightMin = 0
|
|
|
|
self.heightMax = 65000
|
|
|
|
self.sizeChange()
|
2015-06-07 05:10:44 +12:00
|
|
|
|
|
|
|
# -----------------------------------------------------------------------------------------------------
|
|
|
|
# Spellchecking based on http://john.nachtimwald.com/2009/08/22/qplaintextedit-with-in-line-spell-check/
|
|
|
|
|
|
|
|
def setDict(self, d):
|
|
|
|
self.dict = enchant.Dict(d)
|
|
|
|
self.highlighter.rehighlight()
|
|
|
|
|
|
|
|
def toggleSpellcheck(self, v):
|
|
|
|
self.spellcheck = v
|
2015-06-15 22:18:42 +12:00
|
|
|
if self.highlighter:
|
|
|
|
self.highlighter.rehighlight()
|
2015-06-07 05:10:44 +12:00
|
|
|
|
|
|
|
def mousePressEvent(self, event):
|
|
|
|
if event.button() == Qt.RightButton:
|
|
|
|
# Rewrite the mouse event to a left button event so the cursor is
|
|
|
|
# moved to the location of the pointer.
|
|
|
|
event = QMouseEvent(QEvent.MouseButtonPress, event.pos(),
|
|
|
|
Qt.LeftButton, Qt.LeftButton, Qt.NoModifier)
|
|
|
|
QTextEdit.mousePressEvent(self, event)
|
|
|
|
|
|
|
|
class SpellAction(QAction):
|
|
|
|
"A special QAction that returns the text in a signal. Used for spellckech."
|
|
|
|
|
2015-06-08 08:06:57 +12:00
|
|
|
correct = pyqtSignal(str)
|
2015-06-07 05:10:44 +12:00
|
|
|
|
|
|
|
def __init__(self, *args):
|
|
|
|
QAction.__init__(self, *args)
|
|
|
|
|
|
|
|
self.triggered.connect(lambda x: self.correct.emit(
|
2015-06-08 08:06:57 +12:00
|
|
|
str(self.text())))
|
2015-06-07 05:10:44 +12:00
|
|
|
|
|
|
|
def contextMenuEvent(self, event):
|
|
|
|
# Based on http://john.nachtimwald.com/2009/08/22/qplaintextedit-with-in-line-spell-check/
|
|
|
|
|
|
|
|
if not self.spellcheck:
|
|
|
|
QTextEdit.contextMenuEvent(self, event)
|
|
|
|
return
|
|
|
|
|
|
|
|
popup_menu = self.createStandardContextMenu()
|
|
|
|
|
|
|
|
# Select the word under the cursor.
|
|
|
|
cursor = self.textCursor()
|
|
|
|
cursor.select(QTextCursor.WordUnderCursor)
|
|
|
|
self.setTextCursor(cursor)
|
|
|
|
|
|
|
|
# Check if the selected word is misspelled and offer spelling
|
|
|
|
# suggestions if it is.
|
|
|
|
if self.textCursor().hasSelection():
|
2015-06-08 08:06:57 +12:00
|
|
|
text = str(self.textCursor().selectedText())
|
2015-06-07 05:10:44 +12:00
|
|
|
if not self.dict.check(text):
|
2015-06-08 22:01:45 +12:00
|
|
|
spell_menu = QMenu(self.tr('Spelling Suggestions'))
|
2015-06-07 05:10:44 +12:00
|
|
|
for word in self.dict.suggest(text):
|
|
|
|
action = self.SpellAction(word, spell_menu)
|
|
|
|
action.correct.connect(self.correctWord)
|
|
|
|
spell_menu.addAction(action)
|
|
|
|
# Only add the spelling suggests to the menu if there are
|
|
|
|
# suggestions.
|
|
|
|
if len(spell_menu.actions()) != 0:
|
|
|
|
popup_menu.insertSeparator(popup_menu.actions()[0])
|
|
|
|
popup_menu.insertMenu(popup_menu.actions()[0], spell_menu)
|
|
|
|
|
|
|
|
popup_menu.exec_(event.globalPos())
|
|
|
|
|
|
|
|
def correctWord(self, word):
|
|
|
|
'''
|
|
|
|
Replaces the selected text with word.
|
|
|
|
'''
|
|
|
|
cursor = self.textCursor()
|
|
|
|
cursor.beginEditBlock()
|
|
|
|
|
|
|
|
cursor.removeSelectedText()
|
|
|
|
cursor.insertText(word)
|
|
|
|
|
|
|
|
cursor.endEditBlock()
|
|
|
|
|
|
|
|
# -----------------------------------------------------------------------------------------------------
|