diff --git a/sopel_SpiceBot_Core_Prerun/__init__.py b/sopel_SpiceBot_Core_Prerun/__init__.py index 20c1d61..9c8a0cd 100644 --- a/sopel_SpiceBot_Core_Prerun/__init__.py +++ b/sopel_SpiceBot_Core_Prerun/__init__.py @@ -104,7 +104,7 @@ def get_commands(bot, trigger): return commands -def prerun(): +def dispatch(): def actual_decorator(function): @@ -124,3 +124,17 @@ def prerun(): return internal_prerun return actual_decorator + + +@dispatch() +def prerun(): + + def actual_decorator(function): + + @functools.wraps(function) + def internal_prerun(bot, trigger, *args, **kwargs): + + function(bot, trigger, *args, **kwargs) + + return internal_prerun + return actual_decorator