Add feature private messages (#17)

Author: danielsp45

bot/cogs/belts.json

@@ -0,0 +1,63 @@
+{
+    "Branco": {
+        "objectives" : [
+            "- Finalizar um projeto sozinho",
+            "- Apresentar o teu projeto a outros ninjas",
+            "- Saber o nome de 5 ninjas e 2 mentores",
+            "- Completar o Lightbot"],
+        "color" : "FFFFFF"
+        },
+    "Amarelo" : {
+        "objectives" : [
+            "- Estar presente em 4 sessões", 
+            "- Fazer um projeto e apresentar à mesa"],
+        "color" : "F7E129"
+    },
+    "Azul" : {
+        "objectives" : [
+            "- Ajudar como mentor numa sessão",
+            "- Saber o nome de 7 ninjas e 4 mentores",
+            "- Conseguir completar o código que os mentores têm preparado para ti",
+            "- Fazer um site"],
+        "color" : "296DF7"
+        },
+    "Verde" : {
+        "objectives" : [
+            "- Apresentar um projeto para o Dojo inteiro",
+            "- Chegar a 5 Kyu em CodeWars",
+            "- Missão secreta n.º2",
+            "- Projeto em raspberry c/ apresentação"],
+        "color" : "0ACD1B"
+    },
+    "Laranja" : {
+        "objectives" : [
+            "- Projeto de Fuler",
+            "- Com a ajuda de um mentor, montar um computador",
+            "- Introduzir conceitos de bash",
+            "- Missão secreta n.º4 (difícil)"],
+        "color" : "F48005"
+    },
+    "Vermelho" : {
+        "objectives" : [
+            "- Ser mentor 1 sessão",
+            "- Demonstração de conhecimento do paradigma POO",
+            "- Ter um projeto que use API externa"],
+        "color" : "F60909"
+    },
+    "Roxo" : {
+        "objectives" : [
+            "- Conceitos básicos g7",
+            "- Criar uma missão secreta",
+            "- Ser mentor 2 sessões",
+            "- Ter conta e saber usar o Slack",
+            "- Criar um Bot com uma API externa",
+            "- Última missão secreta"],
+        "color" : "A709F6"
+    },
+    "Preto" : {
+        "objectives" : [
+            "Não tens, és um guru da programação!"
+        ],
+        "color" : "000000"
+    }
+}

bot/cogs/belts.py

@@ -1,8 +1,40 @@
 from enum import Enum, unique
 
+import json
+
 import discord
 from discord.ext import commands
 
+class FileHandler():
+    file = "bot/cogs/belts.json"
+
+    def __init__(self, belt):
+        self.belt = belt
+        self.msg = self.get_info()[0]
+        self.color = self.get_info()[1]
+    
+    def get_info(self):
+        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]["objectives"]:
+                msg += '\n' + param
+
+            return (
+                msg,
+                color
+            )
+
+translator_to_emoji = {"Branco" : ":white_circle:",
+                       "Amarelo" : ":yellow_circle:",
+                       "Azul" : ":blue_circle:",
+                       "Verde" : ":green_circle:",
+                       "Laranja" : ":orange_circle:",
+                       "Vermelho" : ":red_circle:",
+                       "Roxo" : ":purple_circle:",
+                       "Preto" : ":black_circle:"}
+
 @unique
 class Belts(Enum):
     Branco = 1
@@ -40,7 +72,7 @@ def get_role_from_name(guild, belt):
         for role in guild.roles:
             if role.name == belt:
                 return role
-    
+
 class BeltsAttributions(commands.Cog):
     def __init__(self, client):
         self.client = client
@@ -62,11 +94,24 @@ async def promove(self, ctx, user, belt):
                 reason = None,
                 atomic = True
             )
-        
+
+            # Public message
             await ctx.send(
                 f'{user} agora és cinturão {belt} :tada:'
             )
-        
+            
+            # Private message 
+            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)
+
         elif belt == ninja.current_belt().name:
             await ctx.reply(
                 f"Esse já é o cinturão do ninja {user}!"
@@ -79,11 +124,24 @@ async def promove(self, ctx, user, belt):
                 reason = None,
                 atomic = True
             )
-        
+
+            # Public message
             await ctx.send(
                 f'{user} agora és cinturão {belt} :tada:'
             )
-        
+            
+            # Private message 
+            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)
+
         elif belt != ninja.next_belt().name:
             await ctx.send(
                 f'{user} esse cinturão não é valido de se ser atribuido.'