From 70db97f64e0d6dbdd60f4d6d147d11c5b6b6a204 Mon Sep 17 00:00:00 2001
From: jesopo
Date: Wed, 19 Feb 2020 17:22:37 +0000
Subject: support @utils.export on functions, to export those functions

---
 src/utils/decorators.py | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

(limited to 'src/utils')

diff --git a/src/utils/decorators.py b/src/utils/decorators.py
index ca47ea6e..855d2b2e 100644
--- a/src/utils/decorators.py
+++ b/src/utils/decorators.py
@@ -37,12 +37,11 @@ def hook(event: str, **kwargs):
         magic.add_hook(event, kwargs)
         return func
     return _hook_func
-def export(setting: str, value: typing.Any):
-    def _export_func(module):
-        magic = get_magic(module)
-        magic.add_export(setting, value)
-        return module
-    return _export_func
+def export(setting: str, value: typing.Any=None):
+    def _export(obj: typing.Any):
+        get_magic(obj).add_export(setting, value)
+        return obj
+    return _export
 
 def _kwarg(key: str, value: typing.Any, func: typing.Any):
     magic = get_magic(func)
-- 
cgit v1.3.1-10-gc9f91