From e0a91df32beabafa990ce09dee13af665c770079 Mon Sep 17 00:00:00 2001 From: Rapptz Date: Wed, 1 Jun 2016 05:13:15 -0400 Subject: Add RESUME support. --- discord/client.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'discord/client.py') diff --git a/discord/client.py b/discord/client.py index 7ac7ca66..29576845 100644 --- a/discord/client.py +++ b/discord/client.py @@ -401,9 +401,10 @@ class Client: while not self.is_closed: try: yield from self.ws.poll_event() - except ReconnectWebSocket: - log.info('Reconnecting the websocket.') - self.ws = yield from DiscordWebSocket.from_client(self) + except (ReconnectWebSocket, ResumeWebSocket) as e: + resume = type(e) is ResumeWebSocket + log.info('Got ' + type(e).__name__) + self.ws = yield from DiscordWebSocket.from_client(self, resume=resume) except ConnectionClosed as e: yield from self.close() if e.code != 1000: -- cgit v1.2.3