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

Fix-up new sync __enter__ not handling the client directly

Browse Source
This commit is contained in:
Lonami Exo 2019-04-13 10:55:51 +02:00
parent 9090ede5db
commit 20b8250037
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
telethon/helpers.py
View File

@ -108,17 +108,27 @@ def _sync_enter(self):
Helps to cut boilerplate on async context
managers that offer synchronous variants.
"""
if self._client.loop.is_running():
if hasattr(self, 'loop'):
loop = self.loop
else:
loop = self._client.loop
if loop.is_running():
raise RuntimeError(
'You must use "async with" if the event loop '
'is running (i.e. you are inside an "async def")'
)
return self._client.loop.run_until_complete(self.__aenter__())
return loop.run_until_complete(self.__aenter__())
def _sync_exit(self, *args):
return self._client.loop.run_until_complete(self.__aexit__(*args))
if hasattr(self, 'loop'):
loop = self.loop
else:
loop = self._client.loop
return loop.run_until_complete(self.__aexit__(*args))
# endregion