asyncio: Add set_protocol / get_protocol methods to Transports

This commit is contained in:
Yury Selivanov 2016-09-11 21:11:02 -04:00
parent 06e18a7c24
commit a05a6ef1ca
7 changed files with 45 additions and 0 deletions

View file

@ -374,6 +374,12 @@ class _UnixReadPipeTransport(transports.ReadTransport):
def resume_reading(self):
self._loop.add_reader(self._fileno, self._read_ready)
def set_protocol(self, protocol):
self._protocol = protocol
def get_protocol(self):
return self._protocol
def is_closing(self):
return self._closing
@ -570,6 +576,12 @@ class _UnixWritePipeTransport(transports._FlowControlMixin,
self._loop.remove_reader(self._fileno)
self._loop.call_soon(self._call_connection_lost, None)
def set_protocol(self, protocol):
self._protocol = protocol
def get_protocol(self):
return self._protocol
def is_closing(self):
return self._closing