-
Notifications
You must be signed in to change notification settings - Fork 6
/
spellcheck_highlighter.py
82 lines (62 loc) · 2.67 KB
/
spellcheck_highlighter.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
################################################################################
### Copyright © 2012-2013 BlackDragonHunt
###
### This file is part of the Super Duper Script Editor.
###
### The Super Duper Script Editor is free software: you can redistribute it
### and/or modify it under the terms of the GNU General Public License as
### published by the Free Software Foundation, either version 3 of the License,
### or (at your option) any later version.
###
### The Super Duper Script Editor is distributed in the hope that it will be
### useful, but WITHOUT ANY WARRANTY; without even the implied warranty of
### MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
### GNU General Public License for more details.
###
### You should have received a copy of the GNU General Public License
### along with the Super Duper Script Editor.
### If not, see <http://www.gnu.org/licenses/>.
################################################################################
from PyQt4 import QtGui, QtCore
from PyQt4.QtGui import QTextCharFormat, QColor
import re
import enchant
from enchant.checker import SpellChecker
from enchant.tokenize import HTMLChunker
import common
RE_ANGLED_APOST = re.compile(ur"[‘’]", re.UNICODE)
class SpellCheckHighlighter(QtGui.QSyntaxHighlighter):
def __init__(self, parent = None):
super(SpellCheckHighlighter, self).__init__(parent)
self.set_language("en_US")
self.format = QTextCharFormat()
self.format.setUnderlineColor(QColor(255, 0, 0))
self.format.setUnderlineStyle(QTextCharFormat.SpellCheckUnderline)
self.errors = []
def set_language(self, lang):
dict = enchant.DictWithPWL(lang, "data/dict/enchant.txt")
self.checker = SpellChecker(dict, chunkers = (HTMLChunker,))
def get_language(self):
return self.checker.dict.tag
def highlightBlock(self, text):
# If there is no previous state, then it's -1, which makes the first line 0.
# And every line after that increases as expected.
line = self.previousBlockState() + 1
self.setCurrentBlockState(line)
# Make sure our error list is long enough to hold this line.
for i in range(len(self.errors), line + 1):
self.errors.append([])
text = common.qt_to_unicode(text)
text = RE_ANGLED_APOST.sub("'", text)
self.errors[line] = []
self.checker.set_text(text)
for err in self.checker:
self.setFormat(err.wordpos, len(err.word), self.format)
self.errors[line].append((err.word, err.wordpos))
def add(self, word):
self.checker.add(word)
self.rehighlight()
def ignore(self, word):
self.checker.ignore_always(word)
self.rehighlight()
### EOF ###