simple_interact.py

# Copyright 2000-2010 Michael Hudson-Doyle <micahel@gmail.com>
# Armin Rigo
#
# All Rights Reserved
#
#
# Permission to use, copy, modify, and distribute this software and
# its documentation for any purpose is hereby granted without fee,
# provided that the above copyright notice appear in all copies and
# that both that copyright notice and this permission notice appear in
# supporting documentation.
#
# THE AUTHOR MICHAEL HUDSON DISCLAIMS ALL WARRANTIES WITH REGARD TO
# THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
# AND FITNESS, IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL,
# INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER
# RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF
# CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
# CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.

"""This is an alternative to python_reader which tries to emulate
the CPython prompt as closely as possible, with the exception of
allowing multiline input and multiline history entries.
"""

from __future__ importannotations

import_sitebuiltins
importfunctools
importos
importsys
importcode

from .readlineimport_get_reader, multiline_input

TYPE_CHECKING=False

ifTYPE_CHECKING:
fromtypingimportAny


_error: tuple[type[Exception], ...] |type[Exception]
try:
from .unix_consoleimport_error
exceptModuleNotFoundError:
from .windows_consoleimport_error

defcheck() ->str:
"""Returns the error message if there is a problem initializing the state."""
try:
_get_reader()
except_errorase:
ifterm:=os.environ.get("TERM", ""):
term=f"; TERM={term}"
returnstr(str(e) orrepr(e) or"unknown error") +term
return""


def_strip_final_indent(text: str) ->str:
# kill spaces and tabs at the end, but only if they follow '\n'.
# meant to remove the auto-indentation only (although it would of
# course also remove explicitly-added indentation).
short=text.rstrip(" \t")
n=len(short)
ifn>0andtext[n-1] =="\n":
returnshort
returntext


def_clear_screen():
reader=_get_reader()
reader.scheduled_commands.append("clear_screen")


REPL_COMMANDS= {
"exit": _sitebuiltins.Quitter('exit', ''),
"quit": _sitebuiltins.Quitter('quit' ,''),
"copyright": _sitebuiltins._Printer('copyright', sys.copyright),
"help": _sitebuiltins._Helper(),
"clear": _clear_screen,
"\x1a": _sitebuiltins.Quitter('\x1a', ''),
}


def_more_lines(console: code.InteractiveConsole, unicodetext: str) ->bool:
# ooh, look at the hack:
src=_strip_final_indent(unicodetext)
try:
code=console.compile(src, "<stdin>", "single")
except (OverflowError, SyntaxError, ValueError):
lines=src.splitlines(keepends=True)
iflen(lines) ==1:
returnFalse

last_line=lines[-1]
was_indented=last_line.startswith((" ", "\t"))
not_empty=last_line.strip() !=""
incomplete=notlast_line.endswith("\n")
return (was_indentedornot_empty) andincomplete
else:
returncodeisNone


defrun_multiline_interactive_console(
console: code.InteractiveConsole,
*,
future_flags: int=0,
) ->None:
from .readlineimport_setup
_setup(console.locals)
iffuture_flags:
console.compile.compiler.flags|=future_flags

more_lines=functools.partial(_more_lines, console)
input_n=0

defmaybe_run_command(statement: str) ->bool:
statement=statement.strip()
ifstatementinconsole.localsorstatementnotinREPL_COMMANDS:
returnFalse

reader=_get_reader()
reader.history.pop() # skip internal commands in history
command=REPL_COMMANDS[statement]
ifcallable(command):
# Make sure that history does not change because of commands
withreader.suspend_history():
command()
returnTrue
returnFalse

while1:
try:
try:
sys.stdout.flush()
exceptException:
pass

ps1=getattr(sys, "ps1", ">>> ")
ps2=getattr(sys, "ps2", "... ")
try:
statement=multiline_input(more_lines, ps1, ps2)
exceptEOFError:
break

ifmaybe_run_command(statement):
continue

input_name=f"<python-input-{input_n}>"
more=console.push(_strip_final_indent(statement), filename=input_name, _symbol="single") # type: ignore[call-arg]
assertnotmore
input_n+=1
exceptKeyboardInterrupt:
r=_get_reader()
ifr.input_transisr.isearch_trans:
r.do_cmd(("isearch-end", [""]))
r.pos=len(r.get_unicode())
r.dirty=True
r.refresh()
r.in_bracketed_paste=False
console.write("\nKeyboardInterrupt\n")
console.resetbuffer()
exceptMemoryError:
console.write("\nMemoryError\n")
console.resetbuffer()