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

Add override to close() method to the websocket subclass.

Browse Source 
This makes it so that _keep_alive is stopped no matter which closing
method is called
This commit is contained in:
CraftSpider 2018-12-01 22:45:37 -05:00 committed by Rapptz
parent dd41dac680
commit 068d953b42
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
discord/gateway.py
View File

@ -510,6 +510,12 @@ class DiscordWebSocket(websockets.client.WebSocketClientProtocol):
log.debug('Updating our voice state to %s.', payload) log.debug('Updating our voice state to %s.', payload)
await self.send_as_json(payload) await self.send_as_json(payload)
async def close(self, code=1000, reason=''):
if self._keep_alive:
self._keep_alive.stop()
await super().close(code, reason)
async def close_connection(self, *args, **kwargs): async def close_connection(self, *args, **kwargs):
if self._keep_alive: if self._keep_alive:
self._keep_alive.stop() self._keep_alive.stop()