Class Interpreter

bpython/repl.py:103–205

Source code interpreter for use in bpython.

class Interpreter(code.InteractiveInterpreter):
    """Source code interpreter for use in bpython."""

    bpython_input_re = LazyReCompile(r"<bpython-input-\d+>")

    def __init__(
        self,
        locals: dict[str, Any] | None = None,
    ) -> None:
        """Constructor.

        The optional 'locals' argument specifies the dictionary in which code
        will be executed; it defaults to a newly created dictionary with key
        "__name__" set to "__main__".

        The syntaxerror callback can be set at any time and will be called
        on a caught syntax error. The purpose for this in bpython is so that
        the repl can be instantiated after the interpreter (which it
        necessarily must be with the current factoring) and then an exception
        callback can be added to the Interpreter instance afterwards - more
        specifically, this is so that autoindentation does not occur after a
        traceback.
        """

        self.syntaxerror_callback: Callable | None = None

        if locals is None:
            # instead of messing with sys.modules, we should modify sys.modules
            # in the interpreter instance
            sys.modules["__main__"] = main_mod = ModuleType("__main__")
            locals = main_mod.__dict__

        super().__init__(locals)
        self.timer = RuntimeTimer()

    def runsource(
        self,
        source: str,
        filename: str | None = None,
        symbol: str = "single",
    ) -> bool:
        """Execute Python code.

        source, filename and symbol are passed on to
        code.InteractiveInterpreter.runsource."""

        if filename is None:
            filename = filename_for_console_input(source)
        with self.timer:
            return super().runsource(source, filename, symbol)

    def showsyntaxerror(self, filename: str | None = None, **kwargs) -> None:
        """Override the regular handler, the code&#x27;s copied and pasted from
        code.py, as per showtraceback, but with the syntaxerror callback called
        and the text in a pretty colour."""
        if self.syntaxerror_callback is not None:
            self.syntaxerror_callback()