mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-11-25 04:34:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 11

bpo-36916: asyncio: Swallow unhandled write() exception (GH-13313)

Browse source
This commit is contained in:
Andrew Svetlov 2019-05-14 19:09:44 +03:00 • committed by Victor Stinner
parent c96be811fa
commit f12ba7cd0a
3 changed files with 15 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

11
Lib/asyncio/streams.py
View file

@ -329,6 +329,13 @@ class StreamReaderProtocol(FlowControlMixin, protocols.Protocol):
closed.exception()
def _swallow_unhandled_exception(task):
# Do a trick to suppress unhandled exception
# if stream.write() was used without await and
# stream.drain() was paused and resumed with an exception
task.exception()
class StreamWriter:
"""Wraps a Transport.
@ -393,7 +400,9 @@ class StreamWriter:
# fast path, the stream is not paused
# no need to wait for resume signal
return self._complete_fut
return self._loop.create_task(self.drain())
ret = self._loop.create_task(self.drain())
ret.add_done_callback(_swallow_unhandled_exception)
return ret
def write_eof(self):
return self._transport.write_eof()