一、功能函数加参数：实现一个可以接收任意数据的加法器
源代码如下：
def show_time(f):
def inner(*x, **y): # 形参
start = time.time()
f(*x, **y) # 相当于add()
end = time.time()
print('spend %s' % (end - start))
return inner
@show_time # @show_time 等于 add = show_time(add)
def add(*a, **b): # 实参 可以接收任意参数
sums = 0
for i in a:
sums += i
print(sums)
time.sleep(1) add(1, 2, 3, 4, 5) # 15
二、装饰器函数加参数：根据需求打印日志
源代码如下：
def logger(flag=''): # flag 默认为空，这一层函数的作用是应用标志位
def show_time(f):
def inner(*x, **y):
start = time.time()
f(*x, **y) # 相当于add()
end = time.time()
print('spend %s' % (end - start))
if flag == 'true':
print('打印日志') return inner
return show_time @logger('true') # @show_time
def add(*a, **b): # 不管有名参数和无名参数都可以接收到
sums = 0
for i in a:
sums += i
print(sums)
time.sleep(1) add(1, 2, 3, 4, 5) @logger('true')
def bar():
print('bar.......')
time.sleep(3) # 停留3秒 bar()
注意：@logger('true') 的执行机制是：先执行 logger('true') 函数，然后返回 show_time 函数的内存地址， 所以 @logger('true') 就相当于 @show_time，两者的区别就是 @logger('true') 要了一个变量flag, 之后的装饰器函数里面就可以使用falg这个标志位了