装饰器的作用:给函数或类添加额外功能或信息
装饰器的本质:将函数作为参数传递,在另一个函数或类中执行原函数。在执行前后可以需要的操作。相当于包装纸,但是很有用。
函数装饰器
- 函数装饰器返回一个闭包,这个闭包会在原函数调用的时候调用。实际上已经不再是执行的原有的函数,只是闭包中调用了它。
- SA:
def decorator(func):
"""函数装饰器"""
def wrapper(*args, **kwargs):
# can do something
return func(*args, **kwargs)
return wrapper
类装饰器
- 类装饰需要实现
__call__
内置函数,该中内置函数可以是该类的实例当做函数一样被回调。 - 用类装饰器装饰时,实际上相当于调用类的
__call__
方法。 - SA:
class Decorator:
"""类装饰器"""
def __init__(self, func):
self.func = func
def __call__(self, *args, **kwargs):
"""
该内置函数(魔法方法)可以使实例可以被回调
such as:
d = Decorator()
var = d()
"""
# can do something
return self.func(*args, **kwargs)
总结
- 类装饰器与函数装饰器使用时并无差别。
- 实现时有所不同,类装饰器稍显麻烦。
- 功能都是为了给已存在的函数或类添加新功能,符合开闭原则。提高了代码可利用性。