在Python程序运行过程中，我们常常需要存储一些临时的中间数据，比如数据处理的缓存、文件上传的暂存内容等，这类数据不需要长期保存，用完之后就可以删除。如果手动指定路径创建文件，很容易出现文件名重复、权限不足或者忘记删除导致磁盘空间占用的问题。Python内置的tempfile模块就是专门解决这类问题的工具，它可以帮我们创建安全、可靠的临时文件和临时目录，并且支持自动清理，避免很多潜在的隐患。
tempfile模块的核心优势
使用tempfile模块创建临时文件相比手动创建有以下几个明显优势：
- 自动生成唯一的文件名，避免文件名冲突
- 默认创建的文件和目录权限更严格，减少安全风险
- 支持自动清理机制，程序退出后自动删除临时资源
- 跨平台兼容性好，不需要适配不同操作系统的临时路径规则
创建临时文件的基本方法
创建匿名临时文件
最常用的方式是使用NamedTemporaryFile或者TemporaryFile，其中TemporaryFile创建的是没有文件名的临时文件，适合不需要通过文件路径访问的场景。
import tempfile
# 创建匿名临时文件，默认模式为w+b，即二进制读写模式
with tempfile.TemporaryFile(mode='w+t', encoding='utf-8') as tmp_file:
# 写入临时数据
tmp_file.write('这是临时存储的测试数据')
# 将文件指针移到开头，方便读取
tmp_file.seek(0)
# 读取临时文件内容
content = tmp_file.read()
print(content) # 输出：这是临时存储的测试数据
# with块结束后，临时文件会自动删除
创建有名字的临时文件
如果需要通过文件路径访问临时文件，或者需要让其他程序也能访问这个临时文件，可以使用NamedTemporaryFile，它会生成一个带有唯一文件名的临时文件。
import tempfile
import os
with tempfile.NamedTemporaryFile(mode='w+t', suffix='.txt', prefix='test_', delete=False) as named_tmp:
# 写入内容
named_tmp.write('有名称的临时文件内容')
# 获取临时文件的路径
tmp_path = named_tmp.name
print(f'临时文件路径：{tmp_path}')
print(f'文件是否存在：{os.path.exists(tmp_path)}') # 输出：True
# 此时文件不会被自动删除，因为设置了delete=False
print(f'with块结束后文件是否存在：{os.path.exists(tmp_path)}') # 输出：True
# 手动删除临时文件
os.remove(tmp_path)
上面的代码中，suffix参数指定文件后缀，prefix参数指定文件名前缀，delete=False表示退出with块后不自动删除文件，默认情况下delete是True，退出后会自动清理。
创建临时目录
如果需要存储多个临时文件，可以创建一个临时目录，tempfile模块提供了 TemporaryDirectory方法来创建临时目录，同样支持自动清理。
import tempfile
import os
with tempfile.TemporaryDirectory(suffix='_dir', prefix='tmp_') as tmp_dir:
print(f'临时目录路径：{tmp_dir}')
# 在临时目录下创建文件
file_path = os.path.join(tmp_dir, 'sub_file.txt')
with open(file_path, 'w', encoding='utf-8') as f:
f.write('临时目录下的文件内容')
print(f'临时目录下的文件是否存在：{os.path.exists(file_path)}') # 输出：True
# with块结束后，临时目录及其中的所有内容都会被自动删除
print(f'临时目录是否存在：{os.path.exists(tmp_dir)}') # 输出：False
临时文件的安全注意事项
使用tempfile模块时，还需要注意以下几个安全相关的问题：
- 尽量不要将
delete参数设置为False后忘记手动清理，避免临时文件长期占用磁盘空间 - 如果临时文件需要被其他进程访问，要注意权限设置，默认的临时文件权限是只有当前用户可读写
- 不要在临时文件中存储敏感信息，除非确认临时文件的存储路径和权限是安全的
- 使用with语句操作临时文件和目录，可以最大程度保证资源被正确清理，即使程序出现异常也会触发清理逻辑
常见场景示例
下面是一个实际的使用场景，比如我们需要处理一个大的CSV文件，先将其拆分到临时文件中做中间处理，最后再合并结果：
import tempfile
import csv
def process_large_csv(input_path):
temp_files = []
try:
# 创建多个临时文件存储拆分后的数据
with open(input_path, 'r', encoding='utf-8') as f:
reader = csv.reader(f)
header = next(reader)
# 每100行数据存到一个临时文件
batch = []
for i, row in enumerate(reader, 1):
batch.append(row)
if i % 100 == 0:
tmp = tempfile.NamedTemporaryFile(mode='w+t', suffix='.csv', delete=False)
writer = csv.writer(tmp)
writer.writerow(header)
writer.writerows(batch)
temp_files.append(tmp.name)
tmp.close()
batch = []
# 处理最后一批不足100行的数据
if batch:
tmp = tempfile.NamedTemporaryFile(mode='w+t', suffix='.csv', delete=False)
writer = csv.writer(tmp)
writer.writerow(header)
writer.writerows(batch)
temp_files.append(tmp.name)
tmp.close()
# 这里可以对临时文件做进一步处理
print(f'拆分出的临时文件数量：{len(temp_files)}')
return temp_files
finally:
# 处理完成后手动清理所有临时文件
for path in temp_files:
import os
if os.path.exists(path):
os.remove(path)
# 假设存在input.csv文件，调用上述函数
# process_large_csv('input.csv')
通过上面的示例可以看到，tempfile模块可以很好地支撑临时文件的创建和管理，只要遵循正确的使用方式，就能避免很多临时文件使用过程中的问题。