在阅读博客中,发现了类装饰器的存在,由于之前就在写类相关的专栏,这次就赶紧补上之前的内容啦。类装饰器这个词是有歧义的,因为类本身可以作为装饰器,一个类也可以被函数装饰器所装饰。今天就简单介绍一下这两种方式吧!
不要以为类装饰器很高端,其功能还是挺平易近人的哦,总会有机会让你用上它的!嘿嘿嘿,不妨试试吧!
附上类使用专栏的其它文章的链接:
- python 类的使用(1) 之类使用的基本框架
- python 类的使用(2) 之类变量
- python 类的使用(3)之类中常用的三个装饰器@classmethod、@staticmethod、@property
- python 类的使用(4)之类常用的魔法方法
- python 类的使用(5)之类装饰器(类的装饰器和类作为装饰器)
- python 类的使用(6)之类的继承
- python 类的使用(7)之类的私有属性和方法 (类中以单下划线或双下划线开头的变量名或函数名)
类作为装饰器
要使用类作为装饰器实际上很简单,只要将某个类中多定义一个__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类的一个实例,
# 表明之前的装饰是成功的。不过这里的装饰只是在创建类实例时调用了一次,其实作用不是很大。
叮
放假咯!可以放松玩一下了,嘿嘿。
发现在生活中能够给我身边的朋友带来一些快乐,大家与我相处时感觉好像挺开心的,虽说很多时候我本不是打算搞笑的。不过能给大家带来快乐我真的觉得非常之荣幸,看到朋友们的笑容,听到他们的笑声,我发自心底觉得开心。也希望大家天天开心啊!
参考:https://www.jianshu.com/p/f758ef3b41d0
参考:https://www.cnblogs.com/wickedpriest/p/11872402.html
参考:https://blog.csdn.net/lilied001/article/details/80593475