Update ampremover.py

BenCos17 · Dec 6, 2024 · 5774164 · 5774164
1 parent c47e5c1
commit 5774164
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/ampremover/ampremover.py b/ampremover/ampremover.py
@@ -23,12 +23,20 @@ async def amputator(self, ctx):
     @amputator.command(name='optin')
     async def opt_in(self, ctx):
         """Opt-in to use the AmputatorBot service"""
-        await ctx.send("This command cannot be used in DMs.")
+        if ctx.guild:
+            await self.config.guild(ctx.guild).opted_in.set(True)
+            await ctx.send("Successfully opted in to use the AmputatorBot service.")
+        else:
+            await ctx.send("This command cannot be used in DMs.")
 
     @amputator.command(name='optout')
     async def opt_out(self, ctx):
         """Opt-out from using the AmputatorBot service"""
-        await ctx.send("This command cannot be used in DMs.")
+        if ctx.guild:
+            await self.config.guild(ctx.guild).opted_in.set(False)
+            await ctx.send("Successfully opted out from using the AmputatorBot service.")
+        else:
+            await ctx.send("This command cannot be used in DMs.")
 
     @amputator.command(name='convert')
     async def convert_amp(self, ctx, *, message: str):