From cf34230a7a98db331a452cd186905a7c3c9721ac Mon Sep 17 00:00:00 2001 From: Kyriog Date: Sat, 4 Aug 2018 01:23:00 +0200 Subject: [PATCH] Adding kwargs to scheduled functions (#104) * Adding kwargs to scheduled functions * Using kwargs as a dict for register_schedule --- disco/bot/plugin.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/disco/bot/plugin.py b/disco/bot/plugin.py index 8422cb7..a5e589e 100644 --- a/disco/bot/plugin.py +++ b/disco/bot/plugin.py @@ -379,7 +379,7 @@ class Plugin(LoggingClass, PluginDeco): """ self.commands.append(Command(self, func, *args, **kwargs)) - def register_schedule(self, func, interval, repeat=True, init=True): + def register_schedule(self, func, interval, repeat=True, init=True, kwargs=None): """ Registers a function to be called repeatedly, waiting for an interval duration. @@ -395,14 +395,19 @@ class Plugin(LoggingClass, PluginDeco): init : bool Whether to run this schedule once immediatly, or wait for the first scheduled iteration. + kwargs: dict + kwargs which will be passed to executed `func` """ + if kwargs is None: + kwargs = {} + def repeat_func(): if init: - func() + func(**kwargs) while True: gevent.sleep(interval) - func() + func(**kwargs) if not repeat: break