Python装饰器
装饰器作用
- 装饰器可也用来扩展函数或者类的功能，在不改变原有主体函数或类的情况下，简便的为其扩展功能。
- 因为装饰器内功能调用在主体前，所以可以拦截主体函数调用或拦截主体类创建实例，并且处理它们。
函数装饰器
一个简单的装饰器，没有任何意义，仅是展示装饰器的用法。
def my_decorator(func):
def wrapper():
print("warpper")
return func()
print("my_decorator")
return wrapper
@my_decorator
def show():
print("show")
show()
print("函数名：%s" % show.__name__)
输出：
my_decorator
warpper
show
函数名：wrapper
这其中有一个问题，show()函数的名字已经改变，因为在装饰器中返回的是包装过的wrapper(*args, **kwargs)函数。如果要解决这个问题，可以引入官方包import functools，代码如下：
import functools
def my_decorator(func):
@functools.wraps(func)
def wrapper():
print("warpper")
return func()
print("my_decorator")
return wrapper
@my_decorator
def show():
print("show")
show()
print("函数名：%s" % show.__name__)
输出：
my_decorator
warpper
show
函数名：show
可以看到函数名已经回复正常。一般装饰器的使用都会是对原有函数进行功能扩充，所以原函数有参的话也可以传入装饰器。两个整数加法，如不是整数提示错误。
import functools
def my_decorator(func):
@functools.wraps(func)
def wrapper(*args, **kwargs):
for i in range(len(args)):
if not isinstance(args[i], int):
print("输入错误，请输入整数")
return
return func(*args, **kwargs)
return wrapper
@my_decorator
def add(x, y):
print("加法结果：%d" % (x + y))
add(11, 22)
类装饰器
可以为一个类添加装饰器，不改变类原有代码的情况下，为类添加或修改属性，也可以为类增加函数。
import functools
def my_decorator(cls):
def show():
print("新加入函数")
@functools.wraps(cls)
def wrapper():
cls.name = "张三"
cls.age = 20
cls.show = show
return cls
return wrapper
@my_decorator
class A:
def __init__(self):
self.name = "小明"
print("init")
a = A()
print(a.age)
print(a.name)
a.show()
输出：
20
张三
新加入函数
类A的def __init__(self)并没有执行，已经被装饰器拦截。装饰器为A添加了属性和函数。
初学python，如对知识点理解有误还请指正。欢迎补充不足之处，看到后我会及时补充进文章中。