python 装饰器（二）： 加参数

接上篇python 闭包&装饰器（一）

1、功能函数加参数：实现一个能够接收任意数据的加法器

源代码以下：

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

2、装饰器函数加参数：根据需求打印日志

源代码以下：

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这个标志位了