From 2a85707a42afb4adcc9b89034f6d34246c6b1553 Mon Sep 17 00:00:00 2001 From: brettlangdon Date: Tue, 9 Mar 2021 20:10:51 -0500 Subject: [PATCH] allow sysaudit.subscribe as a decorator --- sysaudit/__init__.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/sysaudit/__init__.py b/sysaudit/__init__.py index cae68e3..9dc5fec 100644 --- a/sysaudit/__init__.py +++ b/sysaudit/__init__.py @@ -96,7 +96,19 @@ def _subscription_hook(event, args): hook(args) -def subscribe(event, hook): +def subscribe(event, hook=None): + # No hook provided, return a decorator + # @sysaudit.subscribe("event.name") + # def event_name_hook(args): + # pass + if not hook: + + def decorator(fn): + subscribe(event, fn) + return fn + + return decorator + global _subscriptions global _subscription_hook_active