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

Add shard related connection and resume events.

Browse Source 
These include:
* on_shard_resumed
* on_shard_connect
* on_shard_disconnect
This commit is contained in:
Rapptz
2020-04-11 22:35:07 -04:00
parent 80cc5b5edb
commit 8070d39a23
4 changed files with 41 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
discord/gateway.py
View File

@ -438,11 +438,15 @@ class DiscordWebSocket:
self._trace = trace = data.get('_trace', [])
self.sequence = msg['s']
self.session_id = data['session_id']
# pass back shard ID to ready handler
data['__shard_id__'] = self.shard_id
log.info('Shard ID %s has connected to Gateway: %s (Session ID: %s).',
self.shard_id, ', '.join(trace), self.session_id)
elif event == 'RESUMED':
self._trace = trace = data.get('_trace', [])
# pass back the shard ID to the resumed handler
data['__shard_id__'] = self.shard_id
log.info('Shard ID %s has successfully RESUMED session %s under trace %s.',
self.shard_id, self.session_id, ', '.join(trace))

2
discord/shard.py
View File

@ -104,6 +104,7 @@ class Shard:
async def _handle_disconnect(self, e):
self._dispatch('disconnect')
self._dispatch('shard_disconnect', self.id)
if not self._reconnect:
self._queue.put_nowait(EventItem(EventType.close, self, e))
return
@ -136,6 +137,7 @@ class Shard:
async def reidentify(self, exc):
self._cancel_task()
self._dispatch('disconnect')
self._dispatch('shard_disconnect', 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,

5
discord/state.py
View File

@ -1142,5 +1142,10 @@ class AutoShardedConnectionState(ConnectionState):
self._add_private_channel(factory(me=user, data=pm, state=self))
self.dispatch('connect')
self.dispatch('shard_connect', data['__shard_id__'])
if self._ready_task is None:
self._ready_task = asyncio.ensure_future(self._delay_ready(), loop=self.loop)
def parse_resumed(self, data):
self.dispatch('resumed')
self.dispatch('shard_resumed', data['__shard_id__'])