1. 程式人生 > >閉包&裝飾器

閉包&裝飾器

日誌 才有 釋放 for 員工 能夠 調用函數 xxx 變量

閉包

1.函數引用

def test():
    print(--test--)

# 調用函數
test()
# 引用函數
ret = test

print(id(ret))
print(id(test))

# 通過引用調用函數
ret()

#輸出結果
--test--
1718807047704
1718807047704
--test--

2.什麽是閉包

# 定義一個函數
def test(num):
    # 在函數內部再定義一個函數,並且這個函數用到外部函數的變量,那麽將這個函數以及用到的一些變量稱之為閉包
    def inner_test(inner_num):
        
print(inner_num:%d %inner_num) return num +inner_num # 其實這裏返回的就是閉包的結果 return inner_test # 給test函數賦值,這個20就是給參數num ret = test(10) # 這裏的50其實給參數inner_num print(ret(50)) # 這裏的100其實給參數inner_num print(ret(100))
# 輸出結果: inner_num:50 60 inner_num:100 110

3. 一個閉包的實際例子:

def line_conf(a, b):
    def line(x):
        
return a*x + b return line line1 = line_conf(1, 1) line2 = line_conf(4, 5) print(line1(5)) print(line2(5))

這個例子中,函數line與變量a,b構成閉包。在創建閉包的時候,我們通過line_conf的參數a,b說明了這兩個變量的取值,這樣,我們就確定了函數的最終形式(y = x + 1和y = 4x + 5)。我們只需要變換參數a,b,就可以獲得不同的直線表達函數。由此,我們可以看到,閉包也具有提高代碼可復用性的作用。

如果沒有閉包,我們需要每次創建直線函數的時候同時說明a,b,x。這樣,我們就需要更多的參數傳遞,也減少了代碼的可移植性。

註意點:

由於閉包引用了外部函數的局部變量,則外部函數的局部變量沒有及時釋放,消耗內存

4. 修改外部函數中的變量

python3的方法:

def outer(start=0):
    def inner():
        nonlocal start
        start += 1
        return start
    return inner

o1= outer(5)
print(o1())  # 6
print(o1())  # 7

o2 = outer(10)
print(o2())  # 11
print(o2())  # 12

python2的方法:

def outer(start=0):
    count=[start]
    def inner():
        count[0] += 1
        return count[0]
    return inner

o1 = closeure.outer(5)
print(o1())  # 6
print(o1())  # 7
o2 = closeure.outer(10)
print(o2())  # 11
print(o2())  # 12

裝飾器

1、先明白這段代碼

#### 第一波 ####
def foo():
    print(foo)

foo  # 表示是函數
foo()  # 表示執行foo函數

#### 第二波 ####
def foo():
    print(foo)

foo = lambda x: x + 1

foo()  # 執行lambda表達式,而不再是原來的foo函數,因為foo這個名字被重新指向了另外一個匿名函數

函數名僅僅是個變量,只不過指向了定義的函數而已,所以才能通過函數名()調用,如果 函數名=xxx被修改了,那麽當在執行 函數名()時,調用的就不知之前的那個函數了

2、需求

初創公司有N個業務部門,基礎平臺部門負責提供底層的功能,如:數據庫操作、redis調用、監控API等功能。業務部門使用基礎功能時,只需調用基礎平臺提供的功能即可。如下:

############### 基礎平臺提供的功能如下 ###############

def f1():
    print(f1)

def f2():
    print(f2)

def f3():
    print(f3)

def f4():
    print(f4)

############### 業務部門A 調用基礎平臺提供的功能 ###############

f1()
f2()
f3()
f4()

############### 業務部門B 調用基礎平臺提供的功能 ###############

f1()
f2()
f3()
f4()

目前公司有條不紊的進行著,但是,以前基礎平臺的開發人員在寫代碼時候沒有關註驗證相關的問題,即:基礎平臺的提供的功能可以被任何人使用。現在需要對基礎平臺的所有功能進行重構,為平臺提供的所有功能添加驗證機制,即:執行功能前,先進行驗證。

老大把工作交給員工C,他是這麽做的:

############### 基礎平臺提供的功能如下 ############### 

def f1():
    # 驗證1
    # 驗證2
    # 驗證3
    print(f1)

def f2():
    # 驗證1
    # 驗證2
    # 驗證3
    print(f2)

