o
/
EmoteManager
mirror of https://github.com/uhIgnacio/EmoteManager.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

record guild counts too

This commit is contained in:
io mintz 2021-04-06 04:19:37 +00:00
parent 553267d9d8
commit ed9f82163e
2 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
bot.py
View File

@ -69,15 +69,19 @@ 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):
member_count = sum(guild.member_count for guild in self.guilds if guild.shard_id == 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_member_counts (shard_id, member_count) INSERT INTO shard_info (shard_id, guild_count, member_count)
VALUES ($1, $2) VALUES ($1, $2, $3)
ON CONFLICT (shard_id) DO UPDATE ON CONFLICT (shard_id) DO UPDATE SET
SET member_count = EXCLUDED.member_count guild_count = EXCLUDED.guild_count,
member_count = EXCLUDED.member_count
""", """,
shard_id, member_count, shard_id, guild_count, member_count,
) )
def main(): def main():

5
schema.sql
View File

@ -12,8 +12,9 @@ CREATE TABLE invokes (
CREATE INDEX invokes_invoked_at_idx ON invokes (invoked_at); CREATE INDEX invokes_invoked_at_idx ON invokes (invoked_at);
CREATE TABLE shard_member_counts ( CREATE TABLE shard_info (
shard_id INT2 PRIMARY KEY, shard_id INT2 PRIMARY KEY,
-- sum(guild.member_count for guild in shard) -- sum(guild.member_count for guild in shard)
member_count INT4 NOT NULL member_count INT4 NOT NULL,
guild_count INT2 NOT NULL
); );