From 8070d39a23863aacd52b28e1b432e128cf5d3a2e Mon Sep 17 00:00:00 2001 From: Rapptz Date: Sat, 11 Apr 2020 22:35:07 -0400 Subject: Add shard related connection and resume events. These include: * on_shard_resumed * on_shard_connect * on_shard_disconnect --- discord/shard.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'discord/shard.py') diff --git a/discord/shard.py b/discord/shard.py index 0b037398..dfa3849c 100644 --- a/discord/shard.py +++ b/discord/shard.py @@ -104,6 +104,7 @@ class Shard: async def _handle_disconnect(self, e): self._dispatch('disconnect') + self._dispatch('shard_disconnect', self.id) if not self._reconnect: self._queue.put_nowait(EventItem(EventType.close, self, e)) return @@ -136,6 +137,7 @@ class Shard: async def reidentify(self, exc): self._cancel_task() self._dispatch('disconnect') + self._dispatch('shard_disconnect', self.id) log.info('Got a request to %s the websocket at Shard ID %s.', exc.op, self.id) try: coro = DiscordWebSocket.from_client(self._client, resume=exc.resume, shard_id=self.id, -- cgit v1.2.3