python-装饰器-装饰函数带参数(三层def,两层return,语法糖)
# -*- coding:utf-8 -*-
# __author__:
# 2022/5/8
# 需要知道服务器响应速度,还需要知道谁执行了测试用例
import time
def show_time(flag): # 接收装饰函数入参
def show_time_name(func):# 接收被装饰函数本身
def inner(*args,**kwargs): # 接收被装饰函数入参
begin_time = time.time()
func(*args,**kwargs)
end_time = time.time()
print('服务器响应时间为:%s' % (end_time-begin_time))
print('装饰器传入的参数:%s' % flag)
# return func(*args,**kwargs) 也可以在这边去做执行
return inner
return show_time_name
@show_time('aaa') # foo = show_time('aaa')() foo = show_time_name(),这边一定要注意@show_time后面没有括号但是也会去执行一层,加上括号就会执行两层
def foo(*args):
amount = 0
for i in args:
amount += i
print(amount)
time.sleep(1)
foo(3) # inner()
【1】语法糖:@show_time--放在foo被装饰函数上面,指的是在调用foo函数的时候,会先去执行(直接执行,不是函数名)show_time函数,foo被装饰函数名作为入参传入
【2】@show_time---即 foo = show_time(foo),此时foo=show_time_name
【3】@show_time('aaa')---即先执行show_time,show_time的入参为aaa,返回show_time_name;然后再去执行show_time_name('aaa')(foo),即执行show_time_show(foo),返回inner
【4】此时 foo = inner
【5】最终执行foo(),即执行inner(),在inner里面有被装饰函数foo的执行
注:这个博主讲解的很好:https://www.bilibili.com/video/BV1JW411i7HR?spm_id_from=333.337.search-card.all.click