LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-26 03:13:45 +03:00
Code Issues Packages Projects Releases Wiki Activity

Raise RuntimeError on with blocks that need async

Browse Source
This commit is contained in:
Lonami Exo 2018-10-17 11:30:07 +02:00
parent 9cbc088b76
commit 419fe6dca3
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
telethon/client/auth.py
View File

@ -483,6 +483,12 @@ class AuthMethods(MessageParseMethods, UserMethods):
# region with blocks # region with blocks
def __enter__(self): def __enter__(self):
if self._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.start() return self.start()
async def __aenter__(self): async def __aenter__(self):

6
telethon/tl/custom/conversation.py
View File

@ -379,6 +379,12 @@ class Conversation(ChatGetter):
fut.cancel() fut.cancel()
def __enter__(self): 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__()) return self._client.loop.run_until_complete(self.__aenter__())
async def __aenter__(self): async def __aenter__(self):