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

Reduce __enter__/__exit__ boilerplate for sync ctx managers

Browse Source
This commit is contained in:
Lonami Exo
2019-04-13 10:53:33 +02:00
parent badefcec48
commit 9090ede5db
5 changed files with 33 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
telethon/client/chats.py
View File

@@ -3,7 +3,7 @@ import itertools
import string
from .users import UserMethods
from .. import utils
from .. import helpers, utils
from ..requestiter import RequestIter
from ..tl import types, functions, custom
@@ -69,17 +69,8 @@ class _ChatAction:
self._task = None
def __enter__(self):
if self._client.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__())
def __exit__(self, *args):
return self._client.loop.run_until_complete(self.__aexit__(*args))
__enter__ = helpers._sync_enter
__exit__ = helpers._sync_exit
async def _update(self):
try: