haiway 0.13.1__tar.gz → 0.15.0__tar.gz

{haiway-0.13.1 → haiway-0.15.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: haiway
-Version: 0.13.1
+Version: 0.15.0
 Summary: Framework for dependency injection and state management within structured concurrency model.
 Project-URL: Homepage, https://miquido.com
 Project-URL: Repository, https://github.com/miquido/haiway.git

haiway-0.15.0/junit/test-results.xml

@@ -0,0 +1 @@
+<?xml version="1.0" encoding="utf-8"?><testsuites><testsuite name="pytest" errors="0" failures="0" skipped="0" tests="88" time="1.118" timestamp="2025-04-02T07:36:48.794161" hostname="fv-az1784-745"><testcase classname="tests.test_async_queue" name="test_fails_when_stream_fails" time="0.001" /><testcase classname="tests.test_async_queue" name="test_cancels_when_iteration_cancels" time="0.001" /><testcase classname="tests.test_async_queue" name="test_ends_when_stream_ends" time="0.001" /><testcase classname="tests.test_async_queue" name="test_buffers_values_when_not_reading" time="0.001" /><testcase classname="tests.test_async_queue" name="test_delivers_buffer_when_streaming_fails" time="0.001" /><testcase classname="tests.test_async_queue" name="test_delivers_updates_when_sending" time="0.001" /><testcase classname="tests.test_async_queue" name="test_fails_when_sending_to_finished" time="0.001" /><testcase classname="tests.test_async_queue" name="test_ignores_when_finishing_when_finished" time="0.001" /><testcase classname="tests.test_async_stream" name="test_fails_when_stream_fails" time="0.001" /><testcase classname="tests.test_async_stream" name="test_cancels_when_iteration_cancels" time="0.001" /><testcase classname="tests.test_async_stream" name="test_ends_when_stream_ends" time="0.001" /><testcase classname="tests.test_async_stream" name="test_finishes_without_buffer" time="0.001" /><testcase classname="tests.test_async_stream" name="test_fails_without_buffer" time="0.001" /><testcase classname="tests.test_async_stream" name="test_delivers_updates_when_sending" time="0.001" /><testcase classname="tests.test_async_stream" name="test_ignores_when_sending_to_finished" time="0.001" /><testcase classname="tests.test_async_stream" name="test_ignores_when_sending_to_failed" time="0.001" /><testcase classname="tests.test_async_stream" name="test_ignores_when_finishing_when_finished" time="0.001" /><testcase classname="tests.test_async_stream" name="test_delivers_all_when_sending_async" time="0.001" /><testcase classname="tests.test_attribute_path" name="test_id_path_points_to_self" time="0.000" /><testcase classname="tests.test_attribute_path" name="test_attribute_path_points_to_attribute" time="0.000" /><testcase classname="tests.test_attribute_path" name="test_nested_attribute_path_points_to_nested_attribute" time="0.000" /><testcase classname="tests.test_attribute_path" name="test_recursive_attribute_path_points_to_attribute" time="0.000" /><testcase classname="tests.test_attribute_path" name="test_list_item_path_points_to_item" time="0.000" /><testcase classname="tests.test_attribute_path" name="test_tuple_item_path_points_to_item" time="0.000" /><testcase classname="tests.test_attribute_path" name="test_mixed_tuple_item_path_points_to_item" time="0.000" /><testcase classname="tests.test_attribute_path" name="test_dict_item_path_points_to_item" time="0.000" /><testcase classname="tests.test_attribute_path" name="test_id_path_set_updates_self" time="0.000" /><testcase classname="tests.test_attribute_path" name="test_attribute_path_set_updates_attribute" time="0.000" /><testcase classname="tests.test_attribute_path" name="test_nested_attribute_path_set_updates_nested_attribute" time="0.000" /><testcase classname="tests.test_attribute_path" name="test_recursive_attribute_set_updates_attribute" time="0.000" /><testcase classname="tests.test_attribute_path" name="test_list_item_path_set_updates_item" time="0.001" /><testcase classname="tests.test_attribute_path" name="test_tuple_item_path_set_updates_item" time="0.000" /><testcase classname="tests.test_attribute_path" name="test_mixed_tuple_item_set_updates_item" time="0.000" /><testcase classname="tests.test_attribute_path" name="test_dict_item_path_set_updates_item" time="0.000" /><testcase classname="tests.test_auto_retry" name="test_returns_value_without_errors" time="0.001" /><testcase classname="tests.test_auto_retry" name="test_retries_with_errors" time="0.001" /><testcase classname="tests.test_auto_retry" name="test_logs_issue_with_errors" time="0.001" /><testcase classname="tests.test_auto_retry" name="test_fails_with_exceeding_errors" time="0.001" /><testcase classname="tests.test_auto_retry" name="test_fails_with_cancellation" time="0.001" /><testcase classname="tests.test_auto_retry" name="test_retries_with_selected_errors" time="0.001" /><testcase classname="tests.test_auto_retry" name="test_fails_with_not_selected_errors" time="0.001" /><testcase classname="tests.test_auto_retry" name="test_async_returns_value_without_errors" time="0.001" /><testcase classname="tests.test_auto_retry" name="test_async_retries_with_errors" time="0.001" /><testcase classname="tests.test_auto_retry" name="test_async_fails_with_exceeding_errors" time="0.001" /><testcase classname="tests.test_auto_retry" name="test_async_fails_with_cancellation" time="0.001" /><testcase classname="tests.test_auto_retry" name="test_async_fails_when_cancelled" time="0.021" /><testcase classname="tests.test_auto_retry" name="test_async_uses_delay_with_errors" time="0.102" /><testcase classname="tests.test_auto_retry" name="test_async_uses_computed_delay_with_errors" time="0.107" /><testcase classname="tests.test_auto_retry" name="test_async_logs_issue_with_errors" time="0.001" /><testcase classname="tests.test_auto_retry" name="test_async_retries_with_selected_errors" time="0.001" /><testcase classname="tests.test_auto_retry" name="test_async_fails_with_not_selected_errors" time="0.001" /><testcase classname="tests.test_cache" name="test_returns_cache_value_with_same_argument" time="0.001" /><testcase classname="tests.test_cache" name="test_returns_fresh_value_with_different_argument" time="0.000" /><testcase classname="tests.test_cache" name="test_returns_fresh_value_with_limit_exceed" time="0.000" /><testcase classname="tests.test_cache" name="test_returns_same_value_with_repeating_argument" time="0.000" /><testcase classname="tests.test_cache" name="test_fails_with_error" time="0.000" /><testcase classname="tests.test_cache" name="test_returns_fresh_value_with_expiration_time_exceed" time="0.021" /><testcase classname="tests.test_cache" name="test_async_returns_cache_value_with_same_argument" time="0.001" /><testcase classname="tests.test_cache" name="test_async_returns_fresh_value_with_different_argument" time="0.001" /><testcase classname="tests.test_cache" name="test_async_returns_fresh_value_with_limit_exceed" time="0.001" /><testcase classname="tests.test_cache" name="test_async_returns_same_value_with_repeating_argument" time="0.001" /><testcase classname="tests.test_cache" name="test_async_returns_fresh_value_with_expiration_time_exceed" time="0.021" /><testcase classname="tests.test_cache" name="test_async_cancel_waiting_does_not_cancel_task" time="0.502" /><testcase classname="tests.test_cache" name="test_async_expiration_does_not_cancel_task" time="0.021" /><testcase classname="tests.test_cache" name="test_async_fails_with_error" time="0.001" /><testcase classname="tests.test_context" name="test_state_is_available_according_to_context" time="0.001" /><testcase classname="tests.test_context" name="test_state_update_updates_local_context" time="0.001" /><testcase classname="tests.test_context" name="test_exceptions_are_propagated" time="0.001" /><testcase classname="tests.test_state" name="test_basic_initializes_with_arguments" time="0.002" /><testcase classname="tests.test_state" name="test_basic_initializes_with_defaults" time="0.001" /><testcase classname="tests.test_state" name="test_basic_equals_checks_properties" time="0.000" /><testcase classname="tests.test_state" name="test_basic_initializes_with_arguments_and_defaults" time="0.001" /><testcase classname="tests.test_state" name="test_parametrized_initializes_with_proper_parameters" time="0.000" /><testcase classname="tests.test_state" name="test_nested_initializes_with_proper_arguments" time="0.001" /><testcase classname="tests.test_state" name="test_dict_skips_missing_properties" time="0.001" /><testcase classname="tests.test_state" name="test_initialization_allows_missing_properties" time="0.001" /><testcase classname="tests.test_state" name="test_generic_subtypes_validation" time="0.002" /><testcase classname="tests.test_state" name="test_copying_leaves_same_object" time="0.001" /><testcase classname="tests.test_streaming" name="test_fails_when_generator_fails" time="0.001" /><testcase classname="tests.test_streaming" name="test_cancels_when_iteration_cancels" time="0.001" /><testcase classname="tests.test_streaming" name="test_ends_when_generator_ends" time="0.001" /><testcase classname="tests.test_streaming" name="test_delivers_updates_when_generating" time="0.001" /><testcase classname="tests.test_streaming" name="test_streaming_context_variables_access_is_preserved" time="0.001" /><testcase classname="tests.test_streaming" name="test_nested_streaming_streams_correctly" time="0.001" /><testcase classname="tests.test_timeout" name="test_returns_result_when_returning_value" time="0.001" /><testcase classname="tests.test_timeout" name="test_raises_with_error" time="0.001" /><testcase classname="tests.test_timeout" name="test_raises_with_cancel" time="0.011" /><testcase classname="tests.test_timeout" name="test_raises_with_timeout" time="0.012" /></testsuite></testsuites>

{haiway-0.13.1 → haiway-0.15.0}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "hatchling.build"
 [project]
 name = "haiway"
 description = "Framework for dependency injection and state management within structured concurrency model."
-version = "0.13.1"
+version = "0.15.0"
 readme = "README.md"
 maintainers = [
   { name = "Kacper Kaliński", email = "kacper.kalinski@miquido.com" },
@@ -44,6 +44,9 @@ extend-exclude = [".venv", ".git", ".cache"]
 lint.select = ["E", "F", "A", "I", "B", "PL", "W", "C", "RUF", "UP"]
 lint.ignore = ["A005"]
 
+[tool.ruff.lint.pylint]
+max-args = 12
+
 [tool.ruff.lint.per-file-ignores]
 "__init__.py" = ["F401", "E402"]
 "./tests/*.py" = ["PLR2004"]

{haiway-0.13.1 → haiway-0.15.0}/src/haiway/__init__.py

@@ -40,6 +40,7 @@ from haiway.types import (
 )
 from haiway.utils import (
     AsyncQueue,
+    AsyncStream,
     always,
     as_dict,
     as_list,
@@ -63,6 +64,7 @@ __all__ = [
     "MISSING",
     "ArgumentsTrace",
     "AsyncQueue",
+    "AsyncStream",
     "AttributePath",
     "AttributeRequirement",
     "Default",

{haiway-0.13.1 → haiway-0.15.0}/src/haiway/context/access.py

@@ -1,6 +1,7 @@
 from asyncio import (
     CancelledError,
     Task,
+    TaskGroup,
     current_task,
     iscoroutinefunction,
 )
@@ -11,7 +12,6 @@ from collections.abc import (
     Coroutine,
     Iterable,
 )
-from contextvars import Context, copy_context
 from logging import Logger
 from types import TracebackType
 from typing import Any, final, overload
@@ -24,6 +24,7 @@ from haiway.context.state import StateContext
 from haiway.context.tasks import TaskGroupContext
 from haiway.state import State
 from haiway.utils import mimic_function
+from haiway.utils.stream import AsyncStream
 
 __all__ = [
     "ctx",
@@ -37,7 +38,6 @@ class ScopeContext:
         "_identifier",
         "_logger_context",
         "_metrics_context",
-        "_state",
         "_state_context",
         "_task_group_context",
     )
@@ -46,6 +46,7 @@ class ScopeContext:
         self,
         label: str,
         logger: Logger | None,
+        task_group: TaskGroup | None,
         state: tuple[State, ...],
         disposables: Disposables | None,
         metrics: MetricsHandler | None,
@@ -65,15 +66,22 @@ class ScopeContext:
                 logger=logger,
             ),
         )
-        # postponing task group creation to include only when needed
-        self._task_group_context: TaskGroupContext
-        # postponing state creation to include disposables state when prepared
+        self._task_group_context: TaskGroupContext | None
+        object.__setattr__(
+            self,
+            "_task_group_context",
+            TaskGroupContext(
+                task_group=task_group,
+            )
+            if task_group is not None or self._identifier.is_root
+            else None,
+        )
+        # prepare state context to capture current state
         self._state_context: StateContext
-        self._state: tuple[State, ...]
         object.__setattr__(
             self,
-            "_state",
-            state,
+            "_state_context",
+            StateContext.updated(state),
         )
         self._disposables: Disposables | None
         object.__setattr__(
@@ -112,15 +120,12 @@ class ScopeContext:
         )
 
     def __enter__(self) -> str:
+        assert (  # nosec: B101
+            self._task_group_context is None or self._identifier.is_root
+        ), "Can't enter synchronous context with task group"
         assert self._disposables is None, "Can't enter synchronous context with disposables"  # nosec: B101
         self._identifier.__enter__()
         self._logger_context.__enter__()
-        # lazily initialize state
-        object.__setattr__(
-            self,
-            "_state_context",
-            StateContext.updated(self._state),
-        )
         self._state_context.__enter__()
         self._metrics_context.__enter__()
 
@@ -159,34 +164,23 @@ class ScopeContext:
     async def __aenter__(self) -> str:
         self._identifier.__enter__()
         self._logger_context.__enter__()
-        # lazily initialize group when needed
-        object.__setattr__(
-            self,
-            "_task_group_context",
-            TaskGroupContext(),
-        )
-        await self._task_group_context.__aenter__()
+
+        if task_group := self._task_group_context:
+            await task_group.__aenter__()
 
         # lazily initialize state to include disposables results
-        if self._disposables is not None:
+        if disposables := self._disposables:
+            assert self._state_context._token is None  # nosec: B101
             object.__setattr__(
                 self,
                 "_state_context",
-                StateContext.updated(
-                    (
-                        *self._state,
-                        *await self._disposables.__aenter__(),
-                    )
+                StateContext(
+                    state=self._state_context._state.updated(
+                        await disposables.__aenter__(),
+                    ),
                 ),
             )
 
-        else:
-            object.__setattr__(
-                self,
-                "_state_context",
-                StateContext.updated(self._state),
-            )
-
         self._state_context.__enter__()
         self._metrics_context.__enter__()
 
@@ -198,18 +192,19 @@ class ScopeContext:
         exc_val: BaseException | None,
         exc_tb: TracebackType | None,
     ) -> None:
-        if self._disposables is not None:
-            await self._disposables.__aexit__(
+        if disposables := self._disposables:
+            await disposables.__aexit__(
                 exc_type=exc_type,
                 exc_val=exc_val,
                 exc_tb=exc_tb,
             )
 
-        await self._task_group_context.__aexit__(
-            exc_type=exc_type,
-            exc_val=exc_val,
-            exc_tb=exc_tb,
-        )
+        if task_group := self._task_group_context:
+            await task_group.__aexit__(
+                exc_type=exc_type,
+                exc_val=exc_val,
+                exc_tb=exc_tb,
+            )
 
         self._metrics_context.__exit__(
             exc_type=exc_type,
@@ -293,6 +288,7 @@ class ctx:
         *state: State,
         disposables: Disposables | Iterable[Disposable] | None = None,
         logger: Logger | None = None,
+        task_group: TaskGroup | None = None,
         metrics: MetricsHandler | None = None,
     ) -> ScopeContext:
         """
@@ -317,6 +313,10 @@ class ctx:
             logger used within the scope context, when not provided current logger will be used\
              if any, otherwise the logger with the scope name will be requested.
 
+        task_group: TaskGroup | None
+            task group used for spawning and joining tasks within the context. Root scope will
+             always have task group created even when not set.
+
         metrics_store: MetricsStore | None = None
             metrics storage solution responsible for recording and storing metrics.\
              Metrics recroding will be ignored if storage is not provided.
@@ -343,6 +343,7 @@ class ctx:
         return ScopeContext(
             label=label,
             logger=logger,
+            task_group=task_group,
             state=state,
             disposables=resolved_disposables,
             metrics=metrics,
@@ -401,12 +402,12 @@ class ctx:
         return TaskGroupContext.run(function, *args, **kwargs)
 
     @staticmethod
-    def stream[Result, **Arguments](
-        source: Callable[Arguments, AsyncGenerator[Result, None]],
+    def stream[Element, **Arguments](
+        source: Callable[Arguments, AsyncGenerator[Element, None]],
         /,
         *args: Arguments.args,
         **kwargs: Arguments.kwargs,
-    ) -> AsyncIterator[Result]:
+    ) -> AsyncIterator[Element]:
         """
         Stream results produced by a generator within the proper context state.
 
@@ -427,25 +428,22 @@ class ctx:
             iterator for accessing generated results
         """
 
-        # prepare context snapshot
-        context_snapshot: Context = copy_context()
+        output_stream = AsyncStream[Element]()
 
-        # prepare nested context
-        streaming_context: ScopeContext = ctx.scope(
-            getattr(
-                source,
-                "__name__",
-                "streaming",
-            )
-        )
-
-        async def generator() -> AsyncGenerator[Result, None]:
-            async with streaming_context:
+        @ctx.scope("stream")
+        async def stream() -> None:
+            try:
                 async for result in source(*args, **kwargs):
-                    yield result
+                    await output_stream.send(result)
+
+            except BaseException as exc:
+                output_stream.finish(exception=exc)
+
+            else:
+                output_stream.finish()
 
-        # finally return it as an iterator
-        return context_snapshot.run(generator)
+        TaskGroupContext.run(stream)
+        return output_stream
 
     @staticmethod
     def check_cancellation() -> None:
@@ -488,7 +486,7 @@ class ctx:
         StateType
             resolved state instance
         """
-        return StateContext.current(
+        return StateContext.state(
             state,
             default=default,
         )

{haiway-0.13.1 → haiway-0.15.0}/src/haiway/context/state.py

@@ -1,10 +1,12 @@
-from collections.abc import Iterable, MutableMapping
+from asyncio import iscoroutinefunction
+from collections.abc import Callable, Coroutine, Iterable, MutableMapping
 from contextvars import ContextVar, Token
 from types import TracebackType
-from typing import Any, Self, cast, final
+from typing import Any, Self, cast, final, overload
 
 from haiway.context.types import MissingContext, MissingState
 from haiway.state import State
+from haiway.utils.mimic import mimic_function
 
 __all__ = [
     "ScopeState",
@@ -92,7 +94,7 @@ class StateContext:
     _context = ContextVar[ScopeState]("StateContext")
 
     @classmethod
-    def current[StateType: State](
+    def state[StateType: State](
         cls,
         state: type[StateType],
         /,
@@ -179,3 +181,41 @@ class StateContext:
             "_token",
             None,
         )
+
+    @overload
+    def __call__[Result, **Arguments](
+        self,
+        function: Callable[Arguments, Coroutine[Any, Any, Result]],
+    ) -> Callable[Arguments, Coroutine[Any, Any, Result]]: ...
+
+    @overload
+    def __call__[Result, **Arguments](
+        self,
+        function: Callable[Arguments, Result],
+    ) -> Callable[Arguments, Result]: ...
+
+    def __call__[Result, **Arguments](
+        self,
+        function: Callable[Arguments, Coroutine[Any, Any, Result]] | Callable[Arguments, Result],
+    ) -> Callable[Arguments, Coroutine[Any, Any, Result]] | Callable[Arguments, Result]:
+        if iscoroutinefunction(function):
+
+            async def async_context(
+                *args: Arguments.args,
+                **kwargs: Arguments.kwargs,
+            ) -> Result:
+                with self:
+                    return await function(*args, **kwargs)
+
+            return mimic_function(function, within=async_context)
+
+        else:
+
+            def sync_context(
+                *args: Arguments.args,
+                **kwargs: Arguments.kwargs,
+            ) -> Result:
+                with self:
+                    return function(*args, **kwargs)  # pyright: ignore[reportReturnType]
+
+            return mimic_function(function, within=sync_context)  # pyright: ignore[reportReturnType]

{haiway-0.13.1 → haiway-0.15.0}/src/haiway/context/tasks.py

@@ -40,12 +40,13 @@ class TaskGroupContext:
 
     def __init__(
         self,
+        task_group: TaskGroup | None = None,
     ) -> None:
         self._group: TaskGroup
         object.__setattr__(
             self,
             "_group",
-            TaskGroup(),
+            task_group if task_group is not None else TaskGroup(),
         )
         self._token: Token[TaskGroup] | None
         object.__setattr__(

{haiway-0.13.1 → haiway-0.15.0}/src/haiway/helpers/asynchrony.py

@@ -30,12 +30,10 @@ def wrap_async[**Args, Result](
 
 
 @overload
-def asynchronous[**Args, Result]() -> (
-    Callable[
-        [Callable[Args, Result]],
-        Callable[Args, Coroutine[Any, Any, Result]],
-    ]
-): ...
+def asynchronous[**Args, Result]() -> Callable[
+    [Callable[Args, Result]],
+    Callable[Args, Coroutine[Any, Any, Result]],
+]: ...
 
 
 @overload

{haiway-0.13.1 → haiway-0.15.0}/src/haiway/helpers/caching.py

@@ -66,7 +66,7 @@ def cache[**Args, Result, Key](
 ]: ...
 
 
-def cache[**Args, Result, Key](  # noqa: PLR0913
+def cache[**Args, Result, Key](
     function: Callable[Args, Result] | None = None,
     *,
     limit: int | None = None,

{haiway-0.13.1 → haiway-0.15.0}/src/haiway/helpers/metrics.py

@@ -15,6 +15,7 @@ __all_ = [
 
 class MetricsScopeStore:
     __slots__ = (
+        "allow_exit",
         "entered",
         "exited",
         "identifier",
@@ -31,6 +32,7 @@ class MetricsScopeStore:
         self.entered: float = monotonic()
         self.metrics: dict[type[State], State] = {}
         self.exited: float | None = None
+        self.allow_exit: bool = False
         self.nested: list[MetricsScopeStore] = []
 
     @property
@@ -115,7 +117,7 @@ class MetricsHolder:
 
     def __init__(self) -> None:
         self.root_scope: ScopeIdentifier | None = None
-        self.scopes: dict[ScopeIdentifier, MetricsScopeStore] = {}
+        self.scopes: dict[str, MetricsScopeStore] = {}
 
     def record(
         self,
@@ -124,10 +126,10 @@ class MetricsHolder:
         metric: State,
     ) -> None:
         assert self.root_scope is not None  # nosec: B101
-        assert scope in self.scopes  # nosec: B101
+        assert scope.scope_id in self.scopes  # nosec: B101
 
         metric_type: type[State] = type(metric)
-        metrics: dict[type[State], State] = self.scopes[scope].metrics
+        metrics: dict[type[State], State] = self.scopes[scope.scope_id].metrics
         if (current := metrics.get(metric_type)) and hasattr(current, "__add__"):
             metrics[type(metric)] = current.__add__(metric)  # pyright: ignore[reportUnknownMemberType, reportAttributeAccessIssue]
 
@@ -142,29 +144,29 @@ class MetricsHolder:
         merged: bool,
     ) -> Metric | None:
         assert self.root_scope is not None  # nosec: B101
-        assert scope in self.scopes  # nosec: B101
+        assert scope.scope_id in self.scopes  # nosec: B101
 
         if merged:
-            return self.scopes[scope].merged(metric)
+            return self.scopes[scope.scope_id].merged(metric)
 
         else:
-            return cast(Metric | None, self.scopes[scope].metrics.get(metric))
+            return cast(Metric | None, self.scopes[scope.scope_id].metrics.get(metric))
 
     def enter_scope[Metric: State](
         self,
         scope: ScopeIdentifier,
         /,
     ) -> None:
-        assert scope not in self.scopes  # nosec: B101
+        assert scope.scope_id not in self.scopes  # nosec: B101
         scope_metrics = MetricsScopeStore(scope)
-        self.scopes[scope] = scope_metrics
+        self.scopes[scope.scope_id] = scope_metrics
 
         if self.root_scope is None:
             self.root_scope = scope
 
         else:
             for key in self.scopes.keys():
-                if key.scope_id == scope.parent_id:
+                if key == scope.parent_id:
                     self.scopes[key].nested.append(scope_metrics)
                     return
 
@@ -177,8 +179,18 @@ class MetricsHolder:
         scope: ScopeIdentifier,
         /,
     ) -> None:
-        assert scope in self.scopes  # nosec: B101
-        self.scopes[scope].exited = monotonic()
+        assert self.root_scope is not None  # nosec: B101
+        assert scope.scope_id in self.scopes  # nosec: B101
+
+        self.scopes[scope.scope_id].allow_exit = True
+
+        if not all(nested.exited for nested in self.scopes[scope.scope_id].nested):
+            return  # not completed yet
+
+        self.scopes[scope.scope_id].exited = monotonic()
+
+        if scope != self.root_scope and self.scopes[scope.parent_id].allow_exit:
+            self.exit_scope(self.scopes[scope.parent_id].identifier)
 
 
 @final
@@ -213,7 +225,7 @@ class MetricsLogger:
         redact_content: bool,
     ) -> None:
         self.root_scope: ScopeIdentifier | None = None
-        self.scopes: dict[ScopeIdentifier, MetricsScopeStore] = {}
+        self.scopes: dict[str, MetricsScopeStore] = {}
         self.items_limit: int | None = items_limit
         self.redact_content: bool = redact_content
 
@@ -224,10 +236,10 @@ class MetricsLogger:
         metric: State,
     ) -> None:
         assert self.root_scope is not None  # nosec: B101
-        assert scope in self.scopes  # nosec: B101
+        assert scope.scope_id in self.scopes  # nosec: B101
 
         metric_type: type[State] = type(metric)
-        metrics: dict[type[State], State] = self.scopes[scope].metrics
+        metrics: dict[type[State], State] = self.scopes[scope.scope_id].metrics
         if (current := metrics.get(metric_type)) and hasattr(current, "__add__"):
             metrics[type(metric)] = current.__add__(metric)  # pyright: ignore[reportUnknownMemberType, reportAttributeAccessIssue]
 
@@ -248,29 +260,29 @@ class MetricsLogger:
         merged: bool,
     ) -> Metric | None:
         assert self.root_scope is not None  # nosec: B101
-        assert scope in self.scopes  # nosec: B101
+        assert scope.scope_id in self.scopes  # nosec: B101
 
         if merged:
-            return self.scopes[scope].merged(metric)
+            return self.scopes[scope.scope_id].merged(metric)
 
         else:
-            return cast(Metric | None, self.scopes[scope].metrics.get(metric))
+            return cast(Metric | None, self.scopes[scope.scope_id].metrics.get(metric))
 
     def enter_scope[Metric: State](
         self,
         scope: ScopeIdentifier,
         /,
     ) -> None:
-        assert scope not in self.scopes  # nosec: B101
+        assert scope.scope_id not in self.scopes  # nosec: B101
         scope_metrics = MetricsScopeStore(scope)
-        self.scopes[scope] = scope_metrics
+        self.scopes[scope.scope_id] = scope_metrics
 
         if self.root_scope is None:
             self.root_scope = scope
 
         else:
             for key in self.scopes.keys():
-                if key.scope_id == scope.parent_id:
+                if key == scope.parent_id:
                     self.scopes[key].nested.append(scope_metrics)
                     return
 
@@ -283,12 +295,22 @@ class MetricsLogger:
         scope: ScopeIdentifier,
         /,
     ) -> None:
-        assert scope in self.scopes  # nosec: B101
-        self.scopes[scope].exited = monotonic()
+        assert self.root_scope is not None  # nosec: B101
+        assert scope.scope_id in self.scopes  # nosec: B101
+
+        self.scopes[scope.scope_id].allow_exit = True
+
+        if not all(nested.exited for nested in self.scopes[scope.scope_id].nested):
+            return  # not completed yet
+
+        self.scopes[scope.scope_id].exited = monotonic()
+
+        if scope != self.root_scope and self.scopes[scope.parent_id].allow_exit:
+            self.exit_scope(self.scopes[scope.parent_id].identifier)
 
-        if scope == self.root_scope and self.scopes[scope].finished:
+        elif scope == self.root_scope and self.scopes[self.root_scope.scope_id].finished:
             if log := _tree_log(
-                self.scopes[scope],
+                self.scopes[scope.scope_id],
                 list_items_limit=self.items_limit,
                 redact_content=self.redact_content,
             ):

{haiway-0.13.1 → haiway-0.15.0}/src/haiway/utils/__init__.py

@@ -13,9 +13,11 @@ from haiway.utils.logs import setup_logging
 from haiway.utils.mimic import mimic_function
 from haiway.utils.noop import async_noop, noop
 from haiway.utils.queue import AsyncQueue
+from haiway.utils.stream import AsyncStream
 
 __all__ = [
     "AsyncQueue",
+    "AsyncStream",
     "always",
     "as_dict",
     "as_list",