Python基础知识之装饰器decorator

来源:互联网 发布:力劲注塑机怎么调数据 编辑:程序博客网 时间:2024/06/06 06:39

定义

本质是函数,(装饰其他函数)为其他函数添加附加功能。

原则

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