From afe6319db5360ce53b68bcf67d8608e18bf2715c Mon Sep 17 00:00:00 2001 From: eyMarv Date: Sat, 6 Jan 2024 23:43:30 +0100 Subject: [PATCH] Add watchdog error saver --- pyrogram/client.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/pyrogram/client.py b/pyrogram/client.py index e12604f5..ed64835c 100644 --- a/pyrogram/client.py +++ b/pyrogram/client.py @@ -331,6 +331,7 @@ class Client(Methods): # after some idle time has been detected. self.updates_watchdog_task = None self.updates_watchdog_event = asyncio.Event() + self.updates_watchdog_error = None self.last_update_time = datetime.now() self.listeners = {listener_type: [] for listener_type in pyrogram.enums.ListenerTypes} self.loop = asyncio.get_event_loop() @@ -362,8 +363,11 @@ class Client(Methods): else: break - if datetime.now() - self.last_update_time > timedelta(seconds=self.UPDATES_WATCHDOG_INTERVAL): - await self.invoke(raw.functions.updates.GetState()) + try: + if datetime.now() - self.last_update_time > timedelta(seconds=self.UPDATES_WATCHDOG_INTERVAL): + await self.invoke(raw.functions.updates.GetState()) + except Exception as ee: + self.updates_watchdog_error = ee async def authorize(self) -> User: if self.bot_token: