From bd9821320ec2288a4c3cfada980a85e78630f94f Mon Sep 17 00:00:00 2001 From: jack1142 <6032823+jack1142@users.noreply.github.com> Date: Fri, 24 Jul 2020 13:11:45 +0200 Subject: Use a subclass rather than monkey-patching for the websocket --- discord/gateway.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) (limited to 'discord/gateway.py') diff --git a/discord/gateway.py b/discord/gateway.py index 56c1be9e..d54ab597 100644 --- a/discord/gateway.py +++ b/discord/gateway.py @@ -164,14 +164,9 @@ class VoiceKeepAliveHandler(KeepAliveHandler): self.latency = ack_time - self._last_send self.recent_ack_latencies.append(self.latency) -# Monkey patch certain things from the aiohttp websocket code -# Check this whenever we update dependencies. -OLD_CLOSE = aiohttp.ClientWebSocketResponse.close - -async def _new_ws_close(self, *, code: int = 4000, message: bytes = b'') -> bool: - return await OLD_CLOSE(self, code=code, message=message) - -aiohttp.ClientWebSocketResponse.close = _new_ws_close +class DiscordClientWebSocketResponse(aiohttp.ClientWebSocketResponse): + async def close(self, *, code: int = 4000, message: bytes = b'') -> bool: + return await super().close(code=code, message=message) class DiscordWebSocket: """Implements a WebSocket for Discord's gateway v6. -- cgit v1.2.3