函数式装饰器也就是闭包,闭包若是使用不当会造成内存的冗余,这里介绍类装饰器,它不仅功能强大,也更容易理解
首先说一说functool中wraps的作用,他可以保证多个装饰器装饰函数时,函数的name不会被改变,让多个装饰器同时起作用
a)---首先是在类里定义装饰器,但是目前还未懂这个原理,后续会补充的
class Analysis:
import os, re, time
def __init__(self, name):
self.name = name
def class_in(fuc):
def sss(self, *args, **kwargs):
print('这是class_in装饰后的:', self.name)
return fuc(self,*args, **kwargs)
return sss
@class_in
def test(self):
print('我被class_in装饰了!')
这是在类里定义装饰器,拿到外部使用的例子(cookbook上的方法)
from functools import wraps
class Analysis:
def __init__(self, name):
self.name = name
# 将对象方法作为装饰器
def dec(self, fun):
@wraps(fun)
def warps2(*args, **kwargs):
print('对象方法作为装饰器')
return fun(*args, **kwargs)
return warps2
@classmethod
def dec2(cls, fun):
@wraps(fun)
def warps1(*args, **kwargs):
print('类方法作为装饰器')
return fun(*args, **kwargs)
return warps1
# 将类方法作为装饰器
a = Analysis('ssd')
@a.dec
def fun111():
pass
@Analysis.dec2
def func():
pass
b)---给类添加装饰器,但是这种写法代码中会有警告,后续会补充解释
def class_decorate():
print('我是类的装饰器!')
@class_decorate
class Analysis:
........
c)---在类的装饰器中给类添加属性和方法,这里需注意的是,在装饰器中添加方法时,应传入不定长参数,如果使用对象调用该方法时会自动传入self,加上有时需要自定义参数,所以需要使用不定长参数接收。我们可以使用__dic__打印该类查看该类的属性和方法,结果如下:
{'__module__': '__main__', '__init__': <function Person.__init__ at 0x00599660>, '__dict__': <attribute '__dict__' of 'Person' objects>, '__weakref__': <attribute '__weakref__' of 'Person' objects>, '__doc__': None, 'age': 22222222222, 'fun1': <function class_decorate.<locals>.fun1 at 0x00599618>}
可以看到该类中存在通过装饰器添加的属性和方法age、fun1
def class_decorate(obj):
print('我是类的装饰器!,00000000000000000')
obj.age = 22222222222
def fun1(*args, **kwargs):
print(*args, **kwargs)
return '这是类的装饰器添加的方法,22222222222222'
obj.fun1 = fun1
return obj
@class_decorate
class Person:
def __init__(self, name):
self.name = name
d)---给每个类添加自定义值的属性,此处因参数不固定,所以使用不定长参数kwards接收,注意这里是要设置属性名,因此使用setattr给类添加属性及其对应值
def class_decorate(**kwargs):
def add_attr(obj):
for key, value in kwargs.items():
setattr(obj, key, value)
return obj
return add_attr
@class_decorate(age=11, sex='男')
class Person:
def __init__(self, name):
self.name = name
e)----将装饰器定义为类:要把装饰器定义成类,需要确保在类中实现__call__和__get__方法.
这里简单介绍__call__和__get__方法
(1)__call__:一种特殊方法,类作为函数被调用时启用__call__方法
待续
class Test:
def _init__(self, func):
self.func = func
def __call__(self, *args, **kwargs):
result = self.func(*args, **kwargs)
return result
使用时只需在被装饰函数前加@Test
f).把装饰器作用到类和静态方法中(需要保证装饰器在声明时需要放在@classmethod和@staticmethod之前,使用时应放在它们之后)
from functools import wraps
def class1(func):
@wraps(func)
def wrapss(*args, **kwargs):
print(22222222222)
return func(*args, **kwargs)
return wrapss
class A:
@class1
def instace(self):
print('被装饰器修饰的静态方法!')
@classmethod
@class1
def dd(cls):
print('被装饰器修饰的类方法')
网友评论