python 装饰器

前言

装饰器常用于认证、日志。

环境

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()

最后编辑于：2019.05.16 17:04:14