python 装饰器（二）： 加参数

接上篇

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

源代码如下：

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