[O] Better null safety

This commit is contained in:
Hykilpikonna
2020-11-27 09:21:52 -05:00
parent 26c9d5d943
commit b98134ea23
2 changed files with 14 additions and 8 deletions
+11
View File
@@ -51,6 +51,10 @@ class Scheduler:
def start(self, user: str, request):
name = request['name']
if self.isStarted(user, name):
return False
if user not in self.tasks:
self.tasks[user] = {}
@@ -61,7 +65,12 @@ class Scheduler:
request['enabled'] = True
self.database.save()
return True
def stop(self, user: str, name: str):
if not self.isStarted(user, name):
return False
# Stop and remove task
job = self.tasks[user][name]
job.enabled = False
@@ -70,6 +79,8 @@ class Scheduler:
self.database.userRequests[user][name]['enabled'] = False
self.database.save()
return True
def isStarted(self, user: str, name: str):
return user in self.tasks and name in self.tasks[user]