Fixing Decorators (#146)

* Fixing Decorators Minor bug fixes to getargs and getargsorask so that the return of the wrapped function isn't consumed
LanceMaverick · Jun 13, 2017 · 18225d9 · 18225d9
1 parent a836ad4
commit 18225d9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/skybeard/decorators/getargs.py b/skybeard/decorators/getargs.py
@@ -21,9 +21,9 @@ async def g(beard, msg):
             args = get_args(msg, return_string=return_string)
             try:
                 if isinstance(args, str):
-                    await f(beard, msg, args)
+                    return await f(beard, msg, args)
                 else:
-                    await f(beard, msg, *args)
+                    return await f(beard, msg, *args)
             except TypeError as exception:
                 # TODO have configurable messages when args are not enough
                 err_msg = exception.args[0]

diff --git a/skybeard/decorators/getargsorask.py b/skybeard/decorators/getargsorask.py
@@ -25,6 +25,6 @@ async def g(beard, msg):
                     resp = await beard.listener.wait()
                     kwargs[var] = resp['text']
 
-            await f(beard, msg, **kwargs)
+            return await f(beard, msg, **kwargs)
         return g
     return _getargsorask_wrapper