remove inmogi roles on close

probablyjassin · Nov 10, 2024 · 95ec141 · 95ec141
1 parent fd6f39b
commit 95ec141
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/cogs/mogi/calculations.py b/cogs/mogi/calculations.py
@@ -141,6 +141,10 @@ async def apply(self, ctx: MogiApplicationContext):
         await update_roles(ctx, ctx.mogi)
 
         ctx.mogi.finish()
+        for player in ctx.mogi.players:
+            (await ctx.guild.fetch_member(player.discord_id)).remove_roles(
+                ctx.inmogi_role
+            )
         mogi_manager.destroy_mogi(ctx.channel.id)
         return await ctx.respond("# This channel's Mogi is finished and closed.")
 

diff --git a/cogs/mogi/mogi.py b/cogs/mogi/mogi.py
@@ -29,7 +29,12 @@ async def close(self, ctx: MogiApplicationContext):
         )
 
         if await confirmation(ctx, close_confirm_message):
-            mogi_manager.destroy_mogi(ctx.channel.id)
+            for player in ctx.mogi.players:
+                (await ctx.guild.fetch_member(player.discord_id)).remove_roles(
+                    ctx.inmogi_role
+                )
+
+            mogi_manager.destroy_mogi(ctx.mogi.channel_id)
             return await ctx.respond("# This channel's Mogi has been closed.")
 
         await ctx.respond("Cancelled.")