From 8419531b899fbebb2a1f3378b4e1a98a0f45d812 Mon Sep 17 00:00:00 2001
From: ks129 <45097959+ks129@users.noreply.github.com>
Date: Thu, 11 Jun 2020 10:49:47 +0300
Subject: Jams: Move roles adding to another function from inside of command

---
 bot/cogs/jams.py | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

diff --git a/bot/cogs/jams.py b/bot/cogs/jams.py
index 5576adb2d..4173f10fd 100644
--- a/bot/cogs/jams.py
+++ b/bot/cogs/jams.py
@@ -42,14 +42,7 @@ class CodeJams(commands.Cog):
             return
 
         team_channel = await self.create_channels(ctx, team_name, members)
-
-        # Assign team leader role
-        await members[0].add_roles(ctx.guild.get_role(Roles.team_leaders))
-
-        # Assign rest of roles
-        jammer_role = ctx.guild.get_role(Roles.jammers)
-        for member in members:
-            await member.add_roles(jammer_role)
+        await self.add_roles(ctx, members)
 
         await ctx.send(
             f":ok_hand: Team created: {team_channel}\n"
@@ -127,6 +120,16 @@ class CodeJams(commands.Cog):
 
         return team_channel.mention
 
+    async def add_roles(self, ctx: commands.Context, members: t.List[Member]) -> None:
+        """Assign team leader and jammer roles."""
+        # Assign team leader role
+        await members[0].add_roles(ctx.guild.get_role(Roles.team_leaders))
+
+        # Assign rest of roles
+        jammer_role = ctx.guild.get_role(Roles.jammers)
+        for member in members:
+            await member.add_roles(jammer_role)
+
 
 def setup(bot: Bot) -> None:
     """Load the CodeJams cog."""
-- 
cgit v1.2.3