def f3():
    # 驗證1
    # 驗證2
    # 驗證3
    print(f3)

def f4():
    # 驗證1
    # 驗證2
    # 驗證3
    print(f4)

############### 業務部門不變 ############### 
### 業務部門A 調用基礎平臺提供的功能### 

f1()
f2()
f3()
f4()

### 業務部門B 調用基礎平臺提供的功能 ### 

f1()
f2()
f3()
f4()

老大把工作交給員工D,他是這麽做的:

############### 基礎平臺提供的功能如下 ############### 

def check_login():
    # 驗證1
    # 驗證2
    # 驗證3
    pass


def f1():

    check_login()

    print(f1)

def f2():

    check_login()

    print(f2)

def f3():

    check_login()

    print(f3)

def f4():

    check_login()

    print(f4)

老大看了下員工D的實現,嘴角漏出了一絲的欣慰的笑,語重心長的跟員工D聊了個天:

寫代碼要遵循開放封閉原則,雖然在這個原則是用的面向對象開發,但是也適用於函數式編程,簡單來說,它規定已經實現的功能代碼不允許被修改,但可以被擴展,即:

  • 封閉:已實現的功能代碼塊
  • 開放:對擴展開發

如果將開放封閉原則應用在上述需求中,那麽就不允許在函數 f1 、f2、f3、f4的內部進行修改代碼,老板就給了員工D一個實現方案:

def w1(func):
    def inner():
        # 驗證1
        # 驗證2
        # 驗證3
        func()
    return inner

@w1
def f1():
    print(f1)
@w1
def f2():
    print(f2)
@w1
def f3():
    print(f3)
@w1
def f4():
    print(f4)

# 單獨以f1為例:

def w1(func):
    def inner():
        # 驗證1
        # 驗證2
        # 驗證3
        func()
    return inner

@w1
def f1():
    print(f1)

python解釋器就會從上到下解釋代碼,步驟如下:

  1. def w1(func): ==>將w1函數加載到內存
  2. @w1

沒錯,從表面上看解釋器僅僅會解釋這兩句代碼,因為函數在 沒有被調用之前其內部代碼不會被執行。

從表面上看解釋器著實會執行這兩句,但是 @w1 這一句代碼裏卻有大文章, @函數名 是python的一種語法糖。

上例@w1內部會執行一下操作:

執行w1函數

執行w1函數 ,並將 @w1 下面的函數作為w1函數的參數,即:@w1 等價於 w1(f1) 所以,內部就會去執行:

def inner(): 
    #驗證 1 
    #驗證 2 
    #驗證 3 
    f1() # func是參數,此時 func 等於 f1 
return inner# 返回的 inner,inner代表的是函數,非執行函數 ,其實就是將原來的 f1 函數塞進另外一個函數中

w1的返回值

將執行完的w1函數返回值 賦值 給@w1下面的函數的函數名f1 即將w1的返回值再重新賦值給 f1,即:

新f1 = def inner(): 
        #驗證 1 
        #驗證 2 
        #驗證 3 
        原來f1() 
 return inner

所以,以後業務部門想要執行 f1 函數時,就會執行 新f1 函數,在新f1 函數內部先執行驗證,再執行原來的f1函數,然後將原來f1 函數的返回值返回給了業務調用者。

如此一來, 即執行了驗證的功能,又執行了原來f1函數的內容,並將原f1函數返回值 返回給業務調用者

3. 裝飾器(decorator)功能

  1. 引入日誌

  2. 函數執行時間統計

  3. 執行函數前預備處理

  4. 執行函數後清理功能

  5. 權限校驗等場景

  6. 緩存

4. 裝飾器示例

例1:無參數的函數

技術分享圖片
import time

def set_func(foo):
    def call_func():
        start_time = time.time()
        foo()
        end_time = time.time()
        print("運行時間:%f" % (end_time-start_time))
    return call_func

@set_func
def test():
    print("test")
    for i in range(100000):
        pass
test()

test = set_fun(foo)

# test先作為參數賦值給foo後,test接收指向set_fun返回的call_func
# 調用test(),即等價調用call_func()
# 內部函數call_func被引用,所以外部函數的foo變量(自由變量)並沒有釋放
# foo裏保存的是原test函數對象
View Code

例2:帶參數函數的裝飾

