Skip to content

Commit

Permalink
pythongh-111201: A new Python REPL
Browse files Browse the repository at this point in the history
  • Loading branch information
pablogsal committed Oct 31, 2023
1 parent 2445673 commit 17c271e
Show file tree
Hide file tree
Showing 20 changed files with 3,774 additions and 9 deletions.
19 changes: 19 additions & 0 deletions Lib/_pyrepl/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
# Copyright 2000-2008 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.
78 changes: 78 additions & 0 deletions Lib/_pyrepl/__main__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,78 @@
import os
import sys

irc_header = "And now for something completely different"


def interactive_console(mainmodule=None, quiet=False):
# set sys.{ps1,ps2} just before invoking the interactive interpreter. This
# mimics what CPython does in pythonrun.c
if not hasattr(sys, "ps1"):
sys.ps1 = ">>> "
if not hasattr(sys, "ps2"):
sys.ps2 = "... "
#
run_interactive = run_simple_interactive_console
try:
if not os.isatty(sys.stdin.fileno()):
# Bail out if stdin is not tty-like, as pyrepl wouldn't be happy
# For example, with:
# subprocess.Popen(['pypy', '-i'], stdin=subprocess.PIPE)
raise ImportError
from .simple_interact import check

if not check():
raise ImportError
from .simple_interact import run_multiline_interactive_console

run_interactive = run_multiline_interactive_console
# except ImportError:
# pass
except SyntaxError:
print("Warning: 'import pyrepl' failed with SyntaxError")
run_interactive(mainmodule)


def run_simple_interactive_console(mainmodule):
import code

if mainmodule is None:
import __main__ as mainmodule
console = code.InteractiveConsole(mainmodule.__dict__, filename="<stdin>")
# some parts of code.py are copied here because it was impossible
# to start an interactive console without printing at least one line
# of banner. This was fixed in 3.4; but then from 3.6 it prints a
# line when exiting. This can be disabled too---by passing an argument
# that doesn't exist in <= 3.5. So, too much mess: just copy the code.
more = 0
while 1:
try:
if more:
prompt = getattr(sys, "ps2", "... ")
else:
prompt = getattr(sys, "ps1", ">>> ")
try:
line = input(prompt)
except EOFError:
console.write("\n")
break
else:
more = console.push(line)
except KeyboardInterrupt:
console.write("\nKeyboardInterrupt\n")
console.resetbuffer()
more = 0


# ____________________________________________________________

if __name__ == "__main__": # for testing
if os.getenv("PYTHONSTARTUP"):
exec(
compile(
open(os.getenv("PYTHONSTARTUP")).read(),
os.getenv("PYTHONSTARTUP"),
"exec",
)
)
interactive_console()
68 changes: 68 additions & 0 deletions Lib/_pyrepl/_minimal_curses.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,68 @@
"""Minimal '_curses' module, the low-level interface for curses module
which is not meant to be used directly.
Based on ctypes. It's too incomplete to be really called '_curses', so
to use it, you have to import it and stick it in sys.modules['_curses']
manually.
Note that there is also a built-in module _minimal_curses which will
hide this one if compiled in.
"""

import ctypes
import ctypes.util


class error(Exception):
pass


def _find_clib():
trylibs = ["ncursesw", "ncurses", "curses"]

for lib in trylibs:
path = ctypes.util.find_library(lib)
if path:
return path
raise ModuleNotFoundError("curses library not found", name="_minimal_curses")


_clibpath = _find_clib()
clib = ctypes.cdll.LoadLibrary(_clibpath)

clib.setupterm.argtypes = [ctypes.c_char_p, ctypes.c_int, ctypes.POINTER(ctypes.c_int)]
clib.setupterm.restype = ctypes.c_int

clib.tigetstr.argtypes = [ctypes.c_char_p]
clib.tigetstr.restype = ctypes.POINTER(ctypes.c_char)

clib.tparm.argtypes = [ctypes.c_char_p] + 9 * [ctypes.c_int]
clib.tparm.restype = ctypes.c_char_p

OK = 0
ERR = -1

# ____________________________________________________________


def setupterm(termstr, fd):
err = ctypes.c_int(0)
result = clib.setupterm(termstr, fd, ctypes.byref(err))
if result == ERR:
raise error("setupterm() failed (err=%d)" % err.value)


def tigetstr(cap):
if not isinstance(cap, bytes):
cap = cap.encode("ascii")
result = clib.tigetstr(cap)
if ctypes.cast(result, ctypes.c_void_p).value == ERR:
return None
return ctypes.cast(result, ctypes.c_char_p).value


def tparm(str, i1=0, i2=0, i3=0, i4=0, i5=0, i6=0, i7=0, i8=0, i9=0):
result = clib.tparm(str, i1, i2, i3, i4, i5, i6, i7, i8, i9)
if result is None:
raise error("tparm() returned NULL")
return result
Loading

0 comments on commit 17c271e

Please sign in to comment.