装饰装饰器：试着让我的头脑理解它

Question

我正在尝试了解如何装饰装饰器，并想尝试以下方法：

假设我有两个装饰器并将它们应用到函数hello()：

def wrap(f):
    def wrapper():
        return " ".join(f())
    return wrapper


def upper(f):
    def uppercase(*args, **kargs):
        a,b = f(*args, **kargs)
        return a.upper(), b.upper()
    return uppercase

@wrap
@upper
def hello():
    return "hello","world"

print(hello())

然后我必须开始为其他功能添加其他装饰器，但通常@wrap 装饰器会“包装所有这些”

def lower(f):
    def lowercase(*args, **kargs):
        a,b = f(*args, **kargs)
        return a.lower(), b.lower()
    return lowercase

@wrap
@lower
def byebye():
    return "bye", "bye"

如何编写一个装饰器来装饰我的 @lower 和 @upper 装饰器？见下文：

@wrap
def lower():
    ...

@wrap
def upper():
    ...

要达到与上述相同的结果，只需这样做：

@upper
def hello():
    ...

@lower
def byebye():
    ...

Answer 1

def upper(f):
    @wrap
    def uppercase(*args, **kargs):
        a,b = f(*args, **kargs)
        return a.upper(), b.upper()
    return uppercase

---

Python 中的装饰器

 @foo
 def bar(...): ...

就相当于

 def bar(...): ...
 bar = foo(bar)

你想要得到的效果

@wrap
@upper
def hello(): ....

即

hello = wrap(upper(hello))

所以应该在upper的返回值上调用wrap：

def upper_with_wrap(f):
   def uppercase(...): ...
   return wrap(uppercase)

这也相当于在该函数上应用装饰器：

def upper_with_wrap(f):
   @wrap
   def uppercase(...): ...
   # ^ equivalent to 'uppercase = wrap(uppercase)'
   return uppercase

Answer 2

这是一个用装饰器装饰装饰器的通用解决方案（有点复杂）（耶！）。

# A second-order decorator
def decdec(inner_dec):
    def ddmain(outer_dec):
        def decwrapper(f):
            wrapped = inner_dec(outer_dec(f))
            def fwrapper(*args, **kwargs):
               return wrapped(*args, **kwargs)
            return fwrapper
        return decwrapper
    return ddmain

def wrap(f):
    def wrapper():
        return " ".join(f())
    return wrapper


# Decorate upper (a decorator) with wrap (another decorator)
@decdec(wrap)
def upper(f):
    def uppercase(*args, **kargs):
        a,b = f(*args, **kargs)
        return a.upper(), b.upper()
    return uppercase

@upper
def hello():
    return "hello","world"

print(hello())