技術分享圖片
def set_func(foo):
    def call_func(num):
        foo(num)
    return call_func

@set_func
def test(num):
    print(num)

test(11)
View Code

例3:對多個函數進行裝飾

技術分享圖片
def set_func(foo):
    def call_func():
        foo()
    return call_func

@set_func
def test1():
    print("test1")

@set_func
def test2():
    print("test2")

test1()
test2()
View Code

例4:在調用函數之前已經開始裝飾

技術分享圖片
def set_func(foo):
    print("開始裝飾")
    def call_func():
        foo()
    return call_func

@set_func
def test1():
    print("test1")
View Code

例5:對不定長參數的函數裝飾

技術分享圖片
def set_func(foo):
    def call_func(*args, **kwargs):
        foo(*args, **kwargs)
        print("--end--")
    return call_func

@set_func
def test1(num, *args, **kwargs):
    print("test1",num)
    print("test1",args)
    print("test1",kwargs)

test1(11)
test1(11,22,33)
test1(11,22,33,a=55)
View Code

例6;對有返回值的函數裝飾(通用裝飾器)

技術分享圖片
def set_func(foo):
    def call_func(*args, **kwargs):
        return foo(*args, **kwargs)
    return call_func

@set_func
def test1():
    print("test1")
    return "ok"
ret = test1()
print(ret)
View Code

例7:多個裝飾器對一個函數裝飾

技術分享圖片
def set_func1(foo):
    print("裝飾器1開始裝飾")
    def call_func(*args, **kwargs):
        print("裝飾器1功能")
        return foo(*args, **kwargs)
    return call_func

def set_func2(foo):
    print("裝飾器2開始裝飾")
    def call_func(*args, **kwargs):
        print("裝飾器2功能")
        return foo(*args, **kwargs)
    return call_func

@set_func1
@set_func2
def test1():
    print("test1")

test1()

# 輸出結果:
裝飾器2開始裝飾
裝飾器1開始裝飾
裝飾器1功能
裝飾器2功能
test1
View Code 技術分享圖片
def set_func1(foo):
    print("裝飾器1開始裝飾")
    def call_func():
        return "<tr>"+foo()+"</tr>"
    return call_func

def set_func2(foo):
    print("裝飾器2開始裝飾")
    def call_func():
        return "<td>" + foo() + "</td>"
    return call_func

@set_func1
@set_func2
def test1():
    return "ok"
print(test1())

# 輸出結果:
裝飾器2開始裝飾
裝飾器1開始裝飾
<tr><td>ok</td></tr>
View Code

例8:用類對函數進行裝飾

技術分享圖片
class Test(object):
    def __init__(self, func):
        self.func = func

    def __call__(self, *args, **kwargs):
        print("裝飾器的功能")
        return self.func(*args, **kwargs)

@Test
def test():
    return "ok"

print(test())

#說明:
#1. 當用Test來裝作裝飾器對test函數進行裝飾的時候,首先會創建Test的實例對象
#   並且會把test這個函數名當做參數傳遞到__init__方法中
#   即在__init__方法中的屬性func指向了test指向的函數
#
#2. test指向了用Test創建出來的實例對象
#
#3. 當在使用test()進行調用時,就相當於讓這個對象(),因此會調用這個對象的__call__方法
#
#4. 為了能夠在__call__方法中調用原來test指向的函數體,所以在__init__方法中就需要一個實例屬性來保存這個函數體的引用
#   所以才有了self.func = func這句代碼,從而在調用__call__方法中能夠調用到test之前的函數體
View Code

例9:裝飾器帶參數,在原有裝飾器的基礎上,設置外部變量

技術分享圖片
def level(level_num):
    def set_func(func):
        def call_func(*args, **kwargs):
            if level_num == 1:
                print("極品貨")
            elif level_num == 2:
                print("A貨")
            func(*args, **kwargs)
        return call_func
    return set_func


@level(1)
def test1():
    print("ok")

@level(2)
def test2():
    print("ok")

test1()
test2()
# 下面的裝飾過程
# 1. 調用level("1")
# 2. 將步驟1得到的返回值,即set_func返回, 然後set_func(func)
# 3. 將set_func(func)的結果返回,即call_func
# 4. 讓test1 = call_func,即test1現在指向call_func
View Code

閉包&裝飾器