如何在python中使用装饰器更改函数返回值？

Question

我有这个函数和一个装饰器 @decorator(integer)

def spam():
    return "ABCD EFG"

def decorator(number):
    def inside(function):
        *do something*
        return *somthing of type string*
    return inside

我想让装饰器返回 FUNCTION，接受整数并用number之后出现的字母替换每个字母。例如

@decorator(3)
def spam():
    return "ABCD EFG"

和

>>> print(spam())
"DEFG HIJ"

• 我如何让装饰器返回一个函数而不是一个字符串？

Answer 1

为了定义一个装饰器：

def decorator(number):
    def inner(f):
        def decorator_f(*args, **kwargs):
            result = f(*args, **kwargs)
            #make some manipulation on result
            return result
        return decorator_f
    return inner

然后你可以像这样使用它：

@decorator(3)
def spam():
    return "ABCD EFG"