From 5f4e79542fe01f277b17aa5a3ea439b938bb1b79 Mon Sep 17 00:00:00 2001 From: alexvelickiy Date: Thu, 31 Aug 2017 19:57:32 +0300 Subject: [PATCH] Restore updates thread on reconnection and fix restarting it --- telethon/telegram_client.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/telethon/telegram_client.py b/telethon/telegram_client.py index d75ba28b..a5664107 100644 --- a/telethon/telegram_client.py +++ b/telethon/telegram_client.py @@ -133,7 +133,15 @@ class TelegramClient(TelegramBareClient): *args will be ignored. """ - return super().connect() + result = super().connect() + + # Checking if there are update_handlers and if true, start running updates thread. + # This situation may occur on reconnecting. + if result and self._update_handlers: + self._set_updates_thread(running=True) + + return result + def disconnect(self): """Disconnects from the Telegram server @@ -914,7 +922,7 @@ class TelegramClient(TelegramBareClient): name='UpdatesThread', daemon=True, target=self._updates_thread_method) - self._updates_thread.start() + self._updates_thread.start() else: self._updates_thread_running.clear() if self._updates_thread_receiving.is_set():