From 9d3d8e0f0aacb22b0a7d4984f4ed21537db480f8 Mon Sep 17 00:00:00 2001 From: BeatButton Date: Sun, 11 Feb 2018 09:41:18 -0700 Subject: [PATCH] Fix websockets 4.0 support --- discord/gateway.py | 8 ++++---- requirements.txt | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/discord/gateway.py b/discord/gateway.py index c995ef468..6d535a380 100644 --- a/discord/gateway.py +++ b/discord/gateway.py @@ -516,11 +516,11 @@ class DiscordWebSocket(websockets.client.WebSocketClientProtocol): yield from self.send_as_json(payload) @asyncio.coroutine - def close_connection(self, force=False): + def close_connection(self, *args, **kwargs): if self._keep_alive: self._keep_alive.stop() - yield from super().close_connection(force=force) + yield from super().close_connection(*args, **kwargs) class DiscordVoiceWebSocket(websockets.client.WebSocketClientProtocol): """Implements the websocket protocol for handling voice connections. @@ -701,10 +701,10 @@ class DiscordVoiceWebSocket(websockets.client.WebSocketClientProtocol): raise ConnectionClosed(e, shard_id=None) from e @asyncio.coroutine - def close_connection(self, force=False): + def close_connection(self, *args, **kwargs): if self._keep_alive: self._keep_alive.stop() - yield from super().close_connection(force=force) + yield from super().close_connection(*args, **kwargs) diff --git a/requirements.txt b/requirements.txt index 5a7e43e7e..508b44b5a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,2 @@ aiohttp>=2.0.0,<2.3.0 -websockets>=3.1,<4.0 +websockets>=4.0,<5.0