python 类的使用(5)之类装饰器(类的装饰器和类作为装饰器)

python 类的使用(5)之类装饰器(类的装饰器和类作为装饰器)

类作为装饰器

要使用类作为装饰器实际上很简单，只要将某个类中多定义一个__call__方法，这样在类作为装饰器装饰函数时，函数运行时就会运行__call__方法中的内容，这就完成类作为函数装饰器的作用啦。

先看下例子：

from functools import wraps

class animal:
    def __init__(self, func):
        self.func = func
    
    # @wraps   
    def __call__(self, *args, **kwargs):
        print('working here')
        res = self.func(*args, **kwargs)
        return res

@animal
def test(name, kind):
    word = f'{name} belongs to {kind}'
    return word

A = test('cow','mammals')
print(type(test))
print(A)

运行结果：

working here
<class '__main__.animal'>
cow belongs to mammals

这里将类作为一个装饰器，首先使用__init__将函数实例化，但查看type(test)发现test并不是一个函数了，而是__main__.animal。实例化func后，使用self.func即可在任意位置调用函数，但是在__call__中定义的方法才能够装饰到函数中，因此运行函数时先打印了’working here’，接下来使用return 返回了函数运行的结果。

刚刚提到查看type(test)并不是一个函数，说明函数的信息由于被装饰后改变了，这里还是可以使用functools中的wraps模块来解决的。即在__call__上面加一个@wraps的装饰器就可以啦。加上@wraps后函数的信息就回来啦。这里我就不展示了(上面的代码只要将注释的@wraps取消注释，type(test)就不一样了)。

一个简单的类作为装饰器的例子就介绍结束啦，可以看出类作为装饰器还是很容易实现的，但是在一般情况下函数装饰器的使用还是更多的，因为类装饰器写起来还是相对较麻烦一点的。具体什么情况下使用类来写装饰器较好我也还不知道，大家可以自己探索下或者碰上的时候再考虑。

为类加装饰器
其实为类加装饰器也是很简单的，为类加装饰器的效果可能是修改类属性，修改类方法等。

修改类属性的类装饰器

def decorater(cls):             # 传入一个类即cls
    cls.num_of_animals = 10     # 设置一个类属性
    return cls                  # 返回这个被装饰过的类

@decorater
class animal:
    pass
# 这就完成了对类的装饰啦

A = animal()  
# 上面这行代码相当于在运行a = decorater(animal) 运行的结果
# 就是返回了一个被装饰过的新的cls，因此新的cls有了新的属性，我们就可以调用
# 这个num_of_animals的属性啦。

print(A.num_of_animals)

设置类方法的装饰器
你可以在装饰器中多加一些函数，增加更多的类的方法。例如：

def decorater(func):         
    def wrapper(cls):
        cls.num_of_animals = 10                   
        cls.f1 = func    
    	# 这里将传入的func即printd作为类的f1函数，我还不清楚怎么设置有self传入的函数。也可能不行
        return cls
    return wrapper

@decorater(printd)
class animal:
    pass

def printd(*args):
    print('this is a function')
    
A = animal()
A.f1()   # 是可以调用的哦
# 运行结果：this is a function

重写类的装饰器
当然你也可以完全重写一个类，即：

def decorater(cls):
	class wrapper:
		pass     # 这里面可以重写类
	return wrapper
	
@decorater
class animal:
	pass
# 这样是可以的，但是一般来说是没必要这样写哈。

除了上面介绍的内容，其实还有带参数的类装饰器，也都是异曲同工的做法，有兴趣的可以自行探索。
类作为装饰器装饰类
哈哈，写博客的时候有点好奇类能不能装饰类，然后就试了试，没想到真的可以。其实和之前的类作为装饰器装饰函数的方法几乎差不多。例：

class animal:
    def __init__(self, func):
        self.func = func
  
    def __call__(self, *args, **kwargs):
        print('working here')
        res = self.func(*args, **kwargs)
        
        return res

@animal        
class dogs:
    def test2(*args):
        print('look here')

wonwon = dogs()   # 首先实例化
# 结果：working here
wonwon.test2()    # 再执行dogs类中的test2函数
# 结果：look here
# 由于得到了dogs中类的test2函数的运行结果，表明wonwon确实是dogs类的一个实例，
# 表明之前的装饰是成功的。不过这里的装饰只是在创建类实例时调用了一次，其实作用不是很大。