Added exception_propogation parameter.

Signed-off-by: Pavel Kirilin <win10@list.ru>

Author: s3rius

taskiq_dependencies/ctx.py

@@ -20,11 +20,13 @@ def __init__(
         self,
         graph: "DependencyGraph",
         initial_cache: Optional[Dict[Any, Any]] = None,
+        exception_propogation: bool = True,
     ) -> None:
         self.graph = graph
         self.opened_dependencies: List[Any] = []
         self.sub_contexts: "List[Any]" = []
         self.initial_cache = initial_cache or {}
+        self.propogate_excs = exception_propogation
 
     def traverse_deps(  # noqa: C901, WPS210
         self,
@@ -131,7 +133,7 @@ def close(self, *args: Any) -> None:  # noqa: C901
         :param args: exception info if any.
         """
         exception_found = False
-        if args[1] is not None:
+        if args[1] is not None and self.propogate_excs:
             exception_found = True
         for ctx in self.sub_contexts:
             ctx.close(*args)
@@ -232,7 +234,7 @@ async def close(self, *args: Any) -> None:  # noqa: C901
         :param args: exception info if any.
         """
         exception_found = False
-        if args[1] is not None:
+        if args[1] is not None and self.propogate_excs:
             exception_found = True
         for ctx in self.sub_contexts:
             await ctx.close(*args)  # type: ignore

taskiq_dependencies/graph.py

@@ -41,35 +41,43 @@ def is_empty(self) -> bool:
     def async_ctx(
         self,
         initial_cache: Optional[Dict[Any, Any]] = None,
+        exception_propogation: bool = True,
     ) -> AsyncResolveContext:
         """
         Create dependency resolver context.
 
         This context is used to actually resolve dependencies.
 
         :param initial_cache: initial cache dict.
+        :param exception_propogation: If true, all found errors within
+            context will be propogated to dependencies.
         :return: new resolver context.
         """
         return AsyncResolveContext(
             self,
             initial_cache,
+            exception_propogation,
         )
 
     def sync_ctx(
         self,
         initial_cache: Optional[Dict[Any, Any]] = None,
+        exception_propogation: bool = True,
     ) -> SyncResolveContext:
         """
         Create dependency resolver context.
 
         This context is used to actually resolve dependencies.
 
         :param initial_cache: initial cache dict.
+        :param exception_propogation: If true, all found errors within
+            context will be propogated to dependencies.
         :return: new resolver context.
         """
         return SyncResolveContext(
             self,
             initial_cache,
+            exception_propogation,
         )
 
     def _build_graph(self) -> None:  # noqa: C901, WPS210

tests/test_graph.py

@@ -382,7 +382,7 @@ def target(
 
 
 @pytest.mark.anyio
-async def test_async_exception_generators_no_propogation() -> None:
+async def test_async_exception_in_teardown() -> None:
 
     errors_found = 0
 
@@ -400,3 +400,50 @@ def target(_: int = Depends(my_generator)) -> None:
     with pytest.raises(ValueError):
         async with DependencyGraph(target=target).async_ctx() as g:
             target(**(await g.resolve_kwargs()))
+
+
+@pytest.mark.anyio
+async def test_async_propogation_disabled() -> None:
+
+    errors_found = 0
+
+    async def my_generator() -> AsyncGenerator[int, None]:
+        nonlocal errors_found
+        try:
+            yield 1
+        except ValueError:
+            errors_found += 1
+            raise Exception()
+
+    def target(_: int = Depends(my_generator)) -> None:
+        raise ValueError()
+
+    with pytest.raises(ValueError):
+        async with DependencyGraph(target=target).async_ctx(
+            exception_propogation=False,
+        ) as g:
+            target(**(await g.resolve_kwargs()))
+
+    assert errors_found == 0
+
+
+def test_sync_propogation_disabled() -> None:
+
+    errors_found = 0
+
+    def my_generator() -> Generator[int, None, None]:
+        nonlocal errors_found
+        try:
+            yield 1
+        except ValueError:
+            errors_found += 1
+            raise Exception()
+
+    def target(_: int = Depends(my_generator)) -> None:
+        raise ValueError()
+
+    with pytest.raises(ValueError):
+        with DependencyGraph(target=target).sync_ctx(exception_propogation=False) as g:
+            target(**(g.resolve_kwargs()))
+
+    assert errors_found == 0