matthew/enhanced-discord.py
1
0
Fork 0
Code Issues 4 Pull Requests 21 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

6
discord/shard.py
View File

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