用 TemporaryDirectory 隔离一次文件处理
【摘要】 用 Python 标准库管理临时目录,了解清理时机和使用边界。
用 TemporaryDirectory 隔离一次文件处理
处理压缩包、生成中间结果或测试文件转换时,临时文件的清理很容易被异常路径遗漏。Python 标准库的 tempfile.TemporaryDirectory 提供上下文管理器:退出 with 时清理目录及其中内容,即使块内抛出异常也会执行清理。
一个可运行的例子
from pathlib import Path
from tempfile import TemporaryDirectory
def count_nonempty_lines(text: str) -> int:
with TemporaryDirectory(prefix="line-count-") as directory:
input_path = Path(directory) / "input.txt"
input_path.write_text(text, encoding="utf-8")
with input_path.open(encoding="utf-8") as source:
return sum(1 for line in source if line.strip())
assert count_nonempty_lines("first\n\nsecond\n \n") == 2
print("非空行数:", count_nonempty_lines("first\n\nsecond\n"))
此例先将输入写入独立目录,再逐行读取;空白行通过 strip() 排除。return 会先触发上下文退出,再把结果返回给调用方。示例适合说明生命周期,实际只处理内存字符串时不必为了计数写临时文件。
边界与注意事项
临时目录路径只在 with 块内有效;需要保留的产物应在退出前复制到目标位置,并明确处理覆盖行为。清理也可能因权限、仍被占用的文件等原因失败,程序应根据实际平台处理异常。不要依赖临时目录作为持久存储,也不要把敏感内容的删除等同于安全擦除。非正常进程终止时,常规上下文清理不一定有机会执行。
本例在本地 Python 运行,断言通过并输出 非空行数: 2。更多参数和平台差异可参考 Python 标准库 tempfile 文档。
【声明】本内容来自华为云开发者社区博主,不代表华为云及华为云开发者社区的观点和立场。转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息,否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱:
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
评论(0)