在 Go 里面可以通过 defer 语句让函数在结束时执行预定义好的一些操作，举个例子。
package main import "fmt" func main() { defer fmt.Println("高老师总能分享出好东西") fmt.Println("执行结束") /* 执行结束 高老师总能分享出好东西 */ }
这个功能非常方便，比如打开文件之后可以执行 defer fp.Close()，这样函数结束时会自动关闭文件句柄。那么在 Python 里面可不可以实现类似的功能呢？本次来聊一聊 atexit 模块，它能帮我们实现类似的效果。
import atexit def exit_func(words): print(words) # 将函数注册进去 atexit.register(exit_func, "高老师总能分享出好东西") print("Hello") print("World") """ Hello World 高老师总能分享出好东西 """
通过 atexit.register 将函数注册进去之后，会在程序结束之前执行，当然也可以同时注册多个。
import atexit def exit_func(words): print(words) # 将函数注册进去 atexit.register(exit_func, "高老师总能分享出好东西") atexit.register(exit_func, "S 老师今年 18，单身带俩娃") atexit.register(exit_func, "只因^(*￣(oo)￣)^只因大(出海版)") print("Hello") print("World") """ Hello World 只因^(*￣(oo)￣)^只因大(出海版) S 老师今年 18，单身带俩娃 高老师总能分享出好东西 """
如果同时注册了多个函数，那么会按照先入后出的顺序执行。非常简单，其实 atexit 模块就是将我们注册的函数保存在了一个数组中，程序结束的时候，从后往前依次执行。
既然可以注册函数，那么也可以取消注册。
import atexit def exit_func1(words): print(words) def exit_func2(words): print(words) atexit.register(exit_func1, "高老师总能分享出好东西") atexit.register(exit_func1, "S 老师今年 18，单身带俩娃") atexit.register(exit_func2, "只因^(*￣(oo)￣)^只因大(出海版)") # 取消注册，所有注册的 exit_func1 函数都会被删除 atexit.unregister(exit_func1) """ 只因^(*￣(oo)￣)^只因大(出海版) """
而它的逻辑也很简单，就是遍历数组，如果和指定的函数相等，那么就删掉。我们看一下源代码。
如果你想将注册的函数全部取消掉，那么也可以调用 _clear() 函数。
import atexit def exit_func1(words): print(words) def exit_func2(words): print(words) atexit.register(exit_func1, "高老师总能分享出好东西") atexit.register(exit_func1, "S 老师今年 18，单身带俩娃") atexit.register(exit_func2, "只因^(*￣(oo)￣)^只因大(出海版)") atexit._clear()
此时程序不会有任何输出，因为注册的函数全部被清空了，同样可以看一下它的源代码。
最后就是函数的调用时机，我们注册的函数在程序结束时才会调用，可不可以让它们在任意时刻调用呢？
import atexit def exit_func1(words): print(words) def exit_func2(words): print(words) atexit.register(exit_func1, "AAA") atexit.register(exit_func1, "BBB") # 调用注册的函数，调用之后函数会被删除 atexit._run_exitfuncs() print("++++++++++++++++") atexit.register(exit_func2, "CCC") atexit._run_exitfuncs() print("----------------") """ BBB AAA ++++++++++++++++ CCC ---------------- """
输出结果表明，一旦调用了 _run_exitfuncs，所有注册的函数会立即被调用。我们看一下源代码。
以上就是 atexit 模块的用法，那我们如何基于它实现 Golang 的 defer 呢？
from typing import Callable import atexit def defer(func: Callable, *args, **kwargs): atexit.register(func, *args, **kwargs) def get_file_content(file_path): fp = open(file_path, encoding="utf-8") defer(fp.close) # 注册函数 content = fp.read() # do something ... atexit._run_exitfuncs() # 触发注册函数执行 get_file_content("config.py")
不过这个例子明显有点刻意了，因为必须要在函数的结尾调用 atexit._run_exitfuncs，而之所以要实现 Go 的 defer，就是为了避免遗忘某些逻辑。
如果每次都要在函数结尾调用 atexit._run_exitfuncs，那还不如不用，于是我们可以考虑使用装饰器。
from typing import Callable from functools import wraps import atexit def defer(func: Callable, *args, **kwargs): atexit.register(func, *args, **kwargs) # 给函数赋予 defer 功能 def enable_defer(func): @wraps(func) def inner(*args, **kwargs): ret = func(*args, **kwargs) atexit._run_exitfuncs() return ret return inner @enable_defer # 通过装饰器，让函数支持 defer 功能 def get_file_content(file_path): fp = open(file_path, encoding="utf-8") # 注册函数 defer(fp.close) defer(print, "get_file_content 函数实现了 defer 功能") content = fp.read() print("函数执行结束") get_file_content("config.py") print("程序结束") """ 函数执行结束 get_file_content 函数实现了 defer 功能 程序结束 """
输出结果表明，在函数结束后，通过 defer 注册的函数执行了。
以上就是用 Python 实现 Go 的 defer，不过在工作中还是不建议这么做，没啥必要，这里只是想分享一下 atexit 模块。
