Added generator decorator

abots/helpers/black_magic.py

@@ -5,15 +5,6 @@ def infinitedict():
     d = lambda: defaultdict(d)
     return defaultdict(d)
 
-# NOTE - Is this useful? I have no idea, but here it is!
-def decorator(wrapper):
-    def wrapper_decorator(func):
-        @wraps(func)
-        def wrapper_wrapper(*args, **kwargs):
-            return wrapper(*args, **kwargs)
-        return wrapper_wrapper
-    return wrapper_decorator
-
 def debugger(func):
     @wraps(func)
     def wrapper_debug(*args, **kwargs):
@@ -33,3 +24,11 @@ def coroutine(func):
         coro.send(None)
         return coro
     return wrapper_coroutine
+
+@coroutine
+def generator(func):
+    @wraps(func)
+    def wrapper_generator(*args, **kwargs):
+        yield from func(*args, **kwargs)
+    return wrapper_generator
+