使用任意数量的函数作为参数并返回函数组成的Python装饰器

Question

我正在尝试编写一个装饰器，该装饰器将任意数量的函数用作位置参数，然后将这些函数与正在装饰的函数组成。

从概念上讲，我正在尝试这样做：

@format(f1, f2, f3)
def f0(param):
    return value

f0(arg)应该评估为f3(f2(f1(f0(arg))))

这是我一直在努力说明这个概念的一个简单示例。

def lc(s):
    s = s.lower()
    return s

def punc(s):
    if s[-1] != ".":
        s = s + "."
        return s
    else:
        return s

def cap(s):
    s = s[0].upper() + s[1:]
    return s

def format(*args):
    def wrapper(func, *a):
        for f in args:
            func = f(func)
        return func
    return wrapper

@format(lc, punc, cap)
def example_func(param):
    return param

我期望的是：

f0("MY TEST STRING")

My test string.

我收到以下错误消息：

AttributeError: 'function' object has no attribute 'lower'

可能有一种更简单的方法来执行此操作，例如将函数直接传递到原始函数中而不使用装饰器，但是我正在寻找一种以这种方式专门使用装饰器的解决方案。 谢谢。

Answer 1

您需要使用“带有参数的装饰器”模式，如下所示：

def format(*fs):
    def deco(f):
        def wrapper(x):
            val = f(x)
            for func in fs:
                val = func(val)
            return val
        return wrapper
    return deco