diff options
| author | Rapptz <[email protected]> | 2016-06-01 05:13:15 -0400 |
|---|---|---|
| committer | Rapptz <[email protected]> | 2016-06-01 05:13:15 -0400 |
| commit | e0a91df32beabafa990ce09dee13af665c770079 (patch) | |
| tree | d89b01aee4542786949317e731c62afb91b1e9d7 /discord/client.py | |
| parent | [commands] Delete frame objects when done using them. (diff) | |
| download | discord.py-e0a91df32beabafa990ce09dee13af665c770079.tar.xz discord.py-e0a91df32beabafa990ce09dee13af665c770079.zip | |
Add RESUME support.
Diffstat (limited to 'discord/client.py')
| -rw-r--r-- | discord/client.py | 7 |
1 files changed, 4 insertions, 3 deletions
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: |