matthew/enhanced-discord.py
1
0
Fork 0
Code Issues 4 Pull Requests 21 Projects Releases Wiki Activity

Speed up chunking for guilds with presence intent enabled

Browse Source
This commit is contained in:
Rapptz 2020-09-05 13:57:50 -04:00
parent 92e1816114
commit 0ec72660cf
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
discord/state.py
View File

@ -303,6 +303,10 @@ class ConnectionState:
self._add_guild(guild) self._add_guild(guild)
return guild return guild
def _guild_needs_chunking(self, guild):
# If presences are enabled then we get back the old guild.large behaviour
return self._fetch_offline and not guild.chunked and not (self._intents.presences and not guild.large)
def _get_guild_channel(self, data): def _get_guild_channel(self, data):
channel_id = int(data['channel_id']) channel_id = int(data['channel_id'])
try: try:
@ -356,7 +360,8 @@ class ConnectionState:
except asyncio.TimeoutError: except asyncio.TimeoutError:
break break
else: else:
if self._fetch_offline:
if self._guild_needs_chunking(guild):
future = await self.chunk_guild(guild, wait=False) future = await self.chunk_guild(guild, wait=False)
states.append((guild, future)) states.append((guild, future))
else: else:
@ -781,7 +786,7 @@ class ConnectionState:
return return
# check if it requires chunking # check if it requires chunking
if self._fetch_offline: if self._guild_needs_chunking(guild):
asyncio.ensure_future(self._chunk_and_dispatch(guild, unavailable), loop=self.loop) asyncio.ensure_future(self._chunk_and_dispatch(guild, unavailable), loop=self.loop)
return return
@ -1053,7 +1058,7 @@ class AutoShardedConnectionState(ConnectionState):
except asyncio.TimeoutError: except asyncio.TimeoutError:
break break
else: else:
if self._fetch_offline: if self._guild_needs_chunking(guild):
# Chunk the guild in the background while we wait for GUILD_CREATE streaming # Chunk the guild in the background while we wait for GUILD_CREATE streaming
future = asyncio.ensure_future(self.chunk_guild(guild)) future = asyncio.ensure_future(self.chunk_guild(guild))
else: else: