如何使用装饰器更改函数的返回值？ [英] How to change a function's return using a decorator?

问题描述

我想创建一个装饰器来更改函数的返回值，如下所示：

I want create a decorator to change a function's return value like that, How to do that like below?:

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}

推荐答案

好吧...。您调用修饰的函数并更改返回值：

Well.... you call the decorated function and change the return value:

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

这篇关于如何使用装饰器更改函数的返回值？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！