装饰器的多种实现方式(二)

基于类实现装饰器变体

　　在之前的文章里有提过基于类实现的装饰器，这里做一下延申

import types
from functools import wraps


class Decorator:
    def __init__(self, func):
        wraps(func)(self)
        self.calls = 0

    def __call__(self, *args, **kwargs):
        self.calls += 1
        print("preparing")
        return self.__wrapped__(*args, **kwargs)

    def __get__(self, instance, owner):
        """确保绑定方法对象的创建"""
        if instance is None:
            return self
        return types.MethodType(self, instance)


class Valley:
    def __init__(self):
        ...

    @Decorator
    def get(self):
        print("running")


Valley().get()
print(Valley.get.calls)

如果不实现__get__方法，实例方法会缺少调用方对象的传递，大家可以自行尝试

output:

　　preparing
　　running
　　1