装饰器相关

高阶函数的两种情况：

1、

把一个函数名当成实参传给另外一个函数（在不修改被装饰函数源代码的情况下，为其添加功能）

import time


def bar():
    time.sleep(2)
    print('in the bar')


def test1(fun_name):
    t1 = time.time()
    fun_name()
    print('run time:%s'%(time.time()-t1))


test1(bar)

2、

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

import time


def bar():
    time.sleep(2)
    print('in the bar')


def test1(fun_name):
    return  fun_name


bar = test1(bar)

bar()

高阶函数+嵌套函数则得到装饰器

import time


def timer(fun_name):
    def deco():
        start_time = time.time()
        fun_name()
        end_time = time.time()
        print('running time:%s' % (end_time - start_time))

    return deco


def bar():
    time.sleep(2)
    print('in the bar')


bar = timer(bar)
bar()

但我们一般把装饰器在被装饰函数前面定义，并且用如下的写法代替：

import time


def timer(fun_name):
    def deco():
        start_time = time.time()
        fun_name()  # 在这里调用是bar()这个函数
        end_time = time.time()
        print('running time:%s' % (end_time - start_time))

    return deco


@timer  # bar = timer(bar) 这步其实是调用了timer()这个函数，把bar这个函数进行重新定义
def bar():
    time.sleep(2)
    print('in the bar')


bar()  # 在调用bar的时候，其实调用的是deco()这个函数

　　

可以理解为，先用第一种高阶函数加上想要的功能，再重新覆盖定义这个函数，再用函数嵌套，就得到了我们想要的装饰器的效果

如果有参数：

import time


def timer(fun_name):
    def deco(*args,**kwargs):
        start_time = time.time()
        fun_name(*args,**kwargs)
        end_time = time.time()
        print('running time:%s' % (end_time - start_time))

    return deco


@timer  # bar = timer(bar)
def bar():
    time.sleep(2)
    print('in the bar')


@timer
def bar2(name,age):
    print(name,age)


bar()
bar2('simon',22)

　　

终极版：原函数带有返回值，装饰器本身带参数

__author__ = "Alex Li"
import time

user, passwd = 'alex', '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("\033[32;1mUser has passed authentication\033[0m")
                    res = func(*args, **kwargs)  # from home
                    print("---after authenticaion ")
                    return res
                else:
                    exit("\033[31;1mInvalid username or password\033[0m")
            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")


index()
print(home())  # wrapper()
bbs()

 

几个装饰器的小练习和写的答案

一：编写3个函数，每个函数执行的时间是不一样的，

提示：可以使用time.sleep(2)，让程序sleep 2s或更多，

二：编写装饰器，为每个函数加上统计运行时间的功能

提示：在函数开始执行时加上start=time.time()就可纪录当前执行的时间戳，函数执行结束后在time.time() - start就可以拿到执行所用时间

三：编写装饰器，为函数加上认证的功能，即要求认证成功后才能执行函数

四：编写装饰器，为多个函数加上认证的功能（用户的账号密码来源于文件），要求登录成功一次，后续的函数都无需再输入用户名和密码

提示：从文件中读出字符串形式的字典，可以用eval('{"name":"egon","password":"123"}')转成字典格式

答案：

# -*- coding: utf-8 -*-
import time, os

if not os.path.exists('login_for Decorator.txt'):
    with open('login_for Decorator.txt', 'w', encoding='utf-8')as f:
        f.write(str({'name': 'simon', 'password': '123'}))
with open('login_for Decorator.txt', 'r', encoding='utf-8')as f2:
    dict_list = eval(f2.read())

name, password = dict_list['name'], dict_list['password']

login_status = 0


# 验证是否登录,一次登录，后面不再需要登录则可以运行
def login(func):
    def deco(*args, **kwargs):
        global login_status
        if not login_status:
            if_wrong = 1  # 默认密码是错误的
            while if_wrong:  # 如果输入错误，就一直循环输入
                _name = input('your name :').strip()
                _password = input('type your password:').strip()
                if _name == name and _password == password:
                    print('welcome %s login' % _name)
                    login_status = 1
                    if_wrong = 0
                    func()
                else:
                    pass  # print('不应该出现这个')
        else:
            func()

    return deco


# 此装饰器的作用是打印函数运行时间
def timer(func):
    def deco(*args, **kwargs):
        t1 = time.time()
        func()
        print('running time:%s' % (time.time() - t1))

    return deco


@login
def fun1():
    time.sleep(3)
    print('here is fun1')


@login
def fun2():
    time.sleep(3)
    print('here is fun2')


@login
def fun3():
    time.sleep(3)
    print('here is fun3')


fun1()
fun2()
fun3()