Add force option to Dragonfly stop command (#64)

* Add option to specify force task shutdown * Bump version and update changelog * Lint * Update docs/source/changelog.rst Co-authored-by: Bradley Reynolds <[email protected]> Signed-off-by: Robin <[email protected]> --------- Signed-off-by: Robin <[email protected]> Co-authored-by: Bradley Reynolds <[email protected]>
vipyrsec · May 5, 2023 · 7caf47c · 7caf47c
1 parent 3fa2df4
commit 7caf47c
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 4 deletions.
diff --git a/docs/source/changelog.rst b/docs/source/changelog.rst
@@ -4,6 +4,9 @@
 Changelog
 =========
 
+- :release:`2.2.0 <5th May 2023>`
+- :feature:`57` Add force option to Dragonfly stop command
+
 - :release:`2.1.0 <3rd May 2023>`
 - :feature:`59` Add slash command to bring up package modal on demand
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "alessia"
-version = "2.1.0"
+version = "2.2.0"
 dependencies = [
     "python-dotenv==1.0.0",
     "discord.py==2.2.2",

diff --git a/src/bot/exts/dragonfly/dragonfly.py b/src/bot/exts/dragonfly/dragonfly.py
@@ -301,10 +301,14 @@ async def start(self, ctx: commands.Context) -> None:
             await ctx.send("Started task...")
 
     @commands.command()
-    async def stop(self, ctx: commands.Context) -> None:
+    async def stop(self, ctx: commands.Context, force: bool = False) -> None:
         if self.scan_loop.is_running():
-            self.scan_loop.stop()
-            await ctx.send("Stopping task...")
+            if force:
+                self.scan_loop.cancel()
+                await ctx.send("Forcing shutdown...")
+            else:
+                self.scan_loop.stop()
+                await ctx.send("Executing graceful shutdown...")
         else:
             await ctx.send("Task is not running.")