from PySide2.QtCore import Qt, QSize, QRegExp
from PySide2.QtGui import QColor, QTextCharFormat, QFont, QSyntaxHighlighter, QPainter
from PySide2.QtWidgets import QApplication, QMainWindow, QPlainTextEdit, QWidget, QTextEdit
class PythonHighlighter(QSyntaxHighlighter):
def __init__(self, parent=None):
super().__init__(parent)
keyword_format = QTextCharFormat()
keyword_format.setForeground(Qt.blue)
keywords = [
'and', 'as', 'assert', 'break', 'class', 'continue', 'def',
'del', 'elif', 'else', 'except', 'False', 'finally', 'for',
'from', 'global', 'if', 'import', 'in', 'is', 'lambda',
'None', 'nonlocal', 'not', 'or', 'pass', 'raise', 'return',
'True', 'try', 'while', 'with', 'yield'
]
self.keyword_pattern = '|'.join(keywords)
self.keyword_format = keyword_format
integer_format = QTextCharFormat()
integer_format.setForeground(Qt.green)
self.integer_pattern = r'\b\d+\b'
self.integer_format = integer_format
def highlightBlock(self, text):
self.setFormat(0, len(text), QTextCharFormat())
keyword_expression = '\\b(' + self.keyword_pattern + ')\\b'
integer_expression = self.integer_pattern
keyword_format = self.keyword_format
integer_format = self.integer_format
for pattern, format in [(keyword_expression, keyword_format),
(integer_expression, integer_format)]:
expression = QRegExp(pattern)
index = expression.indexIn(text)
while index >= 0:
length = expression.matchedLength()
self.setFormat(index, length, format)
index = expression.indexIn(text, index + length)
class CodeEditor(QPlainTextEdit):
def __init__(self, parent=None):
super().__init__(parent)
font = QFont()
font.setFamily('Courier')
font.setFixedPitch(True)
font.setPointSize(10)
self.setFont(font)
self.highlighter = PythonHighlighter(self)
self.lineNumberArea = LineNumberArea(self)
self.blockCountChanged.connect(self.updateLineNumberAreaWidth)
self.updateRequest.connect(self.updateLineNumberArea)
self.cursorPositionChanged.connect(self.highlightCurrentLine)
self.updateLineNumberAreaWidth(0)
def lineNumberAreaWidth(self):
digits = 1
count = max(1, self.blockCount())
while count >= 10:
count /= 10
digits += 1
return 3 + self.fontMetrics().width('9') * digits
def updateLineNumberAreaWidth(self, _):
print("updateLineNumberAreaWidth")
self.setViewportMargins(self.lineNumberAreaWidth(), 0, 0, 0)
def updateLineNumberArea(self, rect, dy):
print("updateLineNumberArea")
if dy:
self.lineNumberArea.scroll(0, dy)
else:
self.lineNumberArea.update(0, rect.y(), self.lineNumberArea.width(),
rect.height())
if rect.contains(self.viewport().rect()):
self.updateLineNumberAreaWidth(0)
def resizeEvent(self, event):
super().resizeEvent(event)
cr = self.contentsRect()
self.lineNumberArea.setGeometry(cr.left(), cr.top(),
self.lineNumberAreaWidth(), cr.height())
def highlightCurrentLine(self):
print("highlightCurrentLine")
extra_selections = []
if not self.isReadOnly():
selection = QTextEdit.ExtraSelection()
line_color = QColor(Qt.yellow).lighter(160)
selection.format.setBackground(line_color)
selection.format.setProperty(QTextCharFormat.FullWidthSelection, True)
selection.cursor = self.textCursor()
selection.cursor.clearSelection()
extra_selections.append(selection)
self.setExtraSelections(extra_selections)
class LineNumberArea(QWidget):
def __init__(self, editor):
super().__init__(editor)
self.editor = editor
def sizeHint(self):
return QSize(self.editor.lineNumberAreaWidth(), 0)
def paintEvent(self, event):
painter = QPainter(self)
painter.fillRect(event.rect(), Qt.lightGray)
block = self.editor.firstVisibleBlock()
block_number = block.blockNumber()
top = self.editor.blockBoundingGeometry(block).translated(self.editor.contentOffset()).top()
bottom = top + self.editor.blockBoundingRect(block).height()
height = self.editor.fontMetrics().height()
while block.isValid() and top <= event.rect().bottom():
if block.isVisible() and bottom >= event.rect().top():
number = str(block_number + 1)
painter.setPen(Qt.black)
painter.drawText(0, top, self.width(), height, Qt.AlignRight, number)
block = block.next()
top = bottom
bottom = top + self.editor.blockBoundingRect(block).height()
block_number += 1
if __name__ == '__main__':
app = QApplication([])
window = QMainWindow()
editor = CodeEditor()
window.setCentralWidget(editor)
window.show()
app.exec_()