From 25e6bfa9004e19f42ef084439f87acc97865c4f9 Mon Sep 17 00:00:00 2001 From: jesopo Date: Fri, 31 Aug 2018 13:39:40 +0100 Subject: [PATCH] Support replay keyword in EventHookContext.hook --- EventManager.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/EventManager.py b/EventManager.py index 0d4d1dd1..4ca5837c 100644 --- a/EventManager.py +++ b/EventManager.py @@ -57,8 +57,10 @@ class EventHookContext(object): def __init__(self, parent, context): self._parent = parent self.context = context - def hook(self, function, priority=DEFAULT_PRIORITY, **kwargs): - self._parent._context_hook(self.context, function, priority, kwargs) + def hook(self, function, priority=DEFAULT_PRIORITY, replay=False, + **kwargs): + self._parent._context_hook(self.context, function, priority, replay, + kwargs) def on(self, subevent, *extra_subevents, delimiter=DEFAULT_DELIMITER): return self._parent._context_on(self.context, subevent, extra_subevents, delimiter) @@ -103,8 +105,8 @@ class EventHook(object): def hook(self, function, priority=DEFAULT_PRIORITY, replay=False, **kwargs): self._hook(function, None, priority, replay, kwargs) - def _context_hook(self, context, function, priority, kwargs): - self._hook(function, context, priority, False, kwargs) + def _context_hook(self, context, function, priority, replay, kwargs): + self._hook(function, context, priority, replay, kwargs) def _hook(self, function, context, priority, replay, kwargs): callback = EventCallback(function, self.bot, priority, kwargs)