¿Cómo se cambia el retorno de una función con el decorador en python?

Question

Quiero crear un decorador para cambiar el valor de retorno de una función de esa manera, ¿Cómo hacer eso, como a continuación ?:

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} 

Answer 1

Bueno .... se llama a la función decorada y cambia el valor de retorno:

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

Answer 2

voy a tratar de ser bastante general aquí, ya que esto es probablemente un ejemplo de juguete, y usted puede ser que necesite algo con parámetros:

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'