From cd2bcc811504dec6e31f29d9caac530e54253f2e Mon Sep 17 00:00:00 2001
From: Hykilpikonna <me@hydev.org>
Date: Fri, 27 Nov 2020 09:17:17 -0500
Subject: [PATCH] [+] Encapsulate Scheduler.stop

---
 src/scheduler.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/scheduler.py b/src/scheduler.py
index 2a8ae9b..dd0e1ea 100644
--- a/src/scheduler.py
+++ b/src/scheduler.py
@@ -49,7 +49,7 @@ class Scheduler:
 
         return task
 
-    def startTask(self, user: str, request):
+    def start(self, user: str, request):
         name = request['name']
         if user not in self.tasks:
             self.tasks[user] = {}
@@ -61,3 +61,12 @@ class Scheduler:
             request['enabled'] = True
             self.database.save()
 
+    def stop(self, user: str, name: str):
+        # Stop and remove task
+        job = self.tasks[user][name]
+        job.enabled = False
+        job.schedule_removal()
+        self.tasks[user].pop(name, None)
+        self.database.userRequests[user][name]['enabled'] = False
+        self.database.save()
+