python—装饰器

一：什么叫装饰器

      1.装饰器：本质就是函数，为其他函数添加附加功能

                     （原则：不修改被修饰函数的源代码，不修改被修饰函数的调用方法）

import time
def cal(l):
    start_time=time.time()
    res=0
    for i in l:
        res+=i
    stop_time=time.time()
    print('函数的运行时间是%s' %(stop_time-start_time))
    return res
print(cal(range(100)))
#结果：函数的运行时间是0.0
#     4950

     

      2.装饰器=高阶函数+函数嵌套+闭包

         【1】高阶函数定义：

1.函数接收的参数是一个函数名
2.函数的返回值是一个函数名
3.满足上述条件任意一个，都可称之为高阶函数

#函数接收的参数是函数名
import time
def foo():
    print('nihao')
def test(func):
    # print(func)
    start_time=time.time()
    func()
    stop_time=time.time()
    print('函数的运行时间是 %s' %(stop_time-start_time))
test(foo)  #接收的参数是一个函数名foo
#结果：nihao
#     函数的运行时间是 0.0

#函数的返回值是函数名
def foo():
    print('from the foo')
def test(func):
    return func
res=test(foo)
print(res)  #res的内存地址
res()  #res加括号直接运行出来
#结果：<function foo at 0x0000013458554950>
#     from the foo

         【2】函数闭包（嵌套）：

#函数闭包，一层套一层，关闭的一层套一层，把变量闭包到里面
def father(name):
    def son():
        name='zy'
        print('爸爸是%s' %name)
    son()
father('zy')   #永远往上层找
#结果：爸爸是zy

#函数闭包补充：解压序列

>>>list=[1,2,3,8,9,6]
>>>a.*_.c=list
>>>a
1
>>>c
6
#取列表的第一个与最后一个元素
>>>f1=1
>>>f2=2
>>>f1,f2=f2,f1
#直接交换f1，f2的值

      3.装饰器基本实现

#装饰器的框架
def timmer(func):
    def wrapper():
        print(func)
        func()
    return wrapper
例子：
import time
#装饰器的框架
def timmer(func):
    def wrapper():
        #print(func)
        start_time=time.time()
        func()  #就是在运行test（）
        stop_time=time.time()
        print('运行时间是%s' %(stop_time-start_time))
    return wrapper

def test():
    time.sleep(3)
    print('test函数运行完毕')

res=timmer(test)  #返回的是wrapper的地址
res() #执行的是wrapper()
#结果：test函数运行完毕
#    运行时间是3.01176118850708

res=timmer(test)  #返回的是wrapper的地址
res() #执行的是wrapper()
#上面这块可以修改成：
test=timmer(test)
test()
还可以使用语法糖#@timmer（python提供的）替代赋值，在def test()：上面加@timmer装饰器名字就可以

     #函数闭包加参数：

import time
#装饰器的框架
def timmer(func):
    def wrapper(name,age):
        #print(func)
        start_time=time.time()
        res=func(name,age)  #就是在运行test（）
        stop_time=time.time()
        print('运行时间是%s' %(stop_time-start_time))
        return res
    return wrapper

@timmer  #test=timmer(test)
def test(name,age):
    time.sleep(3)
    print('test函数运行完毕,名字是【%s】 年龄是【%s】' %(name,age))
    return '这是test的返回值'

# res=timmer(test)  #返回的是wrapper的地址
res=test('zy',18) #执行的是wrapper()
print(res)
#结果：test函数运行完毕,名字是【zy】 年龄是【18】
#     运行时间是3.0002963542938232
#     这是test的返回值

    #函数闭包为函数加上认证功能：

def auth_func(func):
    def wrapper(*args,**kwargs):
        username=input('用户名：').strip()
        passwd= input('密码：').strip()
        if username=='周圆' and passwd=='123':
            res=func(*args,**kwargs)
            return res
        else:
            print('用户名或密码错误！')
    return wrapper
@auth_func
def index():
    print('欢迎来到京东主页')
@auth_func
def home(name):
    print('欢迎回家%s' %name)
@auth_func
def shopping_car(name):
    print('%s的购物车里有[%s]' %(name,'奶茶'))

index()
home('周圆')
shopping_car('周圆')
#结果：用户名：周圆
#     密码：123
#     欢迎来到京东主页
#     用户名：周圆
#     密码：1234
#     用户名或密码错误！
#     用户名：周圆
#     密码：123
#     周圆的购物车里有[奶茶]

     #函数闭包模拟session：

user_dic={'username':None,'login':False}

def auth_func(func):
    def wrapper(*args,**kwargs):
        if user_dic['username'] and user_dic['login']:
            res = func(*args, **kwargs)
            return res
        username=input('用户名：').strip()
        passwd= input('密码：').strip()
        if username=='周圆' and passwd=='123':
            user_dic['username']=username
            user_dic['login']=True
            res=func(*args,**kwargs)
            return res
        else:
            print('用户名或密码错误！')
    return wrapper
@auth_func
def index():
    print('欢迎来到京东主页')
@auth_func
def home(name):
    print('欢迎回家%s' %name)
@auth_func
def shopping_car(name):
    print('%s的购物车里有[%s]' %(name,'奶茶'))

index()
home('周圆')
shopping_car('周圆')

#用户名：周圆
#密码：123
#欢迎来到京东主页
#欢迎回家周圆
#周圆的购物车里有[奶茶]