diff options
| -rw-r--r-- | src/EventManager.py | 15 |
1 files changed, 8 insertions, 7 deletions
diff --git a/src/EventManager.py b/src/EventManager.py index a9ee6066..d5169f17 100644 --- a/src/EventManager.py +++ b/src/EventManager.py @@ -47,11 +47,11 @@ class EventHook(object): self.log = log self.name = name self.parent = parent - self._children = {} - self._hooks = [] + self._children = {} # type: typing.Dict[str, EventHook] + self._hooks = [] # type: typing.List[EventCallback] self._replayed = False - self._stored_events = [] - self._context_hooks = {} + self._stored_events = [] # type: typing.List[typing.Dict] + self._context_hooks = {} # type: typing.Dict[str, typing.List[EventCallback]] def _make_event(self, kwargs: dict) -> Event: return Event(self._get_path(), **kwargs) @@ -84,9 +84,10 @@ class EventHook(object): if context == None: self._hooks.append(callback) else: + context_str = typing.cast(str, context) if not context in self._context_hooks: - self._context_hooks[context] = [] - self._context_hooks[context].append(callback) + self._context_hooks[context_str] = [] + self._context_hooks[context_str].append(callback) if replay and not self._replayed: for kwargs in self._stored_events: @@ -94,7 +95,7 @@ class EventHook(object): self._replayed = True return callback - def unhook(self, callback: "EventHook"): + def unhook(self, callback: "EventCallback"): if callback in self._hooks: self._hooks.remove(callback) |