haiway 0.10.17__py3-none-any.whl → 0.11.1__py3-none-any.whl

haiway/__init__.py

@@ -9,7 +9,9 @@ from haiway.context import (
     MetricsScopeExiting,
     MissingContext,
     MissingState,
+    ScopeContext,
     ScopeIdentifier,
+    StateContext,
     ctx,
 )
 from haiway.helpers import (
@@ -78,8 +80,10 @@ __all__ = [
     "MissingContext",
     "MissingState",
     "ResultTrace",
+    "ScopeContext",
     "ScopeIdentifier",
     "State",
+    "StateContext",
     "always",
     "as_dict",
     "as_list",

haiway/context/__init__.py

@@ -1,4 +1,4 @@
-from haiway.context.access import ctx
+from haiway.context.access import ScopeContext, ctx
 from haiway.context.disposables import Disposable, Disposables
 from haiway.context.identifier import ScopeIdentifier
 from haiway.context.metrics import (
@@ -9,6 +9,7 @@ from haiway.context.metrics import (
     MetricsScopeEntering,
     MetricsScopeExiting,
 )
+from haiway.context.state import StateContext
 from haiway.context.types import MissingContext, MissingState
 
 __all__ = [
@@ -22,6 +23,8 @@ __all__ = [
     "MetricsScopeExiting",
     "MissingContext",
     "MissingState",
+    "ScopeContext",
     "ScopeIdentifier",
+    "StateContext",
     "ctx",
 ]

haiway/context/access.py

@@ -1,4 +1,9 @@
-from asyncio import CancelledError, Task, current_task, iscoroutinefunction
+from asyncio import (
+    CancelledError,
+    Task,
+    current_task,
+    iscoroutinefunction,
+)
 from collections.abc import (
     AsyncGenerator,
     AsyncIterator,

haiway/helpers/tracing.py

@@ -1,6 +1,6 @@
 from asyncio import iscoroutinefunction
 from collections.abc import Callable, Coroutine, Mapping, Sequence
-from typing import Any, Self, cast
+from typing import Any, Self, cast, overload
 
 from haiway.context import ctx
 from haiway.state import State
@@ -61,28 +61,53 @@ class ResultTrace(State):
     result: Any | Missing
 
 
+@overload
 def traced[**Args, Result](
     function: Callable[Args, Result],
     /,
-) -> Callable[Args, Result]:
-    if __debug__:
-        if iscoroutinefunction(function):
-            return cast(
-                Callable[Args, Result],
-                _traced_async(
-                    function,
-                    label=function.__name__,
-                ),
-            )
+) -> Callable[Args, Result]: ...
+
+
+@overload
+def traced[**Args, Result](
+    *,
+    label: str,
+) -> Callable[[Callable[Args, Result]], Callable[Args, Result]]: ...
 
-        else:
-            return _traced_sync(
-                function,
-                label=function.__name__,
-            )
 
-    else:  # do not trace on non debug runs
-        return function
+def traced[**Args, Result](
+    function: Callable[Args, Result] | None = None,
+    /,
+    *,
+    label: str | None = None,
+) -> Callable[[Callable[Args, Result]], Callable[Args, Result]] | Callable[Args, Result]:
+    def wrap(
+        wrapped: Callable[Args, Result],
+    ) -> Callable[Args, Result]:
+        if __debug__:
+            if iscoroutinefunction(wrapped):
+                return cast(
+                    Callable[Args, Result],
+                    _traced_async(
+                        wrapped,
+                        label=label or wrapped.__name__,
+                    ),
+                )
+
+            else:
+                return _traced_sync(
+                    wrapped,
+                    label=label or wrapped.__name__,
+                )
+
+        else:  # do not trace on non debug runs
+            return wrapped
+
+    if function := function:
+        return wrap(wrapped=function)
+
+    else:
+        return wrap
 
 
 def _traced_sync[**Args, Result](

haiway/utils/queue.py

@@ -23,9 +23,6 @@ class AsyncQueue[Element](AsyncIterator[Element]):
         self._waiting: Future[Element] | None = None
         self._finish_reason: BaseException | None = None
 
-    def __del__(self) -> None:
-        self.finish()
-
     @property
     def is_finished(self) -> bool:
         return self._finish_reason is not None
@@ -34,7 +31,6 @@ class AsyncQueue[Element](AsyncIterator[Element]):
         self,
         element: Element,
         /,
-        *elements: Element,
     ) -> None:
         if self.is_finished:
             raise RuntimeError("AsyncQueue is already finished")
@@ -45,8 +41,6 @@ class AsyncQueue[Element](AsyncIterator[Element]):
         else:
             self._queue.append(element)
 
-        self._queue.extend(elements)
-
     def finish(
         self,
         exception: BaseException | None = None,
@@ -57,7 +51,17 @@ class AsyncQueue[Element](AsyncIterator[Element]):
         self._finish_reason = exception or StopAsyncIteration()
 
         if self._waiting is not None and not self._waiting.done():
-            self._waiting.set_exception(self._finish_reason)
+            # checking loop only on finish as the rest of operations
+            # should always have a valid loop in a typical environment
+            # and we are not supporting multithreading yet
+            if get_running_loop() is not self._loop:
+                self._loop.call_soon_threadsafe(
+                    self._waiting.set_exception,
+                    self._finish_reason,
+                )
+
+            else:
+                self._waiting.set_exception(self._finish_reason)
 
     def cancel(self) -> None:
         self.finish(exception=CancelledError())

{haiway-0.10.17.dist-info → haiway-0.11.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: haiway
-Version: 0.10.17
+Version: 0.11.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.10.17.dist-info → haiway-0.11.1.dist-info}/RECORD

@@ -1,7 +1,7 @@
-haiway/__init__.py,sha256=IEUCyFYKT5IPHnkiUvDVZHdJeHqCaBnG8FhPD20Zgo8,1929
+haiway/__init__.py,sha256=PYMHx7JZwwSsALewho7v6EgIb5MUwYgZEWrVH-OhVx0,2005
 haiway/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-haiway/context/__init__.py,sha256=eRvuhifx7xCd-_6desgk55idzNpD5S5sprmCfGb3_9M,662
-haiway/context/access.py,sha256=Fc2NfoIV2zWE5qPnNgSitK1dy-PME-rJJfUPhoNwJzY,17125
+haiway/context/__init__.py,sha256=feqd0eJnGQwh4B8BZXpS0fQRE-DqoFCFOHipF1jOY8A,762
+haiway/context/access.py,sha256=KeAgxGf78TufqEUSwvw2LPY2aPE00z6xo3hvQ8L6QO4,17146
 haiway/context/disposables.py,sha256=DZjnMp-wMfF-em2Wjhbm1MvXubNpuzFBT70BQNIxC7M,2019
 haiway/context/identifier.py,sha256=Fyb6OHx5FPaSLLRK249HUEr_KlBSG5F-eW01Oxg_Ke8,2570
 haiway/context/logging.py,sha256=ptwgENuyw-WFgokVsYx9OXZGhJENuO_wgfVjcBryUKM,4251
@@ -16,7 +16,7 @@ haiway/helpers/metrics.py,sha256=0oFBiO-hAzihyC5jvXevNrYOoTcUGc2yGhE1A_866Mc,133
 haiway/helpers/retries.py,sha256=gIkyUlqJLDYaxIZd3qzeqGFY9y5Gp8dgZLlZ6hs8hoc,7538
 haiway/helpers/throttling.py,sha256=zo0OwFq64si5KUwhd58cFHLmGAmYwRbFRJMbv9suhPs,3844
 haiway/helpers/timeouted.py,sha256=1xU09hQnFdj6p48BwZl5xUvtIr3zC0ZUXehkdrduCjs,3074
-haiway/helpers/tracing.py,sha256=VDOAhdVELaYs92HxHreEo_ZV8b7e6ZQs10lTNn8xOtQ,3383
+haiway/helpers/tracing.py,sha256=8Gpcc_DguuHAdaxM4rGP0mB-S-8E7DKt7ZGym9f6x6Q,4018
 haiway/state/__init__.py,sha256=emTuwGFn7HyjyTJ_ass69J5jQIA7_WHO4teZz_dR05Y,355
 haiway/state/attributes.py,sha256=plCcYGE5LVU1Nvo0GHkhThqFG96uLR3tFsisQyK1jK0,23122
 haiway/state/path.py,sha256=p-ZVSDU02qAUDln12SDc1pGWzhZ88-1zYEBC0hqceXM,17930
@@ -35,8 +35,8 @@ haiway/utils/freezing.py,sha256=K34ZIMzbkpgkHKH-KF73plEbXExsajNRkRTYp9nJEf4,620
 haiway/utils/logs.py,sha256=oDsc1ZdqKDjlTlctLbDcp9iX98Acr-1tdw-Pyg3DElo,1577
 haiway/utils/mimic.py,sha256=BkVjTVP2TxxC8GChPGyDV6UXVwJmiRiSWeOYZNZFHxs,1828
 haiway/utils/noop.py,sha256=qgbZlOKWY6_23Zs43OLukK2HagIQKRyR04zrFVm5rWI,344
-haiway/utils/queue.py,sha256=oQ3GXCJ-PGNtMEr6EPdgqAvYZoj8lAa7Z2drBKBEoBM,2345
-haiway-0.10.17.dist-info/METADATA,sha256=IwCgN_53mbzSI8qNibPFrXQT_pwJnA34YxZY1BNdrc8,3858
-haiway-0.10.17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-haiway-0.10.17.dist-info/licenses/LICENSE,sha256=GehQEW_I1pkmxkkj3NEa7rCTQKYBn7vTPabpDYJlRuo,1063
-haiway-0.10.17.dist-info/RECORD,,
+haiway/utils/queue.py,sha256=jmz5b4cDvdXpuFnAfrNsGB5wXx4ThCgLHbFlH54o8UY,2657
+haiway-0.11.1.dist-info/METADATA,sha256=RAlc7LEyW8N7fizrJI0n_E1e52LlDscJDuEnD_nJdAM,3857
+haiway-0.11.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+haiway-0.11.1.dist-info/licenses/LICENSE,sha256=GehQEW_I1pkmxkkj3NEa7rCTQKYBn7vTPabpDYJlRuo,1063
+haiway-0.11.1.dist-info/RECORD,,