Like StreamReaderProtocol, but for a subprocess.
| 15 | |
| 16 | |
| 17 | class SubprocessStreamProtocol(streams.FlowControlMixin, |
| 18 | protocols.SubprocessProtocol): |
| 19 | """Like StreamReaderProtocol, but for a subprocess.""" |
| 20 | |
| 21 | def __init__(self, limit, loop): |
| 22 | super().__init__(loop=loop) |
| 23 | self._limit = limit |
| 24 | self.stdin = self.stdout = self.stderr = None |
| 25 | self._transport = None |
| 26 | self._process_exited = False |
| 27 | self._pipe_fds = [] |
| 28 | self._stdin_closed = self._loop.create_future() |
| 29 | |
| 30 | def __repr__(self): |
| 31 | info = [self.__class__.__name__] |
| 32 | if self.stdin is not None: |
| 33 | info.append(f'stdin={self.stdin!r}') |
| 34 | if self.stdout is not None: |
| 35 | info.append(f'stdout={self.stdout!r}') |
| 36 | if self.stderr is not None: |
| 37 | info.append(f'stderr={self.stderr!r}') |
| 38 | return '<{}>'.format(' '.join(info)) |
| 39 | |
| 40 | def connection_made(self, transport): |
| 41 | self._transport = transport |
| 42 | |
| 43 | stdout_transport = transport.get_pipe_transport(1) |
| 44 | if stdout_transport is not None: |
| 45 | self.stdout = streams.StreamReader(limit=self._limit, |
| 46 | loop=self._loop) |
| 47 | self.stdout.set_transport(stdout_transport) |
| 48 | self._pipe_fds.append(1) |
| 49 | |
| 50 | stderr_transport = transport.get_pipe_transport(2) |
| 51 | if stderr_transport is not None: |
| 52 | self.stderr = streams.StreamReader(limit=self._limit, |
| 53 | loop=self._loop) |
| 54 | self.stderr.set_transport(stderr_transport) |
| 55 | self._pipe_fds.append(2) |
| 56 | |
| 57 | stdin_transport = transport.get_pipe_transport(0) |
| 58 | if stdin_transport is not None: |
| 59 | self.stdin = streams.StreamWriter(stdin_transport, |
| 60 | protocol=self, |
| 61 | reader=None, |
| 62 | loop=self._loop) |
| 63 | |
| 64 | def pipe_data_received(self, fd, data): |
| 65 | if fd == 1: |
| 66 | reader = self.stdout |
| 67 | elif fd == 2: |
| 68 | reader = self.stderr |
| 69 | else: |
| 70 | reader = None |
| 71 | if reader is not None: |
| 72 | reader.feed_data(data) |
| 73 | |
| 74 | def pipe_connection_lost(self, fd, exc): |
no outgoing calls
no test coverage detected
searching dependent graphs…