From d02cb84abe2be4df4adfa72043a1fb80aa486690 Mon Sep 17 00:00:00 2001 From: Lonami Exo Date: Sun, 8 Jul 2018 00:04:50 +0200 Subject: [PATCH] Fix end of sync with block warning on disconnect --- telethon/client/auth.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/telethon/client/auth.py b/telethon/client/auth.py index 25c0444c..90d3da05 100644 --- a/telethon/client/auth.py +++ b/telethon/client/auth.py @@ -1,5 +1,6 @@ import getpass import hashlib +import inspect import os import sys @@ -469,7 +470,12 @@ class AuthMethods(MessageParseMethods, UserMethods): return await self.start() def __exit__(self, *args): - self.disconnect() + if self._loop.is_running(): + self._loop.create_task(self.disconnect()) + elif inspect.iscoroutinefunction(self.disconnect): + self._loop.run_until_complete(self.disconnect()) + else: + self.disconnect() async def __aexit__(self, *args): await self.disconnect()