python如何在程序退出前执行代码

# Python如何在程序退出前执行代码

在Python开发中，有时我们需要在程序退出前执行一些清理工作（如关闭文件、释放资源或发送日志）。本文将介绍4种实现方式，并分析其适用场景。

## 一、使用`atexit`模块（推荐）

Python标准库中的`atexit`模块专门用于注册程序退出时的回调函数：

```python
import atexit

def cleanup():
    print("执行清理工作...")

atexit.register(cleanup)  # 注册退出函数

特点： - 支持注册多个函数，按注册的逆序执行 - 适用于正常退出场景（如sys.exit()或主程序执行完毕） - 不适用于强制终止（如os._exit()或kill信号）

二、try...finally语句

对于代码块级别的清理，可以使用finally子句：

try:
    # 主程序代码
    print("程序运行中...")
finally:
    print("无论是否异常都会执行")

适用场景： - 需要确保特定代码段执行后立即清理 - 处理可能抛出异常的代码块

三、上下文管理器（with语句）

通过实现__enter__和__exit__方法创建上下文管理器：

class ResourceManager:
    def __enter__(self):
        print("分配资源")
        return self
    
    def __exit__(self, exc_type, exc_val, exc_tb):
        print("释放资源")

with ResourceManager() as r:
    print("使用资源")

优势： - 资源获取与释放自动配对 - 支持异常处理（通过__exit__参数）

四、信号处理（针对强制终止）

对于SIGTERM等信号，可通过信号处理实现优雅退出：

import signal, sys

def handle_exit(signum, frame):
    print(f"收到信号 {signum}，执行退出处理")
    sys.exit(0)

signal.signal(signal.SIGTERM, handle_exit)

注意： - 对SIGKILL（kill -9）无效 - 需配合atexit使用才能覆盖所有场景

最佳实践建议

1. 常规清理使用atexit + finally组合
2. 资源管理优先采用上下文管理器
3. 服务类程序应实现信号处理
4. 注意处理异常情况下的资源泄漏

测试提示：可通过故意抛出异常或发送SIGTERM信号测试不同场景下的执行情况。 “`

（全文约560字）