From 60d635ccfde32873e6f4125af14046d3ae0ed0b9 Mon Sep 17 00:00:00 2001 From: MarkusRost <2701034-MarkusRost@users.noreply.gitlab.com> Date: Mon, 22 Jul 2024 22:30:07 +0000 Subject: [PATCH] Don't add the same wiki to queue multiple times --- src/bot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/bot.py b/src/bot.py index 83745fb..3778cdf 100644 --- a/src/bot.py +++ b/src/bot.py @@ -153,7 +153,7 @@ class RcQueue: full = set() async with db.pool().acquire() as connection: async with connection.transaction(): - async for db_wiki in connection.cursor('SELECT DISTINCT wiki, row_number() OVER (ORDER BY webhook) AS rowid, webhook, lang, display, rcid FROM rcgcdw WHERE rcid != -1 OR rcid IS NULL ORDER BY webhook'): + async for db_wiki in connection.cursor('SELECT DISTINCT wiki, row_number() OVER (ORDER BY wiki) AS rowid, rcid FROM rcgcdw WHERE rcid != -1 OR rcid IS NULL GROUP BY wiki, rcid ORDER BY rowid'): domain = get_domain(db_wiki["wiki"]) try: if db_wiki["wiki"] not in all_wikis: @@ -253,7 +253,7 @@ async def generate_domain_groups(): domain_wikis = defaultdict(list) async with db.pool().acquire() as connection: async with connection.transaction(): - async for db_wiki in connection.cursor('SELECT DISTINCT wiki, webhook, lang, display, rcid FROM rcgcdw WHERE rcid != -1 OR rcid IS NULL'): + async for db_wiki in connection.cursor('SELECT DISTINCT wiki FROM rcgcdw WHERE rcid != -1 OR rcid IS NULL'): domain_wikis[get_domain(db_wiki["wiki"])].append(QueuedWiki(db_wiki["wiki"], 20)) for group, db_wikis in domain_wikis.items(): yield group, db_wikis