Fixed generator decorator

abots/helpers/black_magic.py

@@ -29,6 +29,9 @@ def coroutine(func):
 def generator(func):
     @wraps(func)
     def wrapper_generator(*args, **kwargs):
-        yield from func(*args, **kwargs)
+        try:
+            yield from func(*args, **kwargs)
+        except GeneratorExit:
+            pass
     return wrapper_generator