Python随心记--装饰器

Posted 2021-01-27

装饰器
    本质就是函数，功能是为其他函数添加附加功能
    原则是：不修改被修饰函数的源代码
          不修改被修饰函数的调用方式

统计函数的运行时间 

import time
def cal(l):
    start_time = time.time()
    res = 0
    for i in l:
        time.sleep(0.1)
        res += i
    stop_time = time.time()
    print(‘函数的运行时间是%s秒‘ %(stop_time - start_time))
    return res
print(cal(range(10)))

加上装饰器方法 统计函数的运行时间

import time
def timmer(func):
    def wapper(*args,**kwargs):
        start_time = time.time()
        res = func(*args,**kwargs)
        stop_time = time.time()
        print(‘函数的运行时间是%s秒‘ %(stop_time - start_time))
        return res
    return wapper
@timmer
def cal(l):
    res = 0
    for i in l:
        time.sleep(0.1)
        res += i
    return res
print(cal(range(10)))

装饰器的知识储备
    装饰器 = 高阶函数+函数嵌套+闭包
    高阶函数
        函数接收的参数是一个函数名
        函数的返回值是一个函数名
        满足上述任意条件一个都是高阶函数

def foo():
    print(‘你好啊林师傅‘)
def test(func):   #高阶函数
    print(func)
    func()
print(test(foo))

import time
def foo():
    print(‘你好啊林师傅‘)
def test(func):
    print(func)
    start_time = time.time()
    func()
    stop_time = time.time()
    print(‘函数的运行时间是%s秒‘ % (stop_time - start_time))
print(test(foo))

注：以上多运行了一次foo函数  

#高阶函数满足不了装饰器

import time
def foo():
    time.sleep(1)
    print(‘来自foo‘)
#不修改foo代码
#不修改foo调用方式
def timer(func):
    start_time = time.time()
    func()
    stop_time = time.time()
    print(‘函数的运行时间是%s秒‘ % (stop_time - start_time))
    return func
foo = timer(foo)
print(foo())

函数嵌套：函数里面定义一个函数

def father(name):
    print(‘from father %s‘ %name)
    def son():
        print(‘from son‘)
        def grandson():
            print(‘from grandson‘)
        grandson()
    son()
father(‘大佬‘)

def father(name):
    print(‘from father %s‘ %name)
    def son():
        print(‘from son‘)
    print(locals())
father(‘大佬‘)

函数嵌套及作用域

def father(name):
    def son():
        print(‘from son %s‘ %name)
    son()
father(‘大佬‘)

def father(name):
    def son():
        name = ‘小豆芽‘
        print(‘from son %s‘ %name)
    son()
father(‘大佬‘)

函数闭包装饰器基本实现

装饰器的框架

import time
def itmmer(func):
    def wrapper():
        strat_time = time.time()
        func()   #执行test函数
        stop_time = time.time()
        print(‘函数的运行时间是%s秒‘ % (stop_time - strat_time))
    return wrapper
def test():
    time.sleep(3)
    print(‘test函数运行完毕‘)
test = itmmer(test)#返回的是wrapper的地址
test()

 

#   @timmer  相当于 test = itmmer(test)  语法糖

import time
def itmmer(func):
    def wrapper():
        strat_time = time.time()
        func()   #执行test函数
        stop_time = time.time()
        print(‘函数的运行时间是%s秒‘ % (stop_time - strat_time))
    return wrapper
@itmmer
def test():
    time.sleep(3)
    print(‘test函数运行完毕‘)
test()

这里添加test返回值，，解决函数最后返回的None

import time
def itmmer(func):
    def wrapper():
        strat_time = time.time()
        res = func()   #执行test函数
        stop_time = time.time()
        print(‘函数的运行时间是%s秒‘ % (stop_time - strat_time))
        return res
    return wrapper
@itmmer
def test():
    time.sleep(1)
    print(‘test函数运行完毕‘)
    return ‘这里添加test返回值，，解决函数最后返回的None‘
res = test()   #这里会最后会返回一个None
print(res)

带参数

import time
def itmmer(func):
    def wrapper(name,age):
        strat_time = time.time()
        res = func(name,age)   #执行test函数
        stop_time = time.time()
        print(‘函数的运行时间是%s秒‘ % (stop_time - strat_time))
        return res
    return wrapper
@itmmer
def test(name,age):
    time.sleep(1)
    print(‘test函数运行完毕,名字是%s,年龄是%s‘ %(name,age))
    return ‘这里添加test返回值，，解决函数最后返回的None‘
test(‘ss‘,20)

带参数 修改为可变参数

import time
def itmmer(func):
    def wrapper(*args,**kwargs):
        strat_time = time.time()
        res = func(*args,**kwargs)   #执行test函数
        stop_time = time.time()
        print(‘函数的运行时间是%s秒‘ % (stop_time - strat_time))
        return res
    return wrapper
@itmmer
def test(name,age):
    time.sleep(1)
    print(‘test函数运行完毕,名字是【%s】,年龄是【%s】‘ %(name,age))
    return ‘这里添加test返回值，，解决函数最后返回的None‘
test(‘ss‘,20)

可变参数 *args,**kwargs 传参实例

def test2(name,age,gender):
    print(name)
    print(age)
    print(gender)
def test1(*args,**kwargs):
    test2(*args,**kwargs)   #test2(*(‘alex‘,18,‘male‘,‘x‘,‘y‘),**{})