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

Chunk every guild if it's a user bot.

Browse Source 
This also increases the timeout for chunking by 30 seconds for every
chunk we're waiting for.
This commit is contained in:
Rapptz 2016-12-24 17:32:07 -05:00
parent 4d59bca4aa
commit a8264f6540
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
discord/state.py
View File

@ -187,7 +187,7 @@ class ConnectionState:
# wait for the chunks # wait for the chunks
if chunks: if chunks:
try: try:
yield from asyncio.wait(chunks, timeout=len(chunks), loop=self.loop) yield from asyncio.wait(chunks, timeout=len(chunks) * 30.0, loop=self.loop)
except asyncio.TimeoutError: except asyncio.TimeoutError:
log.info('Somehow timed out waiting for chunks.') log.info('Somehow timed out waiting for chunks.')
@ -199,7 +199,8 @@ class ConnectionState:
# call GUILD_SYNC after we're done chunking # call GUILD_SYNC after we're done chunking
if not self.is_bot: if not self.is_bot:
compat.create_task(self.syncer([s.id for s in self.servers]), loop=self.loop) log.info('Requesting GUILD_SYNC for %s guilds' % len(self.servers))
yield from self.syncer([s.id for s in self.servers])
# dispatch the event # dispatch the event
self.dispatch('ready') self.dispatch('ready')
@ -212,7 +213,7 @@ class ConnectionState:
servers = self._ready_state.servers servers = self._ready_state.servers
for guild in guilds: for guild in guilds:
server = self._add_server_from_data(guild) server = self._add_server_from_data(guild)
if not self.is_bot and server.large: if not self.is_bot or server.large:
servers.append(server) servers.append(server)
for pm in data.get('private_channels'): for pm in data.get('private_channels'):