1
0
Fork 0
mirror of https://github.com/uhIgnacio/EmoteManager.git synced 2024-08-15 02:23:13 +00:00

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

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
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(
"""
INSERT INTO shard_member_counts (shard_id, member_count)
VALUES ($1, $2)
ON CONFLICT (shard_id) DO UPDATE
SET member_count = EXCLUDED.member_count
INSERT INTO shard_info (shard_id, guild_count, member_count)
VALUES ($1, $2, $3)
ON CONFLICT (shard_id) DO UPDATE SET
guild_count = EXCLUDED.guild_count,
member_count = EXCLUDED.member_count
""",
shard_id, member_count,
shard_id, guild_count, member_count,
)
def main():

View file

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