|
2 | 2 |
|
3 | 3 | import asyncio
|
4 | 4 | import collections.abc
|
5 |
| -import functools |
6 | 5 | import inspect
|
7 | 6 | import types
|
8 | 7 |
|
9 |
| -from . import providers |
10 |
| -from .wiring import _Marker, PatchedCallable |
| 8 | +from .wiring import _Marker |
11 | 9 |
|
12 |
| -from .providers cimport Provider |
| 10 | +from .providers cimport Provider, Resource |
13 | 11 |
|
14 | 12 |
|
15 |
| -def _get_sync_patched(fn, patched: PatchedCallable): |
16 |
| - @functools.wraps(fn) |
17 |
| - def _patched(*args, **kwargs): |
18 |
| - cdef object result |
19 |
| - cdef dict to_inject |
20 |
| - cdef object arg_key |
21 |
| - cdef Provider provider |
| 13 | +def _sync_inject(object fn, tuple args, dict kwargs, dict injections, dict closings, /): |
| 14 | + cdef object result |
| 15 | + cdef dict to_inject |
| 16 | + cdef object arg_key |
| 17 | + cdef Provider provider |
22 | 18 |
|
23 |
| - to_inject = kwargs.copy() |
24 |
| - for arg_key, provider in patched.injections.items(): |
25 |
| - if arg_key not in kwargs or isinstance(kwargs[arg_key], _Marker): |
26 |
| - to_inject[arg_key] = provider() |
| 19 | + to_inject = kwargs.copy() |
| 20 | + for arg_key, provider in injections.items(): |
| 21 | + if arg_key not in kwargs or isinstance(kwargs[arg_key], _Marker): |
| 22 | + to_inject[arg_key] = provider() |
27 | 23 |
|
28 |
| - result = fn(*args, **to_inject) |
| 24 | + result = fn(*args, **to_inject) |
29 | 25 |
|
30 |
| - if patched.closing: |
31 |
| - for arg_key, provider in patched.closing.items(): |
32 |
| - if arg_key in kwargs and not isinstance(kwargs[arg_key], _Marker): |
33 |
| - continue |
34 |
| - if not isinstance(provider, providers.Resource): |
35 |
| - continue |
36 |
| - provider.shutdown() |
| 26 | + if closings: |
| 27 | + for arg_key, provider in closings.items(): |
| 28 | + if arg_key in kwargs and not isinstance(kwargs[arg_key], _Marker): |
| 29 | + continue |
| 30 | + if not isinstance(provider, Resource): |
| 31 | + continue |
| 32 | + provider.shutdown() |
37 | 33 |
|
38 |
| - return result |
39 |
| - return _patched |
| 34 | + return result |
40 | 35 |
|
41 | 36 |
|
42 |
| -async def _async_inject(object fn, tuple args, dict kwargs, dict injections, dict closings): |
| 37 | +async def _async_inject(object fn, tuple args, dict kwargs, dict injections, dict closings, /): |
43 | 38 | cdef object result
|
44 | 39 | cdef dict to_inject
|
45 | 40 | cdef list to_inject_await = []
|
@@ -69,7 +64,7 @@ async def _async_inject(object fn, tuple args, dict kwargs, dict injections, dic
|
69 | 64 | for arg_key, provider in closings.items():
|
70 | 65 | if arg_key in kwargs and isinstance(kwargs[arg_key], _Marker):
|
71 | 66 | continue
|
72 |
| - if not isinstance(provider, providers.Resource): |
| 67 | + if not isinstance(provider, Resource): |
73 | 68 | continue
|
74 | 69 | shutdown = provider.shutdown()
|
75 | 70 | if _isawaitable(shutdown):
|
|
0 commit comments