Beta Shell v2.0

File Manager

~ /usr/local/python-3.12/lib/python3.12/test/support

Name	Size	Perms	Modified	Actions
[ .. / .. ]
__init__.py	78.13 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
__pycache__	-	drwxr-xr-x	2024-04-12 10:29:53	Del
_hypothesis_stubs	-	drwxr-xr-x	2024-04-12 10:29:52	Del
ast_helper.py	1.79 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
asynchat.py	11.33 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
asyncore.py	19.9 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
bytecode_helper.py	4.88 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
hashlib_helper.py	1.86 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
hypothesis_helper.py	1.35 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
import_helper.py	10.48 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
interpreters.py	5.67 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
logging_helper.py	916 B	-rw-r--r--	2024-04-12 10:29:39	Edit Del
os_helper.py	23.72 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
pty_helper.py	2.98 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
script_helper.py	11.45 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
smtpd.py	30.01 KB	-rwxr-xr-x	2024-04-12 10:29:39	Edit Del
socket_helper.py	13.47 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
testcase.py	1.02 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
threading_helper.py	7.86 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
warnings_helper.py	6.69 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del

Editing: pty_helper.py (2.98 KB)

"""
Helper to run a script in a pseudo-terminal.
"""
import os
import selectors
import subprocess
import sys
from contextlib import ExitStack
from errno import EIO

from test.support.import_helper import import_module

def run_pty(script, input=b"dummy input\r", env=None):
    pty = import_module('pty')
    output = bytearray()
    [master, slave] = pty.openpty()
    args = (sys.executable, '-c', script)
    proc = subprocess.Popen(args, stdin=slave, stdout=slave, stderr=slave, env=env)
    os.close(slave)
    with ExitStack() as cleanup:
        cleanup.enter_context(proc)
        def terminate(proc):
            try:
                proc.terminate()
            except ProcessLookupError:
                # Workaround for Open/Net BSD bug (Issue 16762)
                pass
        cleanup.callback(terminate, proc)
        cleanup.callback(os.close, master)
        # Avoid using DefaultSelector and PollSelector. Kqueue() does not
        # work with pseudo-terminals on OS X < 10.9 (Issue 20365) and Open
        # BSD (Issue 20667). Poll() does not work with OS X 10.6 or 10.4
        # either (Issue 20472). Hopefully the file descriptor is low enough
        # to use with select().
        sel = cleanup.enter_context(selectors.SelectSelector())
        sel.register(master, selectors.EVENT_READ | selectors.EVENT_WRITE)
        os.set_blocking(master, False)
        while True:
            for [_, events] in sel.select():
                if events & selectors.EVENT_READ:
                    try:
                        chunk = os.read(master, 0x10000)
                    except OSError as err:
                        # Linux raises EIO when slave is closed (Issue 5380)
                        if err.errno != EIO:
                            raise
                        chunk = b""
                    if not chunk:
                        return output
                    output.extend(chunk)
                if events & selectors.EVENT_WRITE:
                    try:
                        input = input[os.write(master, input):]
                    except OSError as err:
                        # Apparently EIO means the slave was closed
                        if err.errno != EIO:
                            raise
                        input = b""  # Stop writing
                    if not input:
                        sel.modify(master, selectors.EVENT_READ)


######################################################################
## Fake stdin (for testing interactive debugging)
######################################################################

class FakeInput:
    """
    A fake input stream for pdb's interactive debugger.  Whenever a
    line is read, print it (to simulate the user typing it), and then
    return it.  The set of lines to return is specified in the
    constructor; they should not have trailing newlines.
    """
    def __init__(self, lines):
        self.lines = lines

    def readline(self):
        line = self.lines.pop(0)
        print(line)
        return line + '\n'