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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
from qtpy import QtCore
from qtpy import QtGui
from qtpy import QtWidgets
have_pygments = True
try:
from pygments.styles import get_style_by_name
from pygments import lex
from pygments.util import ClassNotFound
from pygments.lexers import get_lexer_for_filename
except ImportError:
have_pygments = False
def highlight_document(edit, filename):
if not have_pygments:
return
doc = edit.document()
try:
lexer = get_lexer_for_filename(filename, stripnl=False)
except ClassNotFound:
return
style = get_style_by_name('default')
font = doc.defaultFont()
base_format = QtGui.QTextCharFormat()
base_format.setFont(font)
token_formats = {}
window = edit.window()
if hasattr(window, 'processEvents'):
processEvents = window.processEvents
else:
processEvents = QtCore.QCoreApplication.processEvents
def get_token_format(token):
if token in token_formats:
return token_formats[token]
if token.parent:
parent_format = get_token_format(token.parent)
else:
parent_format = base_format
fmt = QtGui.QTextCharFormat(parent_format)
font = fmt.font()
if style.styles_token(token):
tstyle = style.style_for_token(token)
if tstyle['color']:
fmt.setForeground(QtGui.QColor('#' + tstyle['color']))
if tstyle['bold']:
font.setWeight(QtGui.QFont.Bold)
if tstyle['italic']:
font.setItalic(True)
if tstyle['underline']:
fmt.setFontUnderline(True)
if tstyle['bgcolor']:
fmt.setBackground(QtGui.QColor('#' + tstyle['bgcolor']))
token_formats[token] = fmt
return fmt
text = doc.toPlainText()
block_count = 0
block = doc.firstBlock()
assert isinstance(block, QtGui.QTextBlock)
block_pos = 0
block_len = block.length()
block_formats = []
for token, ttext in lex(text, lexer):
format_len = len(ttext)
fmt = get_token_format(token)
while format_len > 0:
format_range = QtGui.QTextLayout.FormatRange()
format_range.start = block_pos
format_range.length = min(format_len, block_len)
format_range.format = fmt
block_formats.append(format_range)
block_len -= format_range.length
format_len -= format_range.length
block_pos += format_range.length
if block_len == 0:
block.layout().setAdditionalFormats(block_formats)
doc.markContentsDirty(block.position(), block.length())
block = block.next()
block_pos = 0
block_len = block.length()
block_formats = []
block_count += 1
if block_count % 100 == 0:
processEvents()
if __name__ == '__main__':
app = QtWidgets.QApplication([])
python = QtWidgets.QPlainTextEdit()
with open(__file__, encoding='utf-8') as f:
python.setPlainText(f.read())
python.setWindowTitle('python')
python.show()
highlight_document(python, __file__)
app.exec_()
|
