mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2025-03-30 23:04:14 +03:00
Fix-up new sync __enter__ not handling the client directly
This commit is contained in:
parent
9090ede5db
commit
20b8250037
|
@ -108,17 +108,27 @@ def _sync_enter(self):
|
||||||
Helps to cut boilerplate on async context
|
Helps to cut boilerplate on async context
|
||||||
managers that offer synchronous variants.
|
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(
|
raise RuntimeError(
|
||||||
'You must use "async with" if the event loop '
|
'You must use "async with" if the event loop '
|
||||||
'is running (i.e. you are inside an "async def")'
|
'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):
|
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
|
# endregion
|
||||||
|
|
Loading…
Reference in New Issue
Block a user