diff --git a/src/commands.py b/src/commands.py index 3393c13..ca10643 100644 --- a/src/commands.py +++ b/src/commands.py @@ -173,14 +173,16 @@ def interval(update: Update, context: CallbackContext): name = context.args[0] if name not in database.userRequests[user]: return "*Error:* %s doesn't exist." % name + request = database.userRequests[user][name] # Validate the interval of the interval i = int(context.args[1]) if i < 40 or i > 60*60*24: return "*Error:* %s is too long or too short. (Min: 40s, Max: 60 * 60 * 24s)" % i - database.userRequests[user][name]['interval'] = i + request['interval'] = i database.save() + scheduler.updateInterval(user, name, request) return "Success!" diff --git a/src/scheduler.py b/src/scheduler.py index 08d4d0d..fb6988f 100644 --- a/src/scheduler.py +++ b/src/scheduler.py @@ -84,3 +84,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): + if not self.isStarted(user, name): + return False + + self.stop(user, name) + self.start(user, request) + return True