add protection if hook returns none

1 files changed, 5 insertions, 1 deletions

diff --git a/climl/__init__.py b/climl/__init__.py
index bfbe4fc..147dfc3 100644
--- a/climl/__init__.py
+++ b/climl/__init__.py
@@ -16,7 +16,11 @@ def on_email(content, hooks, conf):
     try:
         for hook in hooks:
             print('on_email, calling hook {}'.format(hook.__name__))
-            context, content = hook(context, content)
+            context2, content2 = hook(context, content)
+            if context2:
+                context = context2
+            if content2:
+                content = content2
     except interface.HookStopIteration:
         pass
     print('now, raising exception')