import asyncio import errno 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. previously_read = expecter.spawn.buffer expecter.spawn.buffer = expecter.spawn.string_type() idx = expecter.new_data(previously_read) if idx is not None: return idx transport, pw = yield from asyncio.get_event_loop()\ .connect_read_pipe(lambda: PatternWaiter(expecter), expecter.spawn) try: return (yield from asyncio.wait_for(pw.fut, timeout)) except asyncio.TimeoutError as e: transport.pause_reading() return expecter.timeout(e) class PatternWaiter(asyncio.Protocol): transport = None def __init__(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.buffer += 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)
Name | Type | Size | Permission | Actions |
---|---|---|---|---|
__pycache__ | Folder | 0755 |
|
|
ANSI.py | File | 11.89 KB | 0644 |
|
FSM.py | File | 13.1 KB | 0644 |
|
__init__.py | File | 3.81 KB | 0644 |
|
async.py | File | 2.28 KB | 0644 |
|
bashrc.sh | File | 85 B | 0644 |
|
exceptions.py | File | 1.04 KB | 0644 |
|
expect.py | File | 10.45 KB | 0644 |
|
fdpexpect.py | File | 5.49 KB | 0644 |
|
popen_spawn.py | File | 5.42 KB | 0644 |
|
pty_spawn.py | File | 34.04 KB | 0644 |
|
pxssh.py | File | 18.43 KB | 0644 |
|
replwrap.py | File | 5.04 KB | 0644 |
|
run.py | File | 6.48 KB | 0644 |
|
screen.py | File | 13.39 KB | 0644 |
|
spawnbase.py | File | 19.62 KB | 0644 |
|
utils.py | File | 4.73 KB | 0644 |
|