Created
October 11, 2021 17:33
-
-
Save Patitotective/5f1c3f57a21c57876f3d135570949a47 to your computer and use it in GitHub Desktop.
A QPlainTextEdit with numbered lines.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import sys | |
from PyQt5.QtWidgets import QFrame, QHBoxLayout, QTextEdit, QPlainTextEdit, QWidget, QMainWindow, QApplication | |
from PyQt5.QtGui import QPainter, QTextFormat, QColor | |
from PyQt5.QtCore import QRect, QVariant, Qt | |
class NumberedLinesTextEdit(QFrame): | |
class NumberBar(QWidget): | |
def __init__(self, edit): | |
super().__init__(edit) | |
self.edit = edit | |
self.adjustWidth(1) | |
def paintEvent(self, event): | |
self.edit.numberbarPaint(self, event) | |
QWidget.paintEvent(self, event) | |
def adjustWidth(self, count): | |
width = self.fontMetrics().width(str(count)) | |
if self.width() != width: | |
self.setFixedWidth(width) | |
def updateContents(self, rect, scroll): | |
if scroll: | |
self.scroll(0, scroll) | |
else: | |
self.update() | |
class PlainTextEdit(QPlainTextEdit): | |
def __init__(self, *args): | |
super().__init__(*args) | |
# self.setFrameStyle(QFrame.NoFrame) | |
self.setFrameStyle(QFrame.NoFrame) | |
self.highlight() | |
self.setLineWrapMode(QPlainTextEdit.NoWrap) | |
self.cursorPositionChanged.connect(self.highlight) | |
def highlight(self): | |
hi_selection = QTextEdit.ExtraSelection() | |
hi_selection.format.setBackground(self.palette().alternateBase()) | |
hi_selection.format.setProperty(QTextFormat.FullWidthSelection, QVariant(True)) | |
hi_selection.cursor = self.textCursor() | |
hi_selection.cursor.clearSelection() | |
self.setExtraSelections([hi_selection]) | |
def numberbarPaint(self, number_bar, event): | |
font_metrics = self.fontMetrics() | |
current_line = self.document().findBlock(self.textCursor().position()).blockNumber() + 1 | |
block = self.firstVisibleBlock() | |
line_count = block.blockNumber() | |
painter = QPainter(number_bar) | |
painter.fillRect(event.rect(), self.palette().base()) | |
# Iterate over all visible text blocks in the document. | |
while block.isValid(): | |
line_count += 1 | |
block_top = self.blockBoundingGeometry(block).translated(self.contentOffset()).top() | |
# Check if the position of the block is out side of the visible | |
# area. | |
if not block.isVisible() or block_top >= event.rect().bottom(): | |
break | |
# We want the line number for the selected line to be bold. | |
if line_count == current_line: | |
font = painter.font() | |
font.setBold(True) | |
painter.setFont(font) | |
else: | |
font = painter.font() | |
font.setBold(False) | |
painter.setFont(font) | |
# Draw the line number right justified at the position of the line. | |
paint_rect = QRect(0, int(block_top), number_bar.width(), font_metrics.height()) | |
painter.drawText(paint_rect, Qt.AlignRight, str(line_count)) | |
block = block.next() | |
painter.end() | |
def __init__(self, *args): | |
super().__init__(*args) | |
self.setFrameStyle(QFrame.StyledPanel | QFrame.Sunken) | |
self.edit = self.PlainTextEdit() | |
self.number_bar = self.NumberBar(self.edit) | |
hbox = QHBoxLayout(self) | |
hbox.setSpacing(0) | |
hbox.setContentsMargins(0, 0, 0, 0) | |
hbox.addWidget(self.number_bar) | |
hbox.addWidget(self.edit) | |
self.edit.blockCountChanged.connect(self.number_bar.adjustWidth) | |
self.edit.updateRequest.connect(self.number_bar.updateContents) | |
def getText(self): | |
return str(self.edit.toPlainText()) | |
def setText(self, text): | |
self.edit.setPlainText(text) | |
def isModified(self): | |
return self.edit.document().isModified() | |
def setModified(self, modified): | |
self.edit.document().setModified(modified) | |
def setLineWrapMode(self, mode): | |
self.edit.setLineWrapMode(mode) | |
class MainWindow(QMainWindow): | |
def __init__(self, *args, **kwargs): | |
super().__init__(*args, **kwargs) | |
self.setCentralWidget(NumberedLinesTextEdit()) | |
self.show() | |
if __name__ == "__main__": | |
app = QApplication(sys.argv) | |
main_window = MainWindow() | |
sys.exit(app.exec_()) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment