在Python项目里，临时资源通常指运行时创建的临时文件、套接字、数据库连接或内存映射对象。如果这些资源没有在合适的时机释放，不仅会造成磁盘空间浪费，还可能因临时文件包含调试信息或凭证带来安全风险。本文围绕临时文件这一最常见场景，说明几种稳妥的清理策略。
使用tempfile模块与上下文管理器
Python标准库中的tempfile提供了TemporaryFile和NamedTemporaryFile，它们天生支持上下文管理器协议。当with代码块执行完毕，无论是否发生异常，文件都会被自动关闭并删除。这是避免文件泄露的首选方案。
下面示例创建一个匿名临时文件并写入数据，离开with块后文件立即消失，无需手动调用删除函数：
import tempfile
with tempfile.TemporaryFile(mode='w+t', encoding='utf-8') as tf:
tf.write('临时敏感数据')
tf.seek(0)
print(tf.read())
# 此处文件已自动清理，磁盘上无残留
如果需要让其他进程通过文件名访问，可以使用NamedTemporaryFile，并设置delete=False延迟删除，随后在finally中自行清理。但更推荐直接依赖上下文管理，降低出错概率。
try-finally与atexit的兜底方案
在无法使用with语句的旧式代码或复杂控制流中，可以用try-finally保证清理逻辑一定执行。atexit模块则适合注册进程退出时的全局清理钩子，处理那些生命周期跨越多个函数的资源。
以下代码演示try-finally模式，即便写入过程出现异常，临时文件仍会被移除：
import os
import tempfile
path = None
try:
fd, path = tempfile.mkstemp(suffix='.tmp')
with os.fdopen(fd, 'w') as f:
f.write('测试内容')
# 模拟后续处理可能报错
raise RuntimeError('处理失败')
except Exception as e:
print('捕获异常:', e)
finally:
if path and os.path.exists(path):
os.remove(path)
print('已清理临时文件')
atexit方案适合在程序启动时统一登记清理函数，例如删除整个临时目录。但要注意atexit钩子在强制杀进程时不会运行，因此不能替代上下文管理器做关键资源保护。
权限与目录策略的安全考量
临时文件默认存放在系统临时目录，多用户环境下若权限设置不当，其他用户可能读取内容。tempfile在类Unix系统上会以0600权限创建文件，已具备一定安全性。但在共享容器或自定义目录时，应显式指定mode并确认umask。
通过下表对比不同策略的适用场景：
| 策略 | 异常安全 | 适用场景 |
|---|---|---|
| with临时文件 | 强 | 函数内短生命周期资源 |
| try-finally | 强 | 兼容旧代码或复杂流 |
| atexit钩子 | 弱（强杀无效） | 进程级全局清理 |
综合来看，优先用tempfile的上下文管理器，在必须跨作用域持有资源时结合try-finally，把atexit作为补充而非主力。这样既能防止文件泄露，也符合最小权限原则。
常见误区与改进
一个典型错误是在函数开头创建临时文件，只在return前删除，却忘了异常分支。这种写法在测试正常时没问题，一旦业务逻辑抛错就会留下垃圾文件。改进方式就是前面提到的上下文或finally。
另一个误区是盲目调用code>shutil.rmtree清理临时目录，却未做路径校验，可能误删相似名称的系统目录。任何删除操作前都应确认路径由tempfile生成或位于明确白名单内。
import tempfile, os, shutil base = tempfile.gettempdir() my_dir = os.path.join(base, 'my_app_tmp') os.makedirs(my_dir, exist_ok=True) # 仅删除自己应用创建的子目录 shutil.rmtree(my_dir, ignore_errors=True)
养成资源随用随清的习惯，并借助语言自带机制而非手工管理，才能写出健壮且安全的Python程序。