Refactor Enum Classes (#27)

* Ci changes

* re-refactor Enum class

Author: danielsp45

bot/cogs/belts.py

@@ -66,9 +66,9 @@ class Ninja:
     def __init__(self, guild: discord.Guild, member: discord.Member):
         self.guild = guild
         self.member = member
-        self.roles = [role for role in member.roles]
+        self.roles = list(member.roles)
 
-    def current_belt(self) -> Belts:
+    def current_belt(self):
         """This function returns the current belt of the ninja."""
 
         highest_belt = None
@@ -79,6 +79,8 @@ def current_belt(self) -> Belts:
 
         return highest_belt
 
+        return highest_belt
+
     def next_belt(self) -> Belts:
         """This function returns the next belt of the ninja."""
 
@@ -94,7 +96,9 @@ def __init__(self, client: commands.Bot):
         self.client = client
 
     @commands.command(name="promove")
-    @commands.has_any_role("🛡️ Admin", "🏆 Champion", "🧑‍🏫 Mentores")
+    @commands.has_any_role(
+        Roles["ADMIN"].name, Roles["CHAMPION"].name, Roles["MENTOR"].name
+    )
     async def promove(
         self, ctx: discord.ext.commands.Context, user: str, belt: str
     ) -> None:

bot/cogs/utils/constants.py

@@ -2,6 +2,7 @@
 
 import discord
 from discord.ext import commands
+from emoji import emojize
 
 translator_to_emoji = {
     "Branco": ":white_circle:",
@@ -14,17 +15,37 @@
     "Preto": ":black_circle:",
 }
 
+Roles = Enum(
+    value="Roles",
+    names=[
+        (emojize(":older_man: Guardiões"), 1),
+        ("GUARDIOES", 1),
+        (emojize(":ninja: Ninja"), 2),
+        ("NINJAS", 2),
+        (emojize(":person_raising_hand: Voluntários"), 3),
+        ("VOLUNTARIOS", 3),
+        (emojize(":teacher: Mentores"), 4),
+        ("MENTOR", 4),
+        (emojize(":shield: Admin"), 5),
+        ("ADMIN", 5),
+        (emojize(":trophy: Champion"), 6),
+        ("CHAMPION", 6),
+    ],
+)
 
-@unique
-class Belts(Enum):
-    Branco = 1
-    Amarelo = 2
-    Azul = 3
-    Verde = 4
-    Laranja = 5
-    Vermelho = 6
-    Roxo = 7
-    Preto = 8
+Belts = Enum(
+    value="Belts",
+    names=[
+        ("Branco", 1),
+        ("Amarelo", 2),
+        ("Azul", 3),
+        ("Verde", 4),
+        ("Laranja", 5),
+        ("Vermelho", 6),
+        ("Roxo", 7),
+        ("Preto", 8),
+    ],
+)
 
 
 def get_role_from_name(guild: discord.Guild, belt: str) -> discord.Role:

requirements.txt

@@ -1,4 +1,5 @@
 discord==1.0.1
 discord.py==1.7.3
+emoji==1.7.0
 python-dotenv==0.20.0
 SQLAlchemy==1.4.37