74 lines
2.2 KiB
Python
74 lines
2.2 KiB
Python
from PyQt5.QtGui import QTextCursor, QFont
|
|
from PyQt5.QtWidgets import (
|
|
QPlainTextEdit,
|
|
QWidget,
|
|
QPushButton,
|
|
QFileDialog,
|
|
QVBoxLayout,
|
|
)
|
|
|
|
|
|
class ConsoleWindow(QWidget):
|
|
def __init__(self):
|
|
super(ConsoleWindow, self).__init__()
|
|
self.layout = QVBoxLayout()
|
|
self.setWindowTitle("Rare Console")
|
|
self.setGeometry(0, 0, 600, 400)
|
|
|
|
self.console = Console()
|
|
self.layout.addWidget(self.console)
|
|
|
|
self.save_button = QPushButton(self.tr("Save output to file"))
|
|
self.layout.addWidget(self.save_button)
|
|
self.save_button.clicked.connect(self.save)
|
|
|
|
self.clear_button = QPushButton(self.tr("Clear"))
|
|
self.layout.addWidget(self.clear_button)
|
|
self.clear_button.clicked.connect(self.console.clear)
|
|
|
|
self.setLayout(self.layout)
|
|
|
|
def save(self):
|
|
file, ok = QFileDialog.getSaveFileName(
|
|
self, "Save output", "", "Log Files (*.log);;All Files (*)"
|
|
)
|
|
if ok:
|
|
if "." not in file:
|
|
file += ".log"
|
|
with open(file, "w") as f:
|
|
f.write(self.console.toPlainText())
|
|
f.close()
|
|
self.save_button.setText(self.tr("Saved"))
|
|
|
|
def log(self, text: str, end: str = "\n"):
|
|
self.console.log(text + end)
|
|
|
|
def error(self, text, end: str = "\n"):
|
|
self.console.error(text + end)
|
|
|
|
|
|
class Console(QPlainTextEdit):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.setReadOnly(True)
|
|
self.setFont(QFont("monospace"))
|
|
self._cursor_output = self.textCursor()
|
|
|
|
def log(self, text):
|
|
html = f"<p style=\"color:#999;white-space:pre\">{text}</p>"
|
|
self._cursor_output.insertHtml(html)
|
|
self.scroll_to_last_line()
|
|
|
|
def error(self, text):
|
|
html = f"<p style=\"color:#eee;white-space:pre\">{text}</p>"
|
|
self._cursor_output.insertHtml(html)
|
|
self.scroll_to_last_line()
|
|
|
|
def scroll_to_last_line(self):
|
|
cursor = self.textCursor()
|
|
cursor.movePosition(QTextCursor.End)
|
|
cursor.movePosition(
|
|
QTextCursor.Up if cursor.atBlockStart() else QTextCursor.StartOfLine
|
|
)
|
|
self.setTextCursor(cursor)
|