Python装饰器15-开始使用类作为装饰器

基于前面的使用函数作为装饰器的理解,将类作为装饰器时需要保证以下几点

  1. 类的实例是可调用的
  2. 类需要一个地方讲被装饰的函数传入到类的实例里

第一条可以通过__call__实现,第二条可以通过__init__实现。

手工将类作为装饰器使用

class Profiled:
    def __init__(self, func):
        # wraps(func)(self)
        self.func = func

    def __call__(self, *args, **kwargs):
        print("call")
        return self.func(*args, **kwargs)
  
def add(x, y):
    return x + y

add = Profiled(add)
result = add(1, 2)
print(result)

执行结果:

call
3

似乎再用语法糖包装一下就可以实现将类作为装饰器来使用了。

使用类作为装饰器使用

from functools import wraps

class Profiled:

    def __init__(self, func):
        # wraps(func)(self)
        self.func = func

    def __call__(self, *args, **kwargs):
        print("call")
        return self.func(*args, **kwargs)
    
@Profiled
def add(x, y):
    return x + y

result = add(1, 2)
print(result)

运行结果同上。

小结

这节我们以我们理解的方法实现了将类作为装饰器使用,看起来都很正确,似乎就是这样使用?但是这种方法还有一些别的问题,最重要的是同样的问题,可以在代码中使用:

print(add.__name__)

发现无法输出__name__,根据约定使用装饰器不能改变函数的__name__属性,因此我们还需要改进将类作为装饰器的使用方法。

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容