python 裝飾器（二）： 加參數

接上篇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這個標誌位了