Add db, prefix cog and update set cog

Author: JannesT3011

cogs/config/__init__.py

@@ -0,0 +1 @@
+from util.checks import is_bot_owner

cogs/config/db.py

@@ -0,0 +1,31 @@
+from discord.ext import commands
+from . import is_bot_owner
+from dataminer import bot_requests
+
+class Db_cmds(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    @is_bot_owner()
+    @commands.group(name="db")
+    async def _db(self, ctx):
+        return
+
+    @is_bot_owner()
+    @commands.cooldown(1, 20.0, commands.BucketType.user)
+    @_db.command(name="add_field", aliases=["addfield", "fieldadd", "field_add"])
+    async def _db_addfield(self, ctx, *, field_name: str, type):
+        """CREATE A NEW FIELD IN THE DATABASE"""
+        bot_requests(ctx.message, str(ctx.command), self.bot.db)
+
+        types = {
+            "list": [],
+            "object": {},
+            "string": "",
+            "int": int
+        }
+        self.bot.db.update_one({"_id": str(ctx.guild.id)}, {"$set": {field_name: types[type]}})
+
+
+def setup(bot):
+    bot.add_cog(Db_cmds(bot))

cogs/config/prefix.py

@@ -0,0 +1,27 @@
+from discord.ext import commands
+import discord
+from dataminer import bot_requests
+from CONFIG import DEFAULT_PREFIX
+
+class Prefix(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    def prefix(self, server_id: str):
+        pr_list = self.bot.db.find({"_id": server_id})
+        for pr in pr_list:
+            prefix = pr["prefix"]
+
+        return prefix, DEFAULT_PREFIX, f"<@{self.bot.user.id}>"
+
+    @commands.cooldown(1, 5.0, commands.BucketType.user)
+    @commands.command(name="prefix")
+    async def _prefix(self, ctx):
+        """SHOWS THE SERVER PREFIX """
+        bot_requests(ctx.message, str(ctx.command), self.bot.db)
+
+        embed = discord.Embed(title="Every prefix you can use on this server:", description=f", \n".join(self.prefix(str(ctx.guild.id))))
+        return await ctx.send(embed=embed)
+
+def setup(bot):
+    bot.add_cog(Prefix(bot))

cogs/config/set.py

@@ -2,6 +2,7 @@
 from . import is_bot_owner
 import discord
 import asyncio
+from dataminer import bot_requests
 
 class Set(commands.Cog):
     def __init__(self, bot):
@@ -12,13 +13,20 @@ def __init__(self, bot):
     @commands.group(name="set")
     async def _set(self, ctx):
         """RETURN ALL AVAILABLE SET FUNCTIONS"""
-        return
+        bot_requests(ctx.message, str(ctx.command), self.bot.db)
+
+        embed = discord.Embed(title="`Set` commands")
+        embed.add_field(name="`set game <NAME>`", value="Set the game/status of the bot", inline=False)
+        embed.add_field(name="`set prefix <PREFIX>`", value="Set the server prefix", inline=False)
+        return await ctx.send(embed=embed)
 
     @is_bot_owner()
     @commands.cooldown(1, 10.0, commands.BucketType.user)
     @_set.command(name="game")
     async def _set_game(self, ctx, *, game_name: str):
         """SET BOT GAME"""
+        bot_requests(ctx.message, str(ctx.command), self.bot.db)
+
         await ctx.send(f"React with 👌 to change the game to `{game_name}`!")
 
         def check(reaction, user):
@@ -38,6 +46,8 @@ def check(reaction, user):
     @_set.command(name="prefix")
     async def _set_prefix(self, ctx, *, prefix: str):
         """SET A SERVER PREFIX"""
+        bot_requests(ctx.message, str(ctx.command), self.bot.db)
+
         self.bot.db.update({"_id": str(ctx.guild.id)}, {"$set": {"prefix": prefix}})
 
         await ctx.send("React with 👌 to change the prefix!")

core/__init__.py

@@ -1,2 +1,3 @@
 from core.bot import run, Bot
-from core.database import DbClient, Database
+from core.database import DbClient, Database
+from CONFIG import DEFAULT_PREFIX

core/cogs.py

@@ -9,4 +9,5 @@
         "util.error",
         "cogs.plot",
         "cogs.config.set",
-        "cogs.config.db"]
+        "cogs.config.db",
+        "cogs.config.prefix"]

util/__init__.py

@@ -0,0 +1 @@
+from CONFIG import OWNER_ID