Explorar el Código

Fixing problem with unit role assignment based on department units list.

BaumSplitter41 hace 4 meses
padre
commit
af46b3be3f
Se han modificado 1 ficheros con 4 adiciones y 7 borrados
  1. 4 7
      VPD_BOT/cogs/unit.py

+ 4 - 7
VPD_BOT/cogs/unit.py

@@ -66,19 +66,16 @@ class unit(commands.Cog):
 
 
 #Command implementation
 #Command implementation
         if unit == "Detective":
         if unit == "Detective":
-            unit_role = int(units[0])
+            unit_role = units[0]
         elif unit == "SWAT":
         elif unit == "SWAT":
-            unit_role = int(units[1])
+            unit_role = units[1]
         elif unit == "Canine":
         elif unit == "Canine":
-            unit_role = int(units[2])
+            unit_role = units[2]
         elif unit == "Air Support":
         elif unit == "Air Support":
-            unit_role = int(units[3])
+            unit_role = units[3]
         else:
         else:
             await ctx.respond("The selected unit does not exist!", ephemeral=True)
             await ctx.respond("The selected unit does not exist!", ephemeral=True)
             return
             return
-        
-
-        unit_role = server.get_role(unit_role)
         if ctx.author == supervisor_role or ctx.author == head_unit_role:
         if ctx.author == supervisor_role or ctx.author == head_unit_role:
             if unit_role in user.roles:
             if unit_role in user.roles:
                 await user.remove_roles(unit_role)
                 await user.remove_roles(unit_role)