feat(bot): Restart heartbeat on shard reconnect.

2024-10-26 14:42:59 -03:00
parent 80408b02d1
commit 6e1fad16f0
1 changed files with 9 additions and 2 deletions
@@ -124,8 +124,15 @@ class Payphone(commands.AutoShardedBot):

        print(f"Shard {shard_id} was resumed!")

-    async def on_shard_disconnect(shard_id: int) -> None:
-    self.update_heartbeat.cancel()
+    async def on_shard_connect(self, shard_id: int) -> None:
+        try:
+            self.update_heartbeat.start()
+
+        except RuntimeError:
+            pass
+
+    async def on_shard_disconnect(self, shard_id: int) -> None:
+        self.update_heartbeat.cancel()

    async def get_or_fetch_channel(
        self, channel_id: int