LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-03-29 06:14:14 +03:00
Code Issues Packages Projects Releases Wiki Activity

Handle AssertionError when cancelling tasks

Browse Source 
Fixes #1478.
This commit is contained in:
Lonami Exo 2020-06-06 13:54:19 +02:00
parent 20a6d7b26b
commit db3e7656e0
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
telethon/helpers.py
View File

@ -123,6 +123,8 @@ async def _cancel(log, **tasks):
except RuntimeError: except RuntimeError:
# Probably: RuntimeError: await wasn't used with future # Probably: RuntimeError: await wasn't used with future
# #
# See: https://github.com/python/cpython/blob/12d3061c7819a73d891dcce44327410eaf0e1bc2/Lib/asyncio/futures.py#L265
#
# Happens with _asyncio.Task instances (in "Task cancelling" state) # Happens with _asyncio.Task instances (in "Task cancelling" state)
# trying to SIGINT the program right during initial connection, on # trying to SIGINT the program right during initial connection, on
# _recv_loop coroutine (but we're creating its task explicitly with # _recv_loop coroutine (but we're creating its task explicitly with
@ -131,6 +133,12 @@ async def _cancel(log, **tasks):
# Since we're aware of this error there's no point in logging it. # Since we're aware of this error there's no point in logging it.
# *May* be https://bugs.python.org/issue37172 # *May* be https://bugs.python.org/issue37172
pass pass
except AssertionError as e:
# In Python 3.6, the above RuntimeError is an AssertionError
# See https://github.com/python/cpython/blob/7df32f844efed33ca781a016017eab7050263b90/Lib/asyncio/futures.py#L328
if e.args != ("yield from wasn't used with future",):
log.exception('Unhandled exception from %s after cancelling '
'%s (%s)', name, type(task), task)
except Exception: except Exception:
log.exception('Unhandled exception from %s after cancelling ' log.exception('Unhandled exception from %s after cancelling '
'%s (%s)', name, type(task), task) '%s (%s)', name, type(task), task)