mirrors/discord.py
1
0
Fork 0
mirror of https://github.com/Rapptz/discord.py.git synced 2025-04-21 00:07:51 +00:00
Code Issues Projects Releases Wiki Activity

Handle connection errors during reidentify flow.

Browse Source
This commit is contained in:
Rapptz 2020-04-11 19:06:22 -04:00
parent f658fcf164
commit 80cc5b5edb
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
discord/shard.py
View File

@ -135,12 +135,16 @@ class Shard:
async def reidentify(self, exc):
self._cancel_task()
log.info('Got a request to %s the websocket at Shard ID %s.', exc.op, self.id)
coro = DiscordWebSocket.from_client(self._client, resume=exc.resume, shard_id=self.id,
session=self.ws.session_id, sequence=self.ws.sequence)
self._dispatch('disconnect')
self.ws = await asyncio.wait_for(coro, timeout=180.0)
self.launch()
log.info('Got a request to %s the websocket at Shard ID %s.', exc.op, self.id)
try:
coro = DiscordWebSocket.from_client(self._client, resume=exc.resume, shard_id=self.id,
session=self.ws.session_id, sequence=self.ws.sequence)
self.ws = await asyncio.wait_for(coro, timeout=180.0)
except self._handled_exceptions as e:
await self._handle_disconnect(e)
else:
self.launch()
async def reconnect(self):
self._cancel_task()