add api connection & refactor belts module

Author: danielsp45

.env.sample

@@ -1,3 +1,5 @@
 BOT_TOKEN='<Your Token Here>'
 GUILD_ID='<Your Guild ID Here>'
 GONGO_CHANNEL_ID='<The Channel For The Daily Reports Here>'
+BOKKEN_JWT='<Jason Web Token provided by the Bokken API here>'
+API_URL='<API url to make the requests from>'

:w

@@ -0,0 +1,186 @@
+import json
+import time
+import asyncio
+
+import discord
+from discord.ext import commands
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+
+from bot.cogs.utils.constants import *
+from bot.cogs.utils.logs import AttributionLogs, Base, log_attribution
+from bot.web import *
+
+# sqlalchemy setup
+engine = create_engine("sqlite:///bot/data/daily_logs.db")
+
+Base.metadata.bind = engine
+
+DBSession = sessionmaker(bind=engine)
+
+session = DBSession()
+
+class BeltsAttributions(commands.Cog):
+    """This is a class to handle the discord attribution of belt roles."""
+
+    def __init__(self, client: commands.Bot):
+        self.client = client
+
+    @commands.command(name="promove")
+    @commands.has_any_role(
+        Roles["ADMIN"].name, Roles["CHAMPION"].name
+    )
+    async def promove(
+        self, ctx: discord.ext.commands.Context, user: str, belt: str
+    ) -> None:
+        """This function promotes a user to the next belt."""
+
+        mentions = ctx.message.raw_mentions
+        guild = ctx.guild
+        member = guild.get_member(mentions[0])
+        ninja = Ninja(guild, member)
+
+        if belt == "Branco" and ninja.current_belt() == None:
+            role = get_role_from_name(guild, belt)
+
+            await member.add_roles(guild.get_role(role.id), reason=None, atomic=True)
+
+            # send request to update ninja belt
+            ninja_username = member.name + '#' + member.discriminator
+            await update_belt(ninja_username, belt)
+
+            # Public message
+            asyncio.create_task(ctx.send(f"{user} agora és cinturão {belt} :tada:"))
+
+            # Private message
+            asyncio.create_task(self.send_private_message(member, belt))
+
+            # Adding the log to the database
+            asyncio.create_task(self.log(ctx, member, belt))
+
+        elif belt == ninja.current_belt().name:
+            await ctx.reply(f"Esse já é o cinturão do ninja {user}!")
+
+        elif belt == ninja.next_belt().name:
+            role = get_role_from_name(guild, belt)
+
+            await member.add_roles(guild.get_role(role.id), reason=None, atomic=True)
+
+            # send request to update ninja belt
+            ninja_username = member.name + '#' + member.discriminator
+            await update_belt(ninja_username, belt)
+
+            # Public message
+            await ctx.send(f"{user} agora és cinturão {belt} :tada:")
+
+            # Private message
+            asyncio.create_task(self.send_private_message(member, belt))
+
+            # Adding the log to the database
+            asyncio.create_task(self.log(ctx, member, belt))
+
+        elif belt != ninja.next_belt().name:
+            await ctx.send(f"{user} esse cinturão não é valido de se ser atribuido.")
+
+
+    async def log(self, ctx, member, belt):
+        """This function logs the belt attribution."""
+
+        new_log = AttributionLogs(
+            ninja_id=str(member),
+            mentor_id=str(ctx.author),
+            belt_attributed=belt,
+            timestamp=int(time.time()),
+        )
+
+        session.add(new_log)
+        session.commit()
+
+    async def send_private_message(self, member, belt):
+        """This function sends a private message to the member."""
+
+        file_handler = FileHandler(belt)
+        emoji = translator_to_emoji[belt]
+        user = member
+        embed = discord.Embed(
+            title=f"{emoji} Parabéns, subiste de cinturão :tada:",
+            description=file_handler.msg,
+            color=file_handler.color,
+        )
+
+        await user.send(embed=embed)
+
+class FileHandler:
+    """
+    This is a class to handle a json file.
+
+    Attributes:
+        file (string): The path to the json file being handled.
+    """
+
+    file = "bot/data/belts.json"
+
+    def __init__(self: str, belt: str):
+        """
+        The constructor for the FileHandler class.
+
+        Parameters:
+            color (int): Color code to be displayed in discord embed.
+        """
+        self.belt = belt
+        self.msg = self.get_info()[0]
+        self.color = self.get_info()[1]
+
+    def get_info(self) -> tuple:
+        """
+        The function to get the info from the belts.json file.
+
+        Returns:
+            msg (string): Variable that contains the message of the respective belt.
+            color (int): Color code to be displayed in discord embed.
+        """
+        with open(self.file) as json_file:
+            data = json.load(json_file)
+            msg = f"Subiste para {self.belt} :clap:\n\nPróximos objetivos:"
+            color = int(data[self.belt]["color"], 16)
+            for param in data[self.belt]["goals"]:
+                msg += "\n" + param
+
+            return (msg, color)
+
+class Ninja:
+    """This is a class to get information about a specific ninja."""
+
+    def __init__(self, guild: discord.Guild, member: discord.Member):
+        self.guild = guild
+        self.member = member
+        self.roles = list(member.roles)
+
+    def current_belt(self):
+        """This function returns the current belt of the ninja."""
+
+        highest_belt = None
+        for role in self.roles:
+            for belt in Belts:
+                if belt.name == role.name:
+                    highest_belt = belt
+
+        print(highest_belt)
+        return highest_belt
+
+    def next_belt(self) -> Belts:
+        """This function returns the next belt of the ninja."""
+
+        if self.current_belt() == None:
+            value = 0
+        else if self.current_belt().value < 8:
+            value = self.current_belt().value + 1
+
+        else:
+            value = 8
+
+        return Belts(value)
+
+
+def setup(client: commands.Bot) -> None:
+    client.add_cog(BeltsAttributions(client))

bot/client.py

@@ -1,5 +1,4 @@
 import logging
-import os
 
 import discord
 from discord.ext import commands