美文网首页
python 装饰器

python 装饰器

作者: lzp1234 | 来源:发表于2019-05-16 16:59 被阅读0次

    前言

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

    环境

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

    相关文章

      网友评论

          本文标题:python 装饰器

          本文链接:https://www.haomeiwen.com/subject/njuiaqtx.html