Jams: Move roles adding to another function from inside of command

1 files 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."""