如何使用类的__call__函数编写装饰器？

Question

以下是我的代码。 给定任何内容，它必须在前面的内容后附加某些HTML标记。 我正在学习使用调用而不是函数闭包编写装饰器。

class decorate:
    def __init__(self, tag=""):
        self.tag = tag
    def __call__(self, function, *args):
        return "<p>{}</p>".format(self.tag, function(*args), self.tag)

@decorate(tag="p")
def get_content(content):
    return content    

print(get_content("I'm awesome"))

# Error i got.
Traceback (most recent call last):
  File "cache.py", line 27, in <module>
    @decorate(tag="p")
  File "cache.py", line 25, in __call__
    return "<p>{}</p>".format(self.tag, function(*args), self.tag)
TypeError: get_content() takes exactly 1 argument (0 given)

Answer 1

更改：

def __call__(self, function, *args):
    return "<p>{}</p>".format(self.tag, function(*args), self.tag)

变成：

def __call__(self, function):
   def wrap(*args):
       return "<{}>{}</{}>".format(self.tag, function(*args), self.tag)
   return wrap

您可能需要研究functools来对wrap进行更精细的装饰，但是除了支持help之类的东西外，这应该可以工作。