hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-06-17 18:46:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Patch __enter__/__exit__ specially in full_sync

Browse Source
This commit is contained in:
Lonami Exo 2018-10-19 18:04:53 +02:00
parent a2bae8374e
commit 39929db2b8
1 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
telethon/full_sync.py
View File

@ -51,7 +51,8 @@ class _SyncGen:
raise StopIteration from None raise StopIteration from None
def _syncify_wrap(t, method_name, loop, thread_ident, syncifier=_sync_result): def _syncify_wrap(t, method_name, loop, thread_ident,
syncifier=_sync_result, rename=None):
method = getattr(t, method_name) method = getattr(t, method_name)
@functools.wraps(method) @functools.wraps(method)
@ -62,11 +63,25 @@ def _syncify_wrap(t, method_name, loop, thread_ident, syncifier=_sync_result):
else syncifier(loop, coro) else syncifier(loop, coro)
) )
setattr(t, method_name, syncified) setattr(t, rename or method_name, syncified)
def _syncify(*types, loop, thread_ident): def _syncify(*types, loop, thread_ident):
for t in types: for t in types:
# __enter__ and __exit__ need special care (VERY dirty hack).
#
# Normally we want them to raise if the loop is running because
# the user can't await there, and they need the async with variant.
#
# However they check if the loop is running to raise, which it is
# with full_sync enabled, so we patch them with the async variant.
if hasattr(t, '__aenter__'):
_syncify_wrap(
t, '__aenter__', loop, thread_ident, rename='__enter__')
_syncify_wrap(
t, '__aexit__', loop, thread_ident, rename='__exit__')
for name in dir(t): for name in dir(t):
if not name.startswith('_') or name == '__call__': if not name.startswith('_') or name == '__call__':
meth = getattr(t, name) meth = getattr(t, name)