diff --git a/src/commands.py b/src/commands.py index a5ba99b..2911983 100644 --- a/src/commands.py +++ b/src/commands.py @@ -171,7 +171,7 @@ def interval(update: Update, context: CallbackContext): request['interval'] = i database.save() - scheduler.updateInterval(user, name, request) + scheduler.updateInterval(user, name) return "Success!" diff --git a/src/scheduler.py b/src/scheduler.py index 0da96ac..2af674f 100644 --- a/src/scheduler.py +++ b/src/scheduler.py @@ -48,12 +48,11 @@ class Scheduler: return task - def start(self, user: str, request): - name = request['name'] - + def start(self, user: str, name: str): if self.isStarted(user, name): return False + request = self.database.reqs[user][name] if user not in self.tasks: self.tasks[user] = {} @@ -83,10 +82,10 @@ class Scheduler: def isStarted(self, user: str, name: str): return user in self.tasks and name in self.tasks[user] - def updateInterval(self, user: str, name: str, request): + def updateInterval(self, user: str, name: str): if not self.isStarted(user, name): return False self.stop(user, name) - self.start(user, request) + self.start(user, name) return True