前言
装饰器常用于认证、日志。
环境
python3.6
正文
装饰器的效果:在不修改方法的情况下,额外增加方法的功能。
这里通过认证来具体介绍装饰器。
1. 首先存在一个获取学生信息的方法:
students_info = [
{"name": "Tom", "age": 12},
{"name": "Lily", "age": 13}
]
def student_get(name):
for i in students_info:
if i.get("name") == name:
return i
return {"name": "None", "age": -1}
print(student_get("Tom"))
通过调用 student_get 方法,传入名称,可以获得该学生的详细信息。但是当前方法没有实现任何的认证,谁都可以调用它。
你当然可以直接修改原有方法,在方法的开始时认证。或者实现两个方法,先调用认证方法,再调用实际功能方法。
但是,官方给出了一个更好的方式,无需改动原有代码,无需增加额外的控制逻辑,即可实现认证,称之为装饰器。
2. 实现认证装饰器
def auth(func):
def wrapper(name, *args, **kwargs):
print("auth start")
if "token" not in kwargs.keys():
print("auth failed")
return
print("auth done")
return func(name)
return wrapper
这里的func就是 student_get(),参数是一个方法。
在auth方法中又定义了wrapper方法,并且最后将这个方法直接返回。装饰器的语义就是这么定义的。
wrapper方法的参数:在使用装饰器时,调用func的参数首先进入wrapper。原来的func只需要接收一个name即可,现在增加了认证功能,当然需要引入额外的参数。
wrapper 方法 最后需要调用func,至于是否return,看需求。
3. 使用装饰器
@auth
def student_get(name):
for i in students_info:
if i.get("name") == name:
return i
return {"name": "None", "age": -1}
print(student_get("Tom", token="abc123"))
student_get() 上出现了一个 @auth 。这个auth就是刚刚创建的auth方法。使用装饰器就是这么简单。并且装饰器可以有多个,按顺序执行。
最后调用student_get()方法时传入的参数多了一个认证信息,这个token会被auth的wrapper使用,而"Tom"原封不动的传给student_get()
网友评论