matthew/enhanced-discord.py
1
0
Fork 0
Code Issues 4 Pull Requests 21 Projects Releases Wiki Activity

Fix websockets 4.0 support

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

8
discord/gateway.py
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)