Skip to content

Commit e55d11e

Browse files
committed
fixup! Add support for async generator injections
1 parent c82cc34 commit e55d11e

File tree

2 files changed

+7
-6
lines changed

2 files changed

+7
-6
lines changed

src/dependency_injector/_cwiring.pyx

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ from .wiring import _Marker
1111
cimport cython
1212

1313

14+
@cython.internal
1415
@cython.no_gc
1516
cdef class KWPair:
1617
cdef str name

src/dependency_injector/wiring.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1046,9 +1046,9 @@ def is_loader_installed() -> bool:
10461046
def _get_async_patched(fn: F, patched: PatchedCallable) -> F:
10471047
@functools.wraps(fn)
10481048
async def _patched(*args: Any, **raw_kwargs: Any) -> Any:
1049-
dr = DependencyResolver(raw_kwargs, patched.injections, patched.closing)
1049+
resolver = DependencyResolver(raw_kwargs, patched.injections, patched.closing)
10501050

1051-
async with dr as kwargs:
1051+
async with resolver as kwargs:
10521052
return await fn(*args, **kwargs)
10531053

10541054
return cast(F, _patched)
@@ -1058,9 +1058,9 @@ async def _patched(*args: Any, **raw_kwargs: Any) -> Any:
10581058
def _get_async_gen_patched(fn: F, patched: PatchedCallable) -> F:
10591059
@functools.wraps(fn)
10601060
async def _patched(*args: Any, **raw_kwargs: Any) -> AsyncIterator[Any]:
1061-
dr = DependencyResolver(raw_kwargs, patched.injections, patched.closing)
1061+
resolver = DependencyResolver(raw_kwargs, patched.injections, patched.closing)
10621062

1063-
async with dr as kwargs:
1063+
async with resolver as kwargs:
10641064
async for obj in fn(*args, **kwargs):
10651065
yield obj
10661066

@@ -1070,9 +1070,9 @@ async def _patched(*args: Any, **raw_kwargs: Any) -> AsyncIterator[Any]:
10701070
def _get_sync_patched(fn: F, patched: PatchedCallable) -> F:
10711071
@functools.wraps(fn)
10721072
def _patched(*args: Any, **raw_kwargs: Any) -> Any:
1073-
dr = DependencyResolver(raw_kwargs, patched.injections, patched.closing)
1073+
resolver = DependencyResolver(raw_kwargs, patched.injections, patched.closing)
10741074

1075-
with dr as kwargs:
1075+
with resolver as kwargs:
10761076
return fn(*args, **kwargs)
10771077

10781078
return cast(F, _patched)

0 commit comments

Comments
 (0)