Python函数装饰器

import time
 
 
#装饰器函数
def show_time(f):
    def inner():
        start = time.time()
        f()
        end = time.time()
        print('spend %s'%(end-start))
    return inner
 
 
#通过@show_time调用
@show_time
def bar():
    print('bar...........')
 
def foo():
    print('foo...')
 
foo()
# bar=show_time(bar)
bar()

装饰器类似与上面，先编写函数的功能模块

通过@showtime来完成函数的功能追加

@showtime作用其实就是调用上面的闭包函数show_time

 上面的函数都是不带参数的函数装饰器，那么如果调用的函数有参数，并且函数个数不确定呢，可以参考下面的例子：

def show_time(f):
    def inner(*x,**y):
        start = time.time()
        f(*x,**y)
        end = time.time()
        print('spend %s'%(end-start))
    return inner
 
@show_time
def add(*x,**y):
    total=0
    for temp in x:
        total+=temp
    print(total)
 
 
# foo()
# bar=show_time(bar)
bar()
 
add(1,5,3,6,2,4)

如果装饰器中需要添加部分其他扩展的功能，可以给装饰器带上参数，如下：

import time
 
def logger(flag=''):
    def show_time(f):
        def inner(*x,**y):
            start = time.time()
            f(*x,**y)
            end = time.time()
            print('spend %s'%(end-start))
            if flag == 'true':
                print('日志记录')
        return inner
    return show_time
 
@logger('true')
def bar():
    print('bar...........')
 
@logger('true')
def add(*x,**y):
    total=0
    for temp in x:
        total+=temp
    print(total)
 
@logger()
def foo():
    print('foo...')
# foo()
# bar=show_time(bar)
#bar()
 
add(1,5,3,6,2,4)
foo()