From 129368fcbdd3f9773ec38b375bfb616475354c5c Mon Sep 17 00:00:00 2001 From: Frisk Date: Thu, 6 Aug 2020 15:39:02 +0200 Subject: [PATCH] Fast fix for adding new wikis --- src/bot.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/src/bot.py b/src/bot.py index baaa9cd..ceb62f4 100644 --- a/src/bot.py +++ b/src/bot.py @@ -105,8 +105,13 @@ class RcQueue: full = [] for db_wiki in fetch_all.fetchall(): domain = get_domain(db_wiki["wiki"]) - current_domain = self[domain] try: + all_wikis[db_wiki["wiki"]] + except KeyError: + all_wikis[db_wiki["wiki"]] = Wiki() + all_wikis[db_wiki["wiki"]].rc_active = True + try: + current_domain = self[domain] if not db_wiki["ROWID"] < current_domain["last_rowid"]: current_domain["query"].append(db_wiki) self.to_remove.remove(db_wiki["wiki"]) @@ -288,7 +293,10 @@ async def discussion_handler(): header["Accept"] = "application/hal+json" async with aiohttp.ClientSession(headers=header, timeout=aiohttp.ClientTimeout(3.0)) as session: - local_wiki = all_wikis[db_wiki["wiki"]] # set a reference to a wiki object from memory + try: + local_wiki = all_wikis[db_wiki["wiki"]] # set a reference to a wiki object from memory + except KeyError: + local_wiki = all_wikis[db_wiki["wiki"]] = Wiki() try: feeds_response = await local_wiki.fetch_feeds(db_wiki["wikiid"], session) except (WikiServerError, WikiError):