Python基础知识之装饰器decorator

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/BearKChan/article/details/78754702

定义

本质是函数，（装饰其他函数）为其他函数添加附加功能。

原则

1. 不能修改被装饰的函数的源代码
2. 不能修改被装饰的函数的调用方式

实现装饰器知识储备

• 函数即“变量”
• 高阶函数
• 嵌套函数

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

1. 函数即“变量”：

定义一个函数就相当于定义一个变量，即将函数体赋值给一个变量名。python的内存回收机制规定：当存储在内存中的内容没有对应的变量名指定时，则当内存回收机制定期清理内存的时候则回收内存。

2. 高阶函数：

• 把一个函数名当作实参传给另外一个函数（在不修改被装饰函数源代码的前提下添加新功能）

    def bar():
        print("in the bar")

    #  test1就是一个高阶函数
    def test1(func):
        print(func)
        func()


    test1(bar)

• 返回值中包含函数名
  （不修改函数的调用方式）

3. 嵌套函数：

        在一个函数的函数体内用def声明一个新的函数。
        def func1():
            def func2():
                pass

装饰器案例剖析

import time

# 这里的timer就是一个装饰器
def timer(func):
    def deco(*args, **kwargs):
        start_time = time.time()
        func(*args, **kwargs)
        end_time = time.time()
        print("the func run time is %s" %(end_time-start_time))
    return deco

# 装饰器的调用
@timer  # test1 = timer(test1)
def test1():
    time.sleep(3)
    print("in the test1")


test1()

装饰器高级案例

user, passwd = 'bear', 'abc123'
def auth(auth_type):
    print("auth func:",auth_type)
    def outer_wrapper(func):
        def wrapper(*args, **kwargs):
            print("wrapper func args:", *args, **kwargs)
            if auth_type == "local":
                username = input("Username:").strip()
                password = input("Password:").strip()
                if user == username and passwd == password:
                    print("User has passed authtication")
                    res = func(*args, **kwargs)
                    return res
                else:
                    exit("Invalid username or password")
            elif auth_type == "ldap":
                print("搞毛线ldap,不会。。。。")
        return wrapper
    return outer_wrapper


def index():
    print("welcome to index page")

@auth(auth_type = "local")  # home = wrapper()
def home():
    print("welcome to home page")
    return "from home"
@auth(auth_type = "ldap")
def bbs():
    print("welcome to bbs page")


print(home())  #wapper()