美文网首页
Python---装饰器详解

Python---装饰器详解

作者: coderST | 来源:发表于2017-10-24 11:18 被阅读29次

    定义:

    本质上是一个函数。作用是用来装饰另一个函数(即被装饰函数),给被装饰函数添加功能。前提是不能改变被装饰函数的源代码和调用方式。这样的一个函数称之为装饰器。

    解析:

    下面我们话不多说,直接用代码说明。下面是一个函数。

    1 def add():
    2     b=1+2
    3     print(b)
    4
    5 add()
    
    程序输出:
    ————————
    
    3
    
    ————————
    
    • 现在我要给这个函数增加一个解释性的句子,如下,我们可以编写一个装饰器:
    #原函数
    def add():
        a=1+2
        print(a)  
    #装饰器
    def decorator(func):
        def warpper():
            print("1+2的结果是:")
            func()
        return warpper
    #注意此句   
    add=decorator(add)
    #调用函数
    add()
    

    程序输出:

    ——————————
    
    1+2的结果是:
    
    3
    
    ——————————
    
    • 这样我们就成功的达成了我们的目的。这里要注意第12行的这一句,这一句是将add这个函数对象传入了decorator()函数,返回的是一个新函数变量,这个新函数对象又重新赋值给add,这样就可以保证不改变被装饰函数的调用方式不变。在Python语法中有一种更优雅的方式可以代替第十二行的语句。如下:
    #装饰器
    def decorator(func):
        def warpper():
            print("1+2的结果是:")
            func()
        return warpper
    
    #add=decorator(add)
    #原函数
    @decorator#换成@符号
    def add():
        a=1+2
        print(a)
    #调用函数
    add()
    

    在被装饰函数前面直接加上“@xxx”(xxx为装饰器函数名)即可

    被装饰函数有参数怎么办?

    • 如果被装饰器函数有参数呢?该怎们班?不用担心,我们可以用不定参数的形式来收集参数。实例代码如下:
    def decorator(func):
        def warpper(*args,**kwargs):
            print("相加的结果是:")
            func(*args,**kwargs)
        return warpper
    
    @decorator
    def add(x,y):
        a=x+y
        print(a)
    
    add(2,3)
    

    程序输出:

    ——————————————————
    相加的结果是:
    5
    ——————————————————
    

    如上,我们给包装函数加上接收参数,然后传给func()函数就行了。这样不管被装饰函数有怎样的参数都不怕了。

    下面写一个页面验证的装饰器。

    • 大家知道有些网站的一部分页面是要求用户登录之后才可以访问的,比如下面的三个函数(分别代表三个页面):
    def index():
        print("welcome to the index page")
    def home():
        print("welcome to the home page")
    def bbs():
        print("welcome to the bbs page")
        return "I am the return contents"
    
    • 假如说现在我们要给home页面和bbs页面加上验证,显然现在更改源代码是不可行的。这个时候我们可以用装饰器,如下:
    username,passwd="jack","abc123"#模拟一个已登录用户
    def decorator(func):
        def warpper(*args,**kwargs):
            Username=input("Username:").strip()
            password=input("Password:").strip()
            if username==Username and passwd==password:
                print("Authenticate Success!")
                func(*args,**kwargs)
            else:
                exit("Username or password is invalid!")
        return warpper
    
    def index():
        print("welcome to the index page")
    @decorator
    def home():
        print("welcome to the home page")
    @decorator
    def bbs():
        print("welcome to the bbs page")
        return "I am the return contents"
    
    index()
    home()
    bbs()
    

    程序结果:

    ————————
    
    welcome to the index page    #index页面未验证直接可以登入
    Username:jack
    Password:abc123
    Authenticate Success!           #登录的而情形
    welcome to the home page
    Username:jack                      #密码或用户名错误的情形
    Password:123
    Username or password is invalid!
    ————————
    
    • 我们注意到bbs()是有返回值的,如果我们把上述代码的最后一句(第25行)改为“print(bbs())”之后再看看他的输出结果:
    ————————
    
    welcome to the index page
    Username:jack
    Password:abc123
    Authenticate Success!
    welcome to the home page
    Username:jack
    Password:abc123
    Authenticate Success!
    welcome to the bbs page
    None                              #返回值能么成None了???
    
    ————————
    
    • What happened! bbs()的返回值打印出来竟然是None。怎么会这样?这样的话不就改变了被装饰函数的源代码了吗?怎样才能解决呢?

      • 我们来分析一下:

      • 我们执行bbs函数其实就相当于执行了装饰器里的wrapper函数,仔细分析装饰器发现wrapper函数却没有返回值,所以为了让他可以正确保证被装饰函数的返回值可以正确返回,那么需要对装饰器进行修改:

    username,passwd="jack","abc123"#模拟一个已登录用户
    def decorator(func):
        def warpper(*args,**kwargs):
            Username=input("Username:").strip()
            password=input("Password:").strip()
            if username==Username and passwd==password:
                print("Authenticate Success!")
               return func(*args,**kwargs)#在这里加一个return就行了
            else:
                exit("Username or password is invalid!")
        return warpper
    
    def index():
        print("welcome to the index page")
    @decorator
    def home():
        print("welcome to the home page")
    @decorator
    def bbs():
        print("welcome to the bbs page")
        return "I am the return contents"
    
    index()
    home()
    bbs()
    
    • 如图加上return就可以解决了。下面我们在看看改后的程序输出:
    ————————
    
    welcome to the index page
    Username:jack
    Password:abc123
    Authenticate Success!
    welcome to the home page
    Username:jack
    Password:abc123
    Authenticate Success!
    welcome to the bbs page
    I am the return contents   #bbs()的返回值得到了正确的返回
    
    ——-——————
    

    好了,返回值的问题解决了.

    既然装饰器是一个函数,那装饰器可以有参数吗?

    • 答案是肯定的。我们同样可以给装饰器加上参数。比如还是上面的三个页面函数作为例子,我们可以根据不同页面的验证方式来给程序不同的验证,而这个验证方式可以以装饰器的参数传入,这样我们就得在装饰器上在嵌套一层函数 了:
    username,passwd="jack","abc123"#模拟一个已登录用户
    def decorator(auth_type):
        def out_warpper(func):
            def warpper(*args,**kwargs):
                Username=input("Username:").strip()
                password=input("Password:").strip()
                if auth_type=="local":
                    if username==Username and passwd==password:
                        print("Authenticate Success!")
                        return func(*args,**kwargs)
                    else:
                        exit("Username or password is invalid!")
                elif auth_type=="unlocal":
                    print("HERE IS UNLOCAL AUTHENTICATE WAYS")
            return warpper
        return out_warpper
    
    def index():
        print("welcome to the index page")
    @decorator(auth_type="local")
    def home():
        print("welcome to the home page")
    @decorator(auth_type="unlocal")
    def bbs():
        print("welcome to the bbs page")
        return "I am the return contents"
    
    index()
    home()
    bbs()
    

    输出:

    ————————
    
    welcome to the index page
    Username:jack
    Password:abc123
    Authenticate Success!
    welcome to the home page
    Username:jack
    Password:abc123
    HERE IS UNLOCAL AUTHENTICATE WAYS
    
    ————————
    
    • 可见,程序分别加入了第2行和第16行和中间的根据auth_type参数的判断的相关内容后, 就解决上述问题了。对于上面的这一个三层嵌套的相关逻辑,大家可以在 pycharm里头加上断点,逐步调试,便可发现其中的道理。

    • 总结

      • 要想学好迭代器就必须理解一下三条:

      • 1.函数即变量(即函数对象的概念)

      • 2.函数嵌套

      • 3.函数式编程

    原文链接:http://www.cnblogs.com/Akkbe-chao/p/6980332.html

    相关文章

      网友评论

          本文标题:Python---装饰器详解

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