Source code for labgrid.util.expect

import pexpect

from ..protocol import ConsoleProtocol
from .exceptions import NoValidDriverError

[docs]class PtxExpect(pexpect.spawn): """labgrid Wrapper of the pexpect module. This class provides pexpect functionality for the ConsoleProtocol classes. driver: ConsoleProtocol object to be passed in """ def __init__(self, driver, logfile=None, timeout=30, cwd=None): self.driver = driver self.logfile = logfile self.linesep = b"\n" pexpect.spawn.__init__( self, None, timeout=timeout, maxread=1, cwd=cwd, logfile=self.logfile, )
[docs] def send(self, s): "Write to underlying transport, return number of bytes written" s = self._coerce_send_string(s) self._log(s, 'send') b = s return self.driver.write(b)
[docs] def read_nonblocking(self, size=1, timeout=-1): """ Pexpects needs a nonblocking read function, simply use pyserial with a timeout of 0""" assert timeout is not None if timeout == -1: timeout = self.timeout return, timeout=timeout)