import asyncio
import collections.abc
import contextvars
from typing import Self


class WrapCoro(collections.abc.Coroutine):
    def __init__(self, coro, context):
        self._coro = coro
        self._context = context

    def __await__(self) -> Self:
        return self

    def __iter__(self) -> Self:
        return self

    def __next__(self):
        return self.send(None)

    def throw(self, *exc_info):
        result = self._context.run(self._coro.throw, *exc_info)
        return result

    def send(self, v):
        result = self._context.run(self._coro.send, v)
        return result

async def coro():
    ...

async def foo():
    await WrapCoro(coro(), contextvars.copy_context())  # warning that WrapCoro is not awaitable

asyncio.run(foo())