Fix websockets 4.0 support

This commit is contained in:
BeatButton 2018-02-11 09:41:18 -07:00 committed by Rapptz
parent ad7506050c
commit 9d3d8e0f0a
2 changed files with 5 additions and 5 deletions

View File

@ -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)

View File

@ -1,2 +1,2 @@
aiohttp>=2.0.0,<2.3.0
websockets>=3.1,<4.0
websockets>=4.0,<5.0