Python使用atexit模块实现Golang的defer功能

package main

import "fmt"

func main() {
    defer fmt.Println("高老师总能分享出好东西")
    fmt.Println("执行结束")
    /*
    执行结束
    高老师总能分享出好东西
    */
}

import atexit

def exit_func(words):
    print(words)

# 将函数注册进去
atexit.register(exit_func, "高老师总能分享出好东西")
print("Hello")
print("World")
"""
Hello
World
高老师总能分享出好东西
"""

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，单身带俩娃
高老师总能分享出好东西
"""

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)￣)^只因大(出海版)
"""

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
----------------
"""

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")

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 功能
程序结束
"""