Callback must return a generator.

PatrickRopp · Jun 24, 2016 · c6ae03f · c6ae03f
1 parent 21660fd
commit c6ae03f
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/inline_requests/decorator.py b/src/inline_requests/decorator.py
@@ -85,10 +85,9 @@ def __init__(self, callback, **kwargs):
 
     def __call__(self, response):
         output = iterate_spider_output(self.callback(response=response, **self.kwargs))
-        if isinstance(output, types.GeneratorType):
-            return self._unwindGenerator(output)
-        else:
-            return output
+        if not isinstance(output, types.GeneratorType):
+            raise ValueError("Callback must return a generator type")
+        return self._unwindGenerator(output)
 
     def _unwindGenerator(self, generator, _prev=None):
         while True: