From f89a5577592c4dc470b249d5a83814c7c54a30bf Mon Sep 17 00:00:00 2001 From: Roman Mogylatov Date: Fri, 13 Nov 2020 15:22:46 -0500 Subject: [PATCH] Add @inject decorator --- src/dependency_injector/wiring.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/dependency_injector/wiring.py b/src/dependency_injector/wiring.py index 5afafbcb..9f12985f 100644 --- a/src/dependency_injector/wiring.py +++ b/src/dependency_injector/wiring.py @@ -21,6 +21,7 @@ from . import providers __all__ = ( 'wire', 'unwire', + 'inject', 'Provide', 'Provider', 'Closing', @@ -202,6 +203,12 @@ def unwire( _unpatch(member, method_name, method) +def inject(fn: Callable[..., Any]) -> Callable[..., Any]: + """Decorate callable with injecting decorator.""" + reference_injections, reference_closing = _fetch_reference_injections(fn) + return _get_patched(fn, reference_injections, reference_closing) + + def _patch_fn( module: ModuleType, name: str,