fixup! Add support for async generator injections

This commit is contained in:
ZipFile 2025-06-03 16:35:58 +00:00
parent c82cc343dd
commit e55d11e6ab
2 changed files with 7 additions and 6 deletions

View File

@ -11,6 +11,7 @@ from .wiring import _Marker
cimport cython
@cython.internal
@cython.no_gc
cdef class KWPair:
cdef str name

View File

@ -1046,9 +1046,9 @@ from ._cwiring import DependencyResolver # noqa: E402
def _get_async_patched(fn: F, patched: PatchedCallable) -> F:
@functools.wraps(fn)
async def _patched(*args: Any, **raw_kwargs: Any) -> Any:
dr = DependencyResolver(raw_kwargs, patched.injections, patched.closing)
resolver = DependencyResolver(raw_kwargs, patched.injections, patched.closing)
async with dr as kwargs:
async with resolver as kwargs:
return await fn(*args, **kwargs)
return cast(F, _patched)
@ -1058,9 +1058,9 @@ def _get_async_patched(fn: F, patched: PatchedCallable) -> F:
def _get_async_gen_patched(fn: F, patched: PatchedCallable) -> F:
@functools.wraps(fn)
async def _patched(*args: Any, **raw_kwargs: Any) -> AsyncIterator[Any]:
dr = DependencyResolver(raw_kwargs, patched.injections, patched.closing)
resolver = DependencyResolver(raw_kwargs, patched.injections, patched.closing)
async with dr as kwargs:
async with resolver as kwargs:
async for obj in fn(*args, **kwargs):
yield obj
@ -1070,9 +1070,9 @@ def _get_async_gen_patched(fn: F, patched: PatchedCallable) -> F:
def _get_sync_patched(fn: F, patched: PatchedCallable) -> F:
@functools.wraps(fn)
def _patched(*args: Any, **raw_kwargs: Any) -> Any:
dr = DependencyResolver(raw_kwargs, patched.injections, patched.closing)
resolver = DependencyResolver(raw_kwargs, patched.injections, patched.closing)
with dr as kwargs:
with resolver as kwargs:
return fn(*args, **kwargs)
return cast(F, _patched)