파이썬에서 데코레이터를 사용하여 함수의 리턴을 변경하는 방법은 무엇입니까?

Question

나는 그런 함수의 반환 값을 변경하기 위해 장식을 만들고 싶어요, 어떻게 아래와 같이 그렇게? 그럼

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

을 .... 당신이 장식 된 함수를 호출하고 반환 값 변경 :

을

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

Answer 2

이 아마 장난감 예이기 때문에 내가 여기에 매우 일반적인 일하려고합니다, 당신은 매개 변수가 뭔가해야 할 수 있습니다

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'