Log invocation of functions

trickeydan · Jul 11, 2021 · 15c5aee · 15c5aee
1 parent b76e06b
commit 15c5aee
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/mqtt_automate/mqtt_automate.py b/mqtt_automate/mqtt_automate.py
@@ -34,14 +34,16 @@ def run(self) -> None:
     def on_message(self, topic: str) -> Callable[[OnMessageHandler], OnMessageHandler]:
         """Register a topic to react to."""
         def decorator(func: OnMessageHandler) -> OnMessageHandler:
-            # Register handler
-            self._handlers[Topic.parse(topic)] = func
 
             async def wrapper(
                 engine: AutomationEngine,
                 match: Match[str],
                 payload: str,
             ) -> None:
+                LOGGER.info(f"INVOKE {topic} -> {func.__name__}")
                 await func(engine, match, payload)
+
+            # Register handler
+            self._handlers[Topic.parse(topic)] = wrapper
             return wrapper
         return decorator