Update item.py

exorde-labs · Oct 16, 2023 · dac9fcd · dac9fcd
1 parent d515ecd
commit dac9fcd
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/exorde/item.py b/exorde/item.py
@@ -84,10 +84,15 @@ async def consumer(
                 yield item
             else:
                 continue
-        except GeneratorExit:
+
+        except GeneratorExit:   
             raise
         except TimeoutError:
             raise GeneratorExit
+        except StopAsyncIteration:
+            logging.info(f"End of iterator {module.__name__} - StopAsyncIteration")
+        except asyncio.TimeoutError:
+            logging.info(f"End of iterator {module.__name__} - TimeoutError")
         except Exception as e:
             traceback_list = traceback.format_exception(
                 type(e), e, e.__traceback__