Add the classmethod Caller.current_future.

pyproject.toml

@@ -50,6 +50,8 @@ Homepage = "https://fleming79.github.io/async-kernel"
 Documentation = "https://fleming79.github.io/async-kernel"
 Source = "https://github.com/fleming79/async-kernel"
 Tracker = "https://github.com/fleming79/async-kernel/issues"
+Changelog = "https://fleming79.github.io/async-kernel/latest/about/changelog/"
+
 
 [project.scripts]
 async-kernel = "async_kernel.command:command_line"

src/async_kernel/caller.py

@@ -307,6 +307,7 @@ class Caller:
     _stopped = False
     _protected = False
     _running = False
+    _future_var: contextvars.ContextVar[Future | None] = contextvars.ContextVar("_future_var", default=None)
     thread: threading.Thread
     "The thread in which the caller will run."
     backend: Backend
@@ -419,6 +420,7 @@ async def _wrap_call(
         args: tuple,
         kwargs: dict,
     ) -> None:
+        self._future_var.set(fut)
         if fut.cancelled():
             fut.set_result(cast("T", None))  # This will cancel
             return
@@ -429,7 +431,7 @@ async def _wrap_call(
                     if (delay_ := delay - time.monotonic() + starttime) > 0:
                         await anyio.sleep(float(delay_))
                     result = func(*args, **kwargs) if callable(func) else func  # pyright: ignore[reportAssignmentType]
-                    if inspect.isawaitable(result):
+                    if inspect.isawaitable(result) and result is not fut:
                         result: T = await result
                     if fut.cancelled() and not scope.cancel_called:
                         scope.cancel()
@@ -726,6 +728,11 @@ async def caller_context() -> None:
         assert isinstance(caller, cls)
         return caller
 
+    @classmethod
+    def current_future(cls) -> Future[Any] | None:
+        """Return the current future when called from inside a function scheduled by Caller."""
+        return cls._future_var.get()
+
     @classmethod
     async def as_completed(
         cls,

tests/test_caller.py

@@ -441,6 +441,12 @@ async def my_func():
         val = await fut
         assert val is True
 
+    async def test_current_future(self, anyio_backend):
+        async with Caller(create=True) as caller:
+            fut = caller.call_soon(Caller.current_future)
+            res = await fut
+            assert res is fut
+
     async def test_closed_in_call_soon(self, anyio_backend):
         ready = threading.Event()
         proceed = threading.Event()