haiway 0.25.0__py3-none-any.whl → 0.25.1__py3-none-any.whl

haiway/context/access.py

@@ -275,6 +275,94 @@ class ScopeContext(Immutable):
         )
 
 
+class DisposablesContext(Immutable):
+    """
+    Immutable async context manager for managing collections of disposables.
+
+    DisposablesContext captures the current contextual state upon initialization
+    and provides an async context manager interface for managing disposable resources.
+    When entered, it prepares the disposables to collect their state, merges it with
+    the captured state, and creates a resolved StateContext. Upon exit, it properly
+    disposes of the disposables and cleans up internal references.
+
+    This class should not be instantiated directly; use the ctx.disposables() factory
+    method to create disposables contexts.
+    """
+
+    _disposables: Disposables
+    _captured_state: Collection[State]
+    _resolved_state_context: StateContext | None
+
+    def __init__(
+        self,
+        disposables: Disposables,
+    ) -> None:
+        # capture current contextual state (without new additions)
+        object.__setattr__(
+            self,
+            "_captured_state",
+            StateContext.current_state(),
+        )
+        # placeholder for temporary, resolved state context
+        object.__setattr__(
+            self,
+            "_resolved_state_context",
+            None,
+        )
+        object.__setattr__(
+            self,
+            "_disposables",
+            disposables,
+        )
+
+    async def __aenter__(self) -> None:
+        assert self._resolved_state_context is None  # nosec: B101
+
+        # Collect all state sources in priority order (lowest to highest priority)
+        collected_state: list[State] = []
+
+        collected_state.extend(self._captured_state)
+
+        collected_state.extend(await self._disposables.prepare())
+
+        # Create resolved state context with all collected state
+        resolved_state_context: StateContext = StateContext(
+            _state=ScopeState(tuple(collected_state))
+        )
+
+        resolved_state_context.__enter__()
+        object.__setattr__(
+            self,
+            "_resolved_state_context",
+            resolved_state_context,
+        )
+
+    async def __aexit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_val: BaseException | None,
+        exc_tb: TracebackType | None,
+    ) -> None:
+        assert self._resolved_state_context is not None  # nosec: B101
+
+        await self._disposables.dispose(
+            exc_type=exc_type,
+            exc_val=exc_val,
+            exc_tb=exc_tb,
+        )
+
+        self._resolved_state_context.__exit__(
+            exc_type=exc_type,
+            exc_val=exc_val,
+            exc_tb=exc_tb,
+        )
+        object.__setattr__(
+            self,
+            "_resolved_state_context",
+            None,
+        )
+
+
 @final
 class ctx:
     """
@@ -496,7 +584,7 @@ class ctx:
     @staticmethod
     def disposables(
         *disposables: Disposable | None,
-    ) -> Disposables:
+    ) -> DisposablesContext:
         """
         Create a container for managing multiple disposable resources.
 
@@ -513,9 +601,9 @@ class ctx:
 
         Returns
         -------
-        Disposables
+        DisposableContext
             A container that manages the lifecycle of all provided disposables
-            and propagates their state to the context when used with ctx.scope()
+            and propagates their state to the context as when used with ctx.scope()
 
         Examples
         --------
@@ -524,16 +612,19 @@ class ctx:
         >>> from haiway import ctx
         >>> async def main():
         ...
-        ...     async with ctx.scope(
-        ...         "database_work",
-        ...         disposables=(database_connection(),)
+        ...     async with ctx.disposables(
+        ...         database_connection(),
         ...     ):
         ...         # ConnectionState is now available in context
         ...         conn_state = ctx.state(ConnectionState)
         ...         await conn_state.connection.execute("SELECT 1")
         """
 
-        return Disposables(*(disposable for disposable in disposables if disposable is not None))
+        return DisposablesContext(
+            disposables=Disposables(
+                *(disposable for disposable in disposables if disposable is not None)
+            )
+        )
 
     @staticmethod
     def spawn[Result, **Arguments](

{haiway-0.25.0.dist-info → haiway-0.25.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: haiway
-Version: 0.25.0
+Version: 0.25.1
 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.25.0.dist-info → haiway-0.25.1.dist-info}/RECORD

@@ -1,7 +1,7 @@
 haiway/__init__.py,sha256=czgQKokK8NkHBv80HrUV5T1LIvMF301u7ZnIAneq-rg,2123
 haiway/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 haiway/context/__init__.py,sha256=5_4gad1PbSs1Swf9IyeOk6gDEDie8uTqXwm75FMyT2Q,1298
-haiway/context/access.py,sha256=ruzXdVTRhS9pdXy6lfVEvRxtubf7TOjE3nXGB9AHLYc,30261
+haiway/context/access.py,sha256=aYEn28Zb64IbsE2MmNmLUsg4CxZfBuJX8GG65UakkAk,33029
 haiway/context/disposables.py,sha256=7Jo-5qzS3UQvZUf4yOqUgfnueMg8I65jwHDp-4g6w54,7998
 haiway/context/identifier.py,sha256=ps7YM1ZnUrj66SPVyxqMhTRMaYOMNSb82J3FfMRVHm4,4690
 haiway/context/observability.py,sha256=rZoZT7g4ZM5_OKIFV0uA0rJodHxondw8X2bMSf_W6_s,20244
@@ -41,7 +41,7 @@ haiway/utils/mimic.py,sha256=xaZiUKp096QFfdSw7cNIKEWt2UIS7vf880KF54gny38,1831
 haiway/utils/noop.py,sha256=U8ocfoCgt-pY0owJDPtrRrj53cabeIXH9qCKWMQnoRk,1336
 haiway/utils/queue.py,sha256=6v2u3pA6A44IuCCTOjmCt3yLyOcm7PCRnrIGo25j-1o,6402
 haiway/utils/stream.py,sha256=lXaeveTY0-AYG5xVzcQYaiC6SUD5fUtHoMXiQcrQAAM,5723
-haiway-0.25.0.dist-info/METADATA,sha256=Ua7LksWkJQNmrEanbmf-69dcxU_Z3dIH_QQTolq64vM,4919
-haiway-0.25.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-haiway-0.25.0.dist-info/licenses/LICENSE,sha256=3phcpHVNBP8jsi77gOO0E7rgKeDeu99Pi7DSnK9YHoQ,1069
-haiway-0.25.0.dist-info/RECORD,,
+haiway-0.25.1.dist-info/METADATA,sha256=ggU_T40b25Eqe886YvYy6VydW4iZRAzJ1U57u7QpUbU,4919
+haiway-0.25.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+haiway-0.25.1.dist-info/licenses/LICENSE,sha256=3phcpHVNBP8jsi77gOO0E7rgKeDeu99Pi7DSnK9YHoQ,1069
+haiway-0.25.1.dist-info/RECORD,,