Keyword Argument Injection with Python Decorators (Python recipe) by Ahmet Emre Aladağ

ActiveState Code (http://code.activestate.com/recipes/577382/)

This solution is a variant that allow to pass parameter not only with kwargs notation.

def injectArguments(inFunction):
    """
    This function allows to reduce code for initialization of parameters of a method through the @-notation
    You need to call this function before the method in this way: @injectArguments
    """
    def outFunction(*args, **kwargs):
        _self = args[0]
        _self.__dict__.update(kwargs)
        # Get all of argument's names of the inFunction
        _total_names = inFunction.func_code.co_varnames[1:inFunction.func_code.co_argcount]
        # Get all of the values
        _values = args[1:]
        # Get only the names that don't belong to kwargs
        _names = [n for n in _total_names if not kwargs.has_key(n)]

        # Match names with values and update __dict__
        d={}
        for n, v in zip(_names,_values):
            d[n] = v
        _self.__dict__.update(d)
        inFunction(*args,**kwargs)

    return outFunction


class Test:
""""""
@injectArguments
def __init__(self, name, surname):
    pass


if __name__=='__main__':
    t = Test('mickey', surname='mouse')
    print t.name, t.surname

Wow! That's what I've been looking for. It's been really useful to learn func_code.co_varnames. Thank you very much. This way, this snippet has became really useful.

In Python 2.4+ the code snippet below from @Filippo Squillace's version:

# Match names with values and update __dict__
d={}
for n, v in zip(_names,_values):
    d[n] = v
_self.__dict__.update(d)

could be replaced with simply:

# Match argument names with values and update __dict__
_self.__dict__.update(zip(_names,_values))

Regardless, both the original and modified recipe's seem elegant and useful.

Both the OP's and Filippo Squillace's variant should end their nested outFunction defintion by returning the result of calling the method being decorated, otherwise it will default to None. In other words, like this:

def injectArguments(inFunction):
    def outFunction(*args, **kwargs):
            ...
        return inFunction(*args, **kwargs) # return result of calling the decorated function

return outFunction

Adapted for python3:

```py

!/usr/bin/env python3

def injectArguments(inFunction): """ Decorator injecting arguments of a method as attributes

Found here: http://code.activestate.com/recipes/577382-keyword-argument-injection-with-python-decorators/

"""

def outFunction(*args, **kwargs):
    _self = args[0]
    _self.__dict__.update(kwargs)
    # Get all of argument's names of the inFunction
    _total_names = inFunction.__code__.co_varnames[1:inFunction.__code__.co_argcount]
    # Get all of the values
    _values = args[1:]
    # Get only the names that don't belong to kwargs
    _names = [n for n in _total_names if not n in kwargs]

    # Match argument names with values and update __dict__
    _self.__dict__.update(zip(_names,_values))

    return inFunction(*args,**kwargs)

return outFunction

if __name__=='__main__': class Test: @injectArguments def __init__(self, name, surname): pass

t = Test('mickey', surname='mouse')
assert(t.name == 'mickey' and t.surname == 'mouse')

```

Thank you!

Adapted for python3:

#!/usr/bin/env python3


def injectArguments(inFunction):
    """
    Decorator injecting arguments of a method as attributes

    Found here: http://code.activestate.com/recipes/577382-keyword-argument-injection-with-python-decorators/

    """

    def outFunction(*args, **kwargs):
        _self = args[0]
        _self.__dict__.update(kwargs)
        # Get all of argument's names of the inFunction
        _total_names = inFunction.__code__.co_varnames[1:inFunction.__code__.co_argcount]
        # Get all of the values
        _values = args[1:]
        # Get only the names that don't belong to kwargs
        _names = [n for n in _total_names if not n in kwargs]

        # Match argument names with values and update __dict__
        _self.__dict__.update(zip(_names,_values))

        return inFunction(*args,**kwargs)

    return outFunction


if __name__=='__main__':
    class Test:
        @injectArguments
        def __init__(self, name, surname):
            pass

    t = Test('mickey', surname='mouse')
    assert(t.name == 'mickey' and t.surname == 'mouse')

Thank you!

Modified to accept default values of non-passed arguments:

#!/usr/bin/env python3
# -*-coding:Utf-8 -*


def injectArguments(inFunction):
    """
    Decorator injecting arguments of a method as attributes

    Found here: http://code.activestate.com/recipes/577382-keyword-argument-injection-with-python-decorators/

    """

    def outFunction(*args, **kwargs):
        _self = args[0]
        _self.__dict__.update(kwargs)

        # Get all of argument's names of the inFunction
        _total_names = inFunction.__code__.co_varnames[1:inFunction.__code__.co_argcount]
        # Get all of the values
        _values = args[1:]
        # Get only the names that don't belong to kwargs
        _names = [n for n in _total_names if not n in kwargs]

        # Match argument names with values and update __dict__
        _self.__dict__.update(zip(_names,_values))

        # Add default value for non-specified arguments
        nb_defaults = len(_names) - len(_values)
        _self.__dict__.update(zip(_names[-nb_defaults:], inFunction.__defaults__[-nb_defaults:]))

        return inFunction(*args,**kwargs)

    return outFunction


if __name__=='__main__':
    class Test:
        @injectArguments
        def __init__(self, name, surname, default='lol'):
            pass

    t = Test('mickey', surname='mouse')
    assert(t.name == 'mickey' and t.surname == 'mouse' and t.default == 'lol')

You can find a new debugged version here: https://gist.github.com/Zzortell/2403054294b6312a32033b747ec0ad2c