diff options
| author | Rapptz <[email protected]> | 2017-02-08 04:37:16 -0500 |
|---|---|---|
| committer | Rapptz <[email protected]> | 2017-02-08 04:37:16 -0500 |
| commit | dc486980f87249583a38a22c79aa0f7370b75d55 (patch) | |
| tree | e2c6b091c40469e51e358b0c736bd81e29137e4f /discord/shard.py | |
| parent | Only defer the lock if we're pre-emptively rate limiting. (diff) | |
| download | discord.py-dc486980f87249583a38a22c79aa0f7370b75d55.tar.xz discord.py-dc486980f87249583a38a22c79aa0f7370b75d55.zip | |
Rewrite RESUME logic to be more in line with what is requested.
Apparently we should always try to RESUME first and if we get
INVALIDATE_SESSION then we should IDENTIFY instead. This is the
preferred way to do RESUMEs.
Diffstat (limited to 'discord/shard.py')
| -rw-r--r-- | discord/shard.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/discord/shard.py b/discord/shard.py index 7bb3a201..dade454d 100644 --- a/discord/shard.py +++ b/discord/shard.py @@ -54,10 +54,9 @@ class Shard: def poll(self): try: yield from self.ws.poll_event() - except (ReconnectWebSocket, ResumeWebSocket) as e: - resume = type(e) is ResumeWebSocket - log.info('Got ' + type(e).__name__) - self.ws = yield from DiscordWebSocket.from_client(self._client, resume=resume, + except ResumeWebSocket as e: + log.info('Got a request to RESUME the websocket.') + self.ws = yield from DiscordWebSocket.from_client(self._client, resume=True, shard_id=self.id, session=self.ws.session_id, sequence=self.ws.sequence) |