o/EmoteManager
1
0
Fork 0
mirror of https://github.com/uhIgnacio/EmoteManager.git synced 2024-08-15 02:23:13 +00:00
Code Issues Projects Releases Wiki Activity

fix AttributeError in on_shard_ready/update_shard

Browse source 
see Rapptz/discord.py#6652
This commit is contained in:
io 2021-06-03 08:38:50 +00:00
parent 25119e49f2
commit 2f4e0a3c28
1 changed files with 8 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
bot.py
View file

@ -76,10 +76,6 @@ class Bot(Bot):
# we use on_shard_ready rather than on_ready because the latter is a bit less reliable # we use on_shard_ready rather than on_ready because the latter is a bit less reliable
async def on_shard_ready(self, shard_id): async def on_shard_ready(self, shard_id):
guilds = [guild for guild in self.guilds if guild.shard_id == shard_id]
guild_count = len(guilds)
member_count = sum(guild.member_count for guild in guilds)
await self.pool.execute( await self.pool.execute(
""" """
INSERT INTO shard_info (shard_id, guild_count, member_count) INSERT INTO shard_info (shard_id, guild_count, member_count)
@ -88,14 +84,10 @@ class Bot(Bot):
guild_count = EXCLUDED.guild_count, guild_count = EXCLUDED.guild_count,
member_count = EXCLUDED.member_count member_count = EXCLUDED.member_count
""", """,
shard_id, guild_count, member_count, shard_id, *self.shard_stats(shard_id),
) )
async def update_shard(self, guild): async def update_shard(self, guild):
guilds = [guild2 for guild2 in self.guilds if guild2.shard_id == guild.shard_id]
guild_count = len(guilds)
member_count = sum(guild.member_count for guild in guilds)
await self.pool.execute( await self.pool.execute(
""" """
UPDATE shard_info UPDATE shard_info
@ -104,11 +96,17 @@ class Bot(Bot):
member_count = $3 member_count = $3
WHERE shard_id = $1 WHERE shard_id = $1
""", """,
guild.shard_id, guild_count, member_count, guild.shard_id, *self.shard_stats(guild.shard_id),
) )
on_guild_join = on_guild_remove = update_shard on_guild_join = on_guild_remove = update_shard
def shard_stats(self, shard_id):
guilds = [guild for guild in self.guilds if guild.shard_id == shard_id]
guild_count = len(guilds)
member_count = sum(getattr(guild, 'member_count', 0) for guild in guilds)
return guild_count, member_count
def main(): def main():
import sys import sys