day 11 - 1 裝飾器進階

裝飾器進階

裝飾器的簡單回顧

裝飾器
開發原則：開放封閉原則
裝飾器的作用：在不改變原函數的調用方式的情況下，在函數的前後添加功能
裝飾器的本質：閉包函數

裝飾器的模式

def wrapper(func):
    def inner(*args,**kwargs):
        print("在被裝飾的函數執行之前做的事")
        ret =func(*args,**kwargs)
        print("在被裝飾的函數執行之前做的事")
        return ret
    return inner

@wrapper    #holiday = wrapper(holiday)
def holiday(day):
    print("放假休息 %s 天"%day)
    return '好開心'

ret = holiday(3)
print(ret)

我們來看兩個雙下方法：__name__ 、__doc__

def wahaha():
   '''
   一個列印娃哈哈的函數
   '''
   print('娃哈哈')

print(wahaha.__name__)  #查看字元串格式的函數名
print(wahaha.__doc__)     #document 顯示函數的註釋(限於多行註釋)
#輸出結果：
'''
wahaha

   一個列印娃哈哈的函數

'''

接下來是 wraps 函數

from functools import wraps  #引用 wraps 函數

def wrapper(func):
    #wraps() 加在最內層函數正上方
    #@wraps(func)   #在不使用 wraps 函數時 返回結果： inner  inner 函數的註釋
    @wraps(func)    #使用 wraps 函數來裝飾 inner 函數 返回結果：holiday  只是一個放假通知
    def inner(*args,**kwargs):
        '''
         inner 函數的註釋
        '''
        print("在被裝飾的函數執行之前做的事")
        ret =func(*args,**kwargs)
        print("在被裝飾的函數執行之前做的事")
        return ret
    return inner

@wrapper    #holiday = wrapper(holiday)
def holiday(day):
    '''
    只是一個放假通知
    '''
    print("放假休息 %s 天"%day)
    return '好開心'

print(holiday.__name__)  #當不調用 wraps 函數時，顯示的為 inner 函數的名稱
print(holiday.__doc__)   #當不調用 wraps 函數時，顯示的為 inner 函數的註釋

ret = holiday(3)    #在使用裝飾器 wraps 函數時，並不影響 holiday 函數的使用
print(ret)          #這正是裝飾器函數的特點
# 返回結果：
'''
在被裝飾的函數執行之前做的事
放假休息 3 天
在被裝飾的函數執行之前做的事
好開心
'''

假如你有成千上萬個函數使用了一個裝飾器，現在你想把這些裝飾器都取消掉，你要怎麼做？
一個一個的取消掉？ 沒日沒夜忙活3天。。。
過兩天領導想通了，再讓你加上。。。

#裝飾器的三層嵌套 為的是給裝飾器傳入變數
import time
#flage = False  #此處控制執行與否
flage = True
def timmer_out(flag):
    def timmer(func):
        def inner(*args,**kwargs):
            if flage:
                start = time.time()
                ret = func(*args,**kwargs)
                end = time.time()
                print(end - start)
                return ret
            else:
                ret = func(*args,**kwargs)
                return ret
        return inner
    return timmer

@timmer_out(flage)  #timmer_out(flage) 表示 timmer_out(flage) == timmer
def func():         #@ 表示 @timmer 即：func == timmer(func)
    time.sleep(0.1)
    print("結束")

@timmer_out(flage)
def func2():
    time.sleep(0.1)
    print("結束")

func()
func2()

多個裝飾器，裝飾一個函數

ef wrapper(func):
    def inner(*args,**kwargs):
        print("-----1-----")
        ret = func(*args,**kwargs)
        print("*****1*****")
        return ret
    return inner

def wrapper2(func):
    def inner2(*args,**kwargs):
        print("-----2-----")
        ret = func(*args,**kwargs)
        print("*****2*****")
        return ret
    return inner2

def wrapper3(func):
    def inner3(*args,**kwargs):
        print("-----3-----")
        ret = func(*args,**kwargs)
        print("*****3*****")
        return ret
    return inner3

@wrapper3
@wrapper2
@wrapper
def f():
    print("in f")
    return '就是這樣'
print(f())

執行過成可參考下圖