1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
from collections import defaultdict
class Hookable:
_hooks = None
def __init__(self):
self._hooks = defaultdict(set)
def register_hook(self, hook_type, hook):
self._hooks[hook_type].add(hook)
def execute_hook(self, hook_type, *args, **kwargs):
for hook in self._hooks[hook_type]:
hook(self, *args, **kwargs)
def hooked(func=None, before=None, after=None):
"""Execute hooks before and after the decorated method."""
def decorator(func):
before_name = before or f"before_{func.__name__}"
after_name = after or f"after_{func.__name__}"
def wrapper(self, *args, **kwargs):
self.execute_hook(before_name, *args, **kwargs)
result = func(self, *args, **kwargs)
self.execute_hook(after_name, result)
return result
return wrapper
# The decorator has been called without parenthesis
if callable(func):
return decorator(func)
return decorator
|
