Add storing of originaly decorated instance in @inject decorator

This commit is contained in:
Roman Mogilatov 2016-03-01 16:28:05 +02:00
parent 92a58cd4f9
commit 86ef17c75a

View File

@ -228,7 +228,7 @@ def inject(*args, **kwargs):
callback = callback_or_cls
if hasattr(callback, 'injections'):
if hasattr(callback, '__INJECT_DECORATED__'):
callback.args += arg_injections
callback.kwargs += kwarg_injections
callback.injections += arg_injections + kwarg_injections
@ -246,6 +246,8 @@ def inject(*args, **kwargs):
return callback(*args, **kwargs)
decorated.__INJECT_DECORATED__ = True
decorated.origin = callback
decorated.args = arg_injections
decorated.kwargs = kwarg_injections
decorated.injections = arg_injections + kwarg_injections