From f65212c2c3241b78fbd04cdb16d7b2d2d8250b50 Mon Sep 17 00:00:00 2001 From: deathbybandaid Date: Mon, 6 May 2019 14:32:51 -0400 Subject: [PATCH] Tweak gif searching --- sopel_modules/SpiceBot_GifSearch/gifsearch.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/sopel_modules/SpiceBot_GifSearch/gifsearch.py b/sopel_modules/SpiceBot_GifSearch/gifsearch.py index 267e7d3..062f268 100644 --- a/sopel_modules/SpiceBot_GifSearch/gifsearch.py +++ b/sopel_modules/SpiceBot_GifSearch/gifsearch.py @@ -77,8 +77,11 @@ def setup(bot): valid_gif_api_dict[gif_api]["apikey"] = None bot.memory["SpiceBot_GifSearch"]['valid_gif_api_dict'][gif_api] = valid_gif_api_dict[gif_api] - for prefixcommand in bot.memory["SpiceBot_GifSearch"]['valid_gif_api_dict'].keys(): - commandsquery_register(bot, "prefix", prefixcommand) + for validgifapi in bot.memory["SpiceBot_GifSearch"]['valid_gif_api_dict'].keys(): + commandsquery_register(bot, "prefix", validgifapi) + + if validgifapi not in bot.memory["SpiceBot_GifSearch"]['cache'].keys(): + bot.memory["SpiceBot_GifSearch"]['cache'][validgifapi] = dict() bot_events_trigger(bot, 2003, "SpiceBot_GifSearch") @@ -235,7 +238,7 @@ def getGif(bot, searchdict): if gifpage and not str(gifpage.status_code).startswith(tuple(["4", "5"])): randombad = False else: - bot.memory["SpiceBot_GifSearch"]["badgiflinks"].append(gifdict["returnurl"]) + bot.memory["SpiceBot_GifSearch"]["badgiflinks"].append(str(gifdict["returnurl"])) newlist = [] for tempdict in gifapiresults: if tempdict["returnurl"] != gifdict["returnurl"]: