Comment changer une fonction de retour avec le décorateur en python?

Je veux créer un décorateur pour changer une fonction de la valeur de retour comme ça, Comment le faire comme ci-dessous?:

def dec(func):
    def wrapper():
        #some code...
        #change return value append 'c':3
    return wrapper

@dec
def foo():
    return {'a':1, 'b':2}

result = foo()
print result
{'a':1, 'b':2, 'c':3}

OriginalL'auteur libaoyin | 2011-08-26

  1. 27

    .... vous appelez le décoré de la fonction et de modifier la valeur de retour:

    def dec(func):
    def wrapper(*args, **kwargs):
        result = func(*args, **kwargs)
        result['c'] = 3
        return result
    return wrapper

    OriginalL'auteur Felix Kling

  2. 10

    Je vais essayer d'être assez général ici puisque c'est probablement un jouet exemple, et vous pourriez avoir besoin de quelque chose paramétrée:

    from collections import MutableMapping

def map_set(k, v):
    def wrapper(func):
        def wrapped(*args, **kwds):
            result = func(*args, **kwds)
            if isinstance(result, MutableMapping):
                result[k] = v 
            return result
        return wrapped
    return wrapper

@map_set('c', 3)
def foo(r=None):
    if r is None:
        return {'a':1, 'b':2}
    else:
        return r

>>> foo()
{'a': 1, 'c': 3, 'b': 2}

>>> foo('bar')
'bar'
    Oui! il est travaillé. et plus vous en ajoutez un mutableMapping vérifier. il est grand.

    OriginalL'auteur eryksun