Admin

name	type	size	perms	modified	actions
[parent directory]
__pycache__/	dir	-	0755	2024-04-23 15:28:34
__init__.py	file	4,089 B	0644	2024-01-30 13:43:49	edit
_async.py	file	907 B	0644	2024-01-30 13:43:49	edit
_async_pre_await.py	file	3,465 B	0644	2024-01-30 13:43:49	edit
_async_w_await.py	file	3,802 B	0644	2024-01-30 13:43:49	edit
ANSI.py	file	12,177 B	0644	2024-01-30 13:43:49	edit
bashrc.sh	file	419 B	0644	2024-01-30 13:43:49	edit
exceptions.py	file	1,068 B	0644	2024-01-30 13:43:49	edit
expect.py	file	13,827 B	0644	2024-01-30 13:43:49	edit
fdpexpect.py	file	5,991 B	0644	2024-01-30 13:43:49	edit
FSM.py	file	13,419 B	0644	2024-01-30 13:43:49	edit
popen_spawn.py	file	6,159 B	0644	2024-01-30 13:43:49	edit
pty_spawn.py	file	37,382 B	0644	2024-01-30 13:43:49	edit
pxssh.py	file	24,445 B	0644	2024-01-30 13:43:49	edit
replwrap.py	file	5,951 B	0644	2024-01-30 13:43:49	edit
run.py	file	6,629 B	0644	2024-01-30 13:43:49	edit
screen.py	file	13,704 B	0644	2024-01-30 13:43:49	edit
socket_pexpect.py	file	4,814 B	0644	2024-01-30 13:43:49	edit
spawnbase.py	file	21,685 B	0644	2024-01-30 13:43:49	edit
utils.py	file	6,019 B	0644	2024-01-30 13:43:49	edit

editing: _async_pre_await.py

"""Implementation of coroutines without using ``async def``/``await`` keywords.

``@asyncio.coroutine`` and ``yield from`` are  used here instead.
"""
import asyncio
import errno
import signal

from pexpect import EOF

@asyncio.coroutine
def expect_async(expecter, timeout=None):
    # First process data that was previously read - if it maches, we don't need
    # async stuff.
    idx = expecter.existing_data()
    if idx is not None:
        return idx
    if not expecter.spawn.async_pw_transport:
        pw = PatternWaiter()
        pw.set_expecter(expecter)
        transport, pw = yield from asyncio.get_event_loop().connect_read_pipe(
            lambda: pw, expecter.spawn
        )
        expecter.spawn.async_pw_transport = pw, transport
    else:
        pw, transport = expecter.spawn.async_pw_transport
        pw.set_expecter(expecter)
        transport.resume_reading()
    try:
        return (yield from asyncio.wait_for(pw.fut, timeout))
    except asyncio.TimeoutError as e:
        transport.pause_reading()
        return expecter.timeout(e)

@asyncio.coroutine
def repl_run_command_async(repl, cmdlines, timeout=-1):
    res = []
    repl.child.sendline(cmdlines[0])
    for line in cmdlines[1:]:
        yield from repl._expect_prompt(timeout=timeout, async_=True)
        res.append(repl.child.before)
        repl.child.sendline(line)

# Command was fully submitted, now wait for the next prompt
    prompt_idx = yield from repl._expect_prompt(timeout=timeout, async_=True)
    if prompt_idx == 1:
        # We got the continuation prompt - command was incomplete
        repl.child.kill(signal.SIGINT)
        yield from repl._expect_prompt(timeout=1, async_=True)
        raise ValueError("Continuation prompt found - input was incomplete:")
    return "".join(res + [repl.child.before])

class PatternWaiter(asyncio.Protocol):
    transport = None

def set_expecter(self, expecter):
        self.expecter = expecter
        self.fut = asyncio.Future()

def found(self, result):
        if not self.fut.done():
            self.fut.set_result(result)
            self.transport.pause_reading()

def error(self, exc):
        if not self.fut.done():
            self.fut.set_exception(exc)
            self.transport.pause_reading()

def connection_made(self, transport):
        self.transport = transport

def data_received(self, data):
        spawn = self.expecter.spawn
        s = spawn._decoder.decode(data)
        spawn._log(s, "read")

if self.fut.done():
            spawn._before.write(s)
            spawn._buffer.write(s)
            return

try:
            index = self.expecter.new_data(s)
            if index is not None:
                # Found a match
                self.found(index)
        except Exception as e:
            self.expecter.errored()
            self.error(e)

def eof_received(self):
        # N.B. If this gets called, async will close the pipe (the spawn object)
        # for us
        try:
            self.expecter.spawn.flag_eof = True
            index = self.expecter.eof()
        except EOF as e:
            self.error(e)
        else:
            self.found(index)

def connection_lost(self, exc):
        if isinstance(exc, OSError) and exc.errno == errno.EIO:
            # We may get here without eof_received being called, e.g on Linux
            self.eof_received()
        elif exc is not None:
            self.error(exc)

cancel