内容摘要：熟悉 Golang 的同学都知道，Golang 里面有一个关键词叫做defer，它可以实现延迟调用。实际上在 Python 里面也有相关的语法，那就是contextlib.ExitStack。我们来看

熟悉 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(退出缩进) 

运行效果如下图所示：

可以看出以下特点：

被添加的回调函数进入了一个栈，所以后添加的回调函数先调用 回调函数会在结束缩进的亿华云时候被调用

现在我们来人工构造一个异常：

可以看到，即使缩进里面出现了报错，回调函数仍然可以正常运行。等所有回调函数运行完成以后，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公众号。