python_装饰器

装饰器(语法糖)

定义： 本质是行数，（装饰其他函数）就是为了其他函数添加附加功能！

 

原则： 1.不能就该被装饰的函数的源代码。

　　　 2.不能修改被装饰的函数的调用方式。

 

实现装饰器的知识储备：

　　1.函数即“变量”。（在Python中一切皆为对象）

　　2.高阶函数：a 把一个函数名当作实参传给另外一个函数。

　　　　　  　　 b 返回值中包含函数名

　　3.嵌套函数

　　　　def func():

　　　　　　def func1():　　#局部变量（函数即变量）

　　　　　　　　pass

　　　　　　pass

 

高阶函数+嵌套函数 ==> 装饰器

 

通用的装饰器：(装饰器带参数，装饰的函数带参数)

 

import time

def auth(auth_type):
    print('auth func:', auth_type)

    def out_wrapper(func):
        def wrapper(*args, **kwargs):   #通用被装饰函数传参
            print('wrapper func args:', *args, **kwargs)
            if auth_type == 'local':
                print('\033[32;1mUser has passed authentication\033[0m')
                res = func(*args, **kwargs) #通用被装饰函数返回值
                print('--after authenticaion')
                return res

            elif auth_type == 'ldap':
                print('哈哈，我不会ldap')
                res = func(*args, **kwargs) #通用被装饰函数返回值
                print('--after authenticaion')
                return res
        return wrapper
    return out_wrapper
        

@auth(auth_type = 'local')  #先执行auth（arges），再用out_wrapper(home)装饰
def home():
    print('welcome to home  page')
    return 'from home'

@auth(auth_type = 'ldap')
def bbs():
    print('welcome to bbs page')
    return 'from bbs'


home()
bbs()