Raise ResumeWebsocket in case of RECONNECT

This commit is contained in:
Anis B 2017-10-11 05:40:30 +02:00 committed by GitHub
parent 192f57efef
commit 1cd50919a1

View File

@ -45,10 +45,6 @@ __all__ = [ 'ReconnectWebSocket', 'DiscordWebSocket',
'KeepAliveHandler', 'VoiceKeepAliveHandler',
'DiscordVoiceWebSocket', 'ResumeWebSocket' ]
class ReconnectWebSocket(Exception):
"""Signals to handle the RECONNECT opcode."""
pass
class ResumeWebSocket(Exception):
"""Signals to initialise via RESUME opcode instead of IDENTIFY."""
pass
@ -339,7 +335,7 @@ class DiscordWebSocket(websockets.client.WebSocketClientProtocol):
# internal exception signalling to reconnect.
log.info('Received RECONNECT opcode.')
yield from self.close()
raise ReconnectWebSocket()
raise ResumeWebsocket()
if op == self.HEARTBEAT_ACK:
self._keep_alive.ack()