From e2250d402e8ad035b2653eb411c8e744cc9eb3bf Mon Sep 17 00:00:00 2001 From: Rapptz Date: Sat, 31 Jul 2021 20:12:40 -0400 Subject: [PATCH] Add on_socket_event_type event --- discord/gateway.py | 6 +++--- docs/api.rst | 12 ++++++++++++ 2 files changed, 15 insertions(+), 3 deletions(-) diff --git a/discord/gateway.py b/discord/gateway.py index aea1aad47..163f480a2 100644 --- a/discord/gateway.py +++ b/discord/gateway.py @@ -423,7 +423,9 @@ class DiscordWebSocket: msg = utils.from_json(msg) log.debug('For Shard ID %s: WebSocket Event: %s', self.shard_id, msg) - self._dispatch('socket_response', msg) + event = msg.get('t') + if event: + self._dispatch('socket_event_type', event) op = msg.get('op') data = msg.get('d') @@ -476,8 +478,6 @@ class DiscordWebSocket: log.warning('Unknown OP code %s.', op) return - event = msg.get('t') - if event == 'READY': self._trace = trace = data.get('_trace', []) self.sequence = msg['s'] diff --git a/docs/api.rst b/docs/api.rst index f10099139..77be49ca6 100644 --- a/docs/api.rst +++ b/docs/api.rst @@ -297,6 +297,18 @@ to handle it, which defaults to print a traceback and ignoring the exception. :param kwargs: The keyword arguments for the event that raised the exception. +.. function:: on_socket_event_type(event_type) + + Called whenever a websocket event is received from the WebSocket. + + This is mainly useful for logging how many events you are receiving + from the Discord gateway. + + .. versionadded:: 2.0 + + :param event_type: The event type from Discord that is received, e.g. ``'READY'``. + :type event_type: :class:`str` + .. function:: on_socket_raw_receive(msg) Called whenever a message is received from the WebSocket, before