From 341b3141048a72b59593d17713bf492d5e115ac2 Mon Sep 17 00:00:00 2001
From: jesopo
Date: Sat, 25 Jan 2020 13:58:13 +0000
Subject: change command specs to be compiled at runtime by a decorator

---
 src/utils/decorators.py | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

(limited to 'src/utils/decorators.py')

diff --git a/src/utils/decorators.py b/src/utils/decorators.py
index dc4adf3a..ca47ea6e 100644
--- a/src/utils/decorators.py
+++ b/src/utils/decorators.py
@@ -1,4 +1,5 @@
 import typing
+from .parse import argument_spec
 
 BITBOT_MAGIC = "__bitbot"
 
@@ -42,10 +43,18 @@ def export(setting: str, value: typing.Any):
         magic.add_export(setting, value)
         return module
     return _export_func
+
+def _kwarg(key: str, value: typing.Any, func: typing.Any):
+    magic = get_magic(func)
+    magic.add_kwarg(key, value)
+    return func
+
 def kwarg(key: str, value: typing.Any):
     def _kwarg_func(func):
-        magic = get_magic(func)
-        magic.add_kwarg(key, value)
-        return func
+        return _kwarg(key, value, func)
     return _kwarg_func
 
+def spec(spec: str):
+    def _spec_func(func):
+        return _kwarg("spec", argument_spec(spec), func)
+    return _spec_func
-- 
cgit v1.3.1-10-gc9f91