17370845950

如何创建临时目录并在 with 结束后递归删除
tempfile.TemporaryDirectory 是 Python 3.2+ 推荐的临时目录管理方式,with 语句确保自动创建与递归删除(含异常时),比 mkdtemp + rmtree 更安全;需注意 Windows 路径长度、文件占用及权限问题,支持 prefix/suffix/dir 参数定制

,禁用 NFS 路径。

tempfile.TemporaryDirectory 最省心

Python 3.2+ 自带的 tempfile.TemporaryDirectory 就是专为这个场景设计的:自动创建临时目录,with 块退出时自动递归删除,包括非空目录。它比手写 mkdtemp + rmtree 更安全,因为即使内部抛异常,__exit__ 仍会触发清理。

常见错误是误用 tempfile.mkdtemp() 后手动删——容易漏掉 except 分支或忘记加 shutil.rmtree,导致残留目录。

  • 必须用 with 语句进入,不能只调 TemporaryDirectory() 然后赋值给变量
  • 返回对象有 .name 属性,就是目录路径,直接传给 os.chdiropen 都可以
  • 如果 with 块里发生异常,删除仍会执行;但若删除过程本身失败(如权限不足、文件被占用),会抛出新异常并压制原异常(Python 3.7+ 改为附加到 __context__
import tempfile
import os

with tempfile.TemporaryDirectory() as tmpdir: print("临时目录:", tmpdir) with open(os.path.join(tmpdir, "test.txt"), "w") as f: f.write("hello")

退出 with 后 tmpdir 及其所有内容已被递归删除

Windows 下要注意长路径和文件占用

在 Windows 上,TemporaryDirectory 删除失败最常见原因不是代码逻辑问题,而是系统级限制:路径过长(>260 字符)、文件正被其他进程打开(比如资源管理器窗口停留在该目录)、或某些文件设置了只读/系统属性。

这些情况不会提前报错,而是在 with 结束时才暴露,且错误信息往往只显示 PermissionErrorWindowsError: [Error 5],不指明具体哪个文件卡住。

  • 避免在临时目录内启动子进程并让它后台运行(如 subprocess.Popen(..., shell=True) 开 cmd 窗口),否则目录可能被锁住
  • 测试时可先用 os.listdir(tmpdir) 检查是否真为空,再人工验证是否有隐藏文件(如 Thumbs.db
  • 如需更高容错,可捕获 OSError 并重试几次(配合 time.sleep),但别无脑循环——根本原因得从进程隔离或路径长度入手

需要自定义前缀或位置时怎么设参数

TemporaryDirectory 支持 suffixprefixdir 三个关键字参数,行为和 mkdtemp 一致,但注意:dir 是父目录路径,不是最终路径;prefix 默认为空,设为 "myapp_" 会让目录名类似 myapp_abc123

容易忽略的是:如果指定的 dir 不存在,会抛 FileNotFoundError;如果 dir 存在但无写权限,则抛 PermissionError —— 这些都在 __enter__ 阶段报,不会等到 __exit__

  • prefixsuffix 只影响目录名生成,不影响内容,也不保证唯一性(底层仍靠随机字符串)
  • 不要把 dir 设成 /tmp 以外的 NFS 或网络挂载点,某些文件系统对 unlinkrmdir 的原子性支持不佳,可能造成清理不彻底
  • 若需跨平台稳定行为,避免依赖 os.environ["TEMP"],优先用 tempfile.gettempdir()

不用 with 时递归删除要小心 onerror

如果因兼容旧 Python 或特殊流程无法用 TemporaryDirectory,改用手动 mkdtemp + shutil.rmtree,必须显式处理删除失败。直接调 rmtree 在 Windows 上遇到只读文件会崩,得靠 onerror 回调先改权限。

这不是“多写几行”的事,而是权限模型差异带来的必选项:Linux 下只读文件能直接删,Windows 不行。

  • onerror 函数签名固定为 func, path, exc_info,其中 func 是失败的系统调用(如 os.remove),path 是目标路径
  • 典型修复逻辑是:对 os.remove 失败的文件,先 os.chmod(path, stat.S_IWRITE) 再重试;对 os.rmdir 失败的目录,同样改权限后重试
  • 别在 onerror 里吞掉所有异常,至少 log 一下 path,否则调试时完全不知道哪卡住了

临时目录的“自动清理”听着简单,实际落地时 Windows 权限、文件句柄泄漏、NFS 语义差异这三类问题占了八成以上故障;写死 rmtree 调用不如先确认 TemporaryDirectory 是否真不适用。