Làm thế nào thay đổi trở lại của hàm với trang trí trong python?

Question

Tôi muốn tạo ra một trang trí để thay đổi giá trị trả về của một chức năng như vậy, như thế nào để làm điều đó như dưới đây ?:

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

Vâng .... bạn gọi hàm được trang trí và thay đổi giá trị trả về:

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

Answer 2

tôi sẽ cố gắng để được khá chung ở đây vì đây có lẽ là một ví dụ đồ chơi, và bạn có thể cần một cái gì đó tham số:

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'