| Index: third_party/pexpect/pexpect/popen_spawn.py
|
| diff --git a/third_party/pexpect/pexpect/popen_spawn.py b/third_party/pexpect/pexpect/popen_spawn.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..680dd8a273d4ac76b2793665c5b47d1f2fd3587a
|
| --- /dev/null
|
| +++ b/third_party/pexpect/pexpect/popen_spawn.py
|
| @@ -0,0 +1,179 @@
|
| +"""Provides an interface like pexpect.spawn interface using subprocess.Popen
|
| +"""
|
| +import os
|
| +import threading
|
| +import subprocess
|
| +import sys
|
| +import time
|
| +import signal
|
| +import shlex
|
| +
|
| +try:
|
| + from queue import Queue, Empty # Python 3
|
| +except ImportError:
|
| + from Queue import Queue, Empty # Python 2
|
| +
|
| +from .spawnbase import SpawnBase, PY3
|
| +from .exceptions import EOF
|
| +
|
| +class PopenSpawn(SpawnBase):
|
| + if PY3:
|
| + crlf = '\n'.encode('ascii')
|
| + else:
|
| + crlf = '\n'
|
| +
|
| + def __init__(self, cmd, timeout=30, maxread=2000, searchwindowsize=None,
|
| + logfile=None, cwd=None, env=None, encoding=None,
|
| + codec_errors='strict'):
|
| + super(PopenSpawn, self).__init__(timeout=timeout, maxread=maxread,
|
| + searchwindowsize=searchwindowsize, logfile=logfile,
|
| + encoding=encoding, codec_errors=codec_errors)
|
| +
|
| + kwargs = dict(bufsize=0, stdin=subprocess.PIPE,
|
| + stderr=subprocess.STDOUT, stdout=subprocess.PIPE,
|
| + cwd=cwd, env=env)
|
| +
|
| + if sys.platform == 'win32':
|
| + startupinfo = subprocess.STARTUPINFO()
|
| + startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
|
| + kwargs['startupinfo'] = startupinfo
|
| + kwargs['creationflags'] = subprocess.CREATE_NEW_PROCESS_GROUP
|
| +
|
| + if not isinstance(cmd, (list, tuple)):
|
| + cmd = shlex.split(cmd)
|
| +
|
| + self.proc = subprocess.Popen(cmd, **kwargs)
|
| + self.closed = False
|
| + self._buf = self.string_type()
|
| +
|
| + self._read_queue = Queue()
|
| + self._read_thread = threading.Thread(target=self._read_incoming)
|
| + self._read_thread.setDaemon(True)
|
| + self._read_thread.start()
|
| +
|
| + _read_reached_eof = False
|
| +
|
| + def read_nonblocking(self, size, timeout):
|
| + buf = self._buf
|
| + if self._read_reached_eof:
|
| + # We have already finished reading. Use up any buffered data,
|
| + # then raise EOF
|
| + if buf:
|
| + self._buf = buf[size:]
|
| + return buf[:size]
|
| + else:
|
| + self.flag_eof = True
|
| + raise EOF('End Of File (EOF).')
|
| +
|
| + if timeout == -1:
|
| + timeout = self.timeout
|
| + elif timeout is None:
|
| + timeout = 1e6
|
| +
|
| + t0 = time.time()
|
| + while (time.time() - t0) < timeout and size and len(buf) < size:
|
| + try:
|
| + incoming = self._read_queue.get_nowait()
|
| + except Empty:
|
| + break
|
| + else:
|
| + if incoming is None:
|
| + self._read_reached_eof = True
|
| + break
|
| +
|
| + buf += self._decoder.decode(incoming, final=False)
|
| +
|
| + r, self._buf = buf[:size], buf[size:]
|
| +
|
| + self._log(r, 'read')
|
| + return r
|
| +
|
| + def _read_incoming(self):
|
| + """Run in a thread to move output from a pipe to a queue."""
|
| + fileno = self.proc.stdout.fileno()
|
| + while 1:
|
| + buf = b''
|
| + try:
|
| + buf = os.read(fileno, 1024)
|
| + except OSError as e:
|
| + self._log(e, 'read')
|
| +
|
| + if not buf:
|
| + # This indicates we have reached EOF
|
| + self._read_queue.put(None)
|
| + return
|
| +
|
| + self._read_queue.put(buf)
|
| +
|
| + def write(self, s):
|
| + '''This is similar to send() except that there is no return value.
|
| + '''
|
| + self.send(s)
|
| +
|
| + def writelines(self, sequence):
|
| + '''This calls write() for each element in the sequence.
|
| +
|
| + The sequence can be any iterable object producing strings, typically a
|
| + list of strings. This does not add line separators. There is no return
|
| + value.
|
| + '''
|
| + for s in sequence:
|
| + self.send(s)
|
| +
|
| + def send(self, s):
|
| + '''Send data to the subprocess' stdin.
|
| +
|
| + Returns the number of bytes written.
|
| + '''
|
| + s = self._coerce_send_string(s)
|
| + self._log(s, 'send')
|
| +
|
| + b = self._encoder.encode(s, final=False)
|
| + if PY3:
|
| + return self.proc.stdin.write(b)
|
| + else:
|
| + # On Python 2, .write() returns None, so we return the length of
|
| + # bytes written ourselves. This assumes they all got written.
|
| + self.proc.stdin.write(b)
|
| + return len(b)
|
| +
|
| + def sendline(self, s=''):
|
| + '''Wraps send(), sending string ``s`` to child process, with os.linesep
|
| + automatically appended. Returns number of bytes written. '''
|
| +
|
| + n = self.send(s)
|
| + return n + self.send(self.linesep)
|
| +
|
| + def wait(self):
|
| + '''Wait for the subprocess to finish.
|
| +
|
| + Returns the exit code.
|
| + '''
|
| + status = self.proc.wait()
|
| + if status >= 0:
|
| + self.exitstatus = status
|
| + self.signalstatus = None
|
| + else:
|
| + self.exitstatus = None
|
| + self.signalstatus = -status
|
| + self.terminated = True
|
| + return status
|
| +
|
| + def kill(self, sig):
|
| + '''Sends a Unix signal to the subprocess.
|
| +
|
| + Use constants from the :mod:`signal` module to specify which signal.
|
| + '''
|
| + if sys.platform == 'win32':
|
| + if sig in [signal.SIGINT, signal.CTRL_C_EVENT]:
|
| + sig = signal.CTRL_C_EVENT
|
| + elif sig in [signal.SIGBREAK, signal.CTRL_BREAK_EVENT]:
|
| + sig = signal.CTRL_BREAK_EVENT
|
| + else:
|
| + sig = signal.SIGTERM
|
| +
|
| + os.kill(self.proc.pid, sig)
|
| +
|
| + def sendeof(self):
|
| + '''Closes the stdin pipe from the writing end.'''
|
| + self.proc.stdin.close()
|
|
|