[[413590]]

熟悉 Golang 的同學都知道，Golang 里面有一個關鍵詞叫做defer，它可以實現延遲調用。

實際上在 Python 里面也有相關的語法，那就是contextlib.ExitStack。

我們來看這樣一個場景：

我有一個函數parse，它的作用是從 Redis 中持續讀入數據，并寫入到MongoDB 中。示例代碼如下：

import json 
import redis 
import pymongo 
 
client = redis.Redis() 
handler = pymongo.MongoClient().test.data 
 
def parse(): 
    data = client.lpop('test') 
    if not data: 
        return 
    handler.insert_one(json.loads(data)) 

但現在我想增加一個需求，當Redis 讀取結束或者讀取數據報錯的時候，能把當前的時間也寫入到MongoDB 中。

那么代碼可能變成下面這樣：

import json 
import redis 
import datetime 
import pymongo 
 
client = redis.Redis() 
handler = pymongo.MongoClient().test.data 
 
def parse(): 
    while True: 
        try: 
            data = client.lpop('test') 
            if not data: 
                handler.insert_one({'finished': True, 'ts': datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}) 
                return 
            handler.insert_one(json.loads(data)) 
        except Exception: 
            handler.insert_one({'finished': True, 'ts': datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S'}) 

可以看到，代碼變得很難看了。

現在，我們可以使用延遲調用來讓代碼變得更好看。

要實現這個目的，就可以開始使用ExitStack了。它可以注冊多個回調函數，在退出上下文縮進時執行。

我們先來看一個簡單的例子：

import contextlib 
 
def callback_1(): 
    print('我是第一個回調函數') 
 
def callback_2(x): 
    print(f'我是第二個回調函數，傳入參數：{x}') 
 
 
with contextlib.ExitStack() as stack: 
    stack.callback(callback_1) 
    stack.callback(callback_2, 100) 
    print(12345) 
    print('xxxx') 
print('退出縮進') 

運行效果如下圖所示：

可以看出以下特點：

1. 被添加的回調函數進入了一個棧，所以后添加的回調函數先調用
2. 回調函數會在結束縮進的時候被調用

現在我們來人工構造一個異常：

可以看到，即使縮進里面出現了報錯，回調函數仍然可以正常運行。等所有回調函數運行完成以后，Python 才會退出。

基于以上特點，我們就可以來重構最開始的代碼了：

import json 
import redis 
import datetime 
import pymongo 
import contextlib 
 
client = redis.Redis() 
handler = pymongo.MongoClient().test.data 
 
def add_ts(): 
    handler.insert_one({'finished': True, 'ts': datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}) 
 
def parse(): 
    with contextlib.ExitStack() as stack: 
        stack.callback(add_ts) 
        while True: 
            data = client.lpop('test') 
            if not data: 
                return 
            handler.insert_one(json.loads(data)) 

無論是正常運行結束還是運行過程中報錯，add_ts函數都會正常運行，確保始終增加一條日期數據。

本文轉載自微信公眾號「未聞Code」，可以通過以下二維碼關注。轉載本文請聯系未聞Code公眾號。