如何基于程序状态实现单atexit函数的差异化输出

嘿，这个问题我刚好碰到过！要给atexit()注册的单个函数传递程序状态（比如整数），不用注册多个函数，下面几种方法都能搞定，按优雅程度排序：

1. 闭包（最推荐，无全局污染） #

闭包可以让内部函数访问外层函数的变量，完美适配atexit要求的无参回调：

import atexit

def make_exit_handler(program_status):
    # 内部函数会记住外层的program_status参数
    def exit_callback():
        if program_status == 0:
            print("🎉 程序正常退出啦")
        elif program_status == 1:
            print("❌ 程序因错误退出")
        else:
            print(f"⚠️ 程序以状态码 {program_status} 退出")
    return exit_callback

# 模拟程序运行中得到的状态
current_status = 0  # 这里可以根据业务逻辑动态修改

# 注册闭包返回的回调函数
atexit.register(make_exit_handler(current_status))

如果你的状态会在程序运行过程中动态变化，用**可变对象（比如列表）**包裹状态值，就能让回调拿到最新状态：

import atexit

def make_exit_handler(status_container):
    def exit_callback():
        latest_status = status_container[0]
        print(f"程序最终退出状态：{latest_status}")
    return exit_callback

# 用列表包裹整数（可变对象）
status = [0]
atexit.register(make_exit_handler(status))

# 后面程序运行中修改状态
status[0] = 2

2. functools.partial绑定参数 #

用partial可以直接把参数和目标函数绑定，生成一个无参的可调用对象，非常简洁：

import atexit
from functools import partial

def exit_handler(program_status):
    if program_status == 0:
        print("程序正常结束")
    elif program_status == 1:
        print("程序出错退出")
    else:
        print(f"退出状态码：{program_status}")

current_status = 1
# 绑定参数后注册
atexit.register(partial(exit_handler, current_status))

3. 全局变量（简单但不推荐） #

如果你的程序逻辑很简单，也可以用全局变量存状态，但要注意全局变量容易在复杂程序里引发意外问题：

import atexit

# 全局变量存储状态
program_status = 0

def exit_handler():
    # 声明使用全局变量
    global program_status
    if program_status == 0:
        print("正常退出")
    else:
        print(f"异常退出，状态码{program_status}")

atexit.register(exit_handler)

# 程序运行中修改全局状态
# program_status = 1

以上几种方法都能实现只用一个注册函数，根据程序状态输出不同内容，优先选闭包或者partial的方式哦～

内容的提问来源于stack exchange，提问作者Kevin Gardenhire