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

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-0.11.0.dist-info → haiway-0.11.1.dist-info}/METADATA

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

@@ -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
@@ -36,7 +36,7 @@ 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=jmz5b4cDvdXpuFnAfrNsGB5wXx4ThCgLHbFlH54o8UY,2657
-haiway-0.11.0.dist-info/METADATA,sha256=HWxQi7lFsvowR9D-wF-zNBHnIfMVq0i5Jlze3BAbVEo,3857
-haiway-0.11.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-haiway-0.11.0.dist-info/licenses/LICENSE,sha256=GehQEW_I1pkmxkkj3NEa7rCTQKYBn7vTPabpDYJlRuo,1063
-haiway-0.11.0.dist-info/RECORD,,
+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,,