console.py

Go to the documentation of this file.

import io
import sys
import inspect
import traceback
import contextlib
 
from PyQt5.QtWidgets import *
 
 
class Console(QWidget):
    def __init__(self, *a):
        super().__init__(*a)
        self.context_globalcontext_global = {}
        self.context_localcontext_local = {}
 
        layout = QVBoxLayout()
        self.setLayout(layout)
 
        self.lineslines = QPlainTextEdit()
        self.lineslines.setReadOnly(True)
        layout.addWidget(self.lineslines)
 
        self.inputinput = QLineEdit()
        self.inputinput.returnPressed.connect(self.on_inputon_input)
        layout.addWidget(self.inputinput)
 
    def set_font(self, font):
        self.lineslines.setFont(font)
        self.inputinput.setFont(font)
 
    def define(self, name, value):
        self.context_globalcontext_global[name] = value
 
    def on_input(self):
        self.evaleval(self.inputinput.text())
        self.inputinput.clear()
 
    def eval(self, line):
        self.lineslines.appendPlainText(">>> " + line)
 
        try:
            try:
                expression = True
                code = compile(line, "<console>", "eval")
            except SyntaxError:
                expression = False
                code = compile(line, "<console>", "single")
 
            sterrout = io.StringIO()
            with contextlib.redirect_stderr(sterrout):
                with contextlib.redirect_stdout(sterrout):
                    if expression:
                        value = eval(code, self.context_globalcontext_global, self.context_localcontext_local)
                    else:
                        value = None
                        exec(code, self.context_globalcontext_global, self.context_localcontext_local)
 
            stdstreams = sterrout.getvalue()
            if stdstreams:
                if stdstreams.endswith("\n"):
                    stdstreams = stdstreams[:-1]
                self.lineslines.appendPlainText(stdstreams)
            if value is not None:
                self.lineslines.appendPlainText(repr(value))
        except Exception:
            etype, value, tb = sys.exc_info()
            # Find how many frames to print to hide the current frame and above
            parent = inspect.currentframe().f_back
            i = 0
            for frame, ln in traceback.walk_tb(tb):
                i += 1
                if frame.f_back == parent:
                    i = 0
 
            file = io.StringIO()
            traceback.print_exception(etype, value, tb, limit=-i, file=file)
            self.lineslines.appendPlainText(file.getvalue())
 
        self.lineslines.verticalScrollBar().setValue(self.lineslines.verticalScrollBar().maximum())