From af5daa28c30b7b009626218aedbdc5f51b887b5a Mon Sep 17 00:00:00 2001
From: sgoudham <sgoudham@gmail.com>
Date: Sat, 25 Jul 2020 18:43:38 +0100
Subject: [PATCH] Adding functions to store modlog channels within cache

---
 settings.py | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)

diff --git a/settings.py b/settings.py
index ba9c36c7..9a6487a9 100644
--- a/settings.py
+++ b/settings.py
@@ -52,6 +52,58 @@ colour_list = [c for c in colors.values()]
 # Store guildID's and modlog channel within a cached dictionary
 modlogs = {}
 
+
+# Updating the prefix within the dict and database when the method is called
+async def storage_modlog_for_guild(ctx, channelID):
+    modlogs[str(ctx.guild.id)] = channelID
+
+    with db.connection() as connection:
+        # Update the existing prefix within the database
+        update_query = """UPDATE guilds SET modlogs = (?) WHERE guildID = (?)"""
+        update_vals = channelID, ctx.guild.id,
+
+        # Using the connection cursor
+        with closing(connection.cursor()) as cur:
+            # Execute the query
+            cur.execute(update_query, update_vals)
+            print(cur.rowcount, f"Modlog channel for guild {ctx.guild.name} has been updated")
+
+    channel = ctx.guild.get_channel(channelID)
+    # Let the user know that the guild prefix has been updated
+    await ctx.send(f"**Modlog Channel for **{ctx.guild.name}** has been updated to {channel.mention}**")
+
+
+# Method to store the cached modlog channels
+def cache_modlogs(guildid, channel):
+    modlogs[guildid] = channel
+
+
+# Deleting the key - value pair for guild/modlogs
+def del_modlog_channel(guildid):
+    del modlogs[guildid]
+
+
+# Get the modlog channel of the guild that the user is in
+def get_modlog_for_guild(guildid):
+    channel = modlogs[guildid]
+    if channel is not None:
+        return channel
+    return Exception
+
+
+# Before initialising the cache. Store the prefixes from the database within the cache
+with db.connection() as conn:
+    # Grab the prefix of the server from the database
+    select_query = """SELECT * FROM guilds"""
+    with closing(conn.cursor()) as cursor:
+        # Execute the query
+        cursor.execute(select_query)
+        results = cursor.fetchall()
+
+        # Store the guildids and modlog channels
+        for row in results:
+            cache_modlogs(row[0], row[2])
+
 # Storing the prefixes and guildID's in the cache
 cached_prefixes = {}