tempfile库

tempfile 是 Python 标准库中的一个模块，提供了创建临时文件和目录的功能。这些临时文件和目录在使用完之后可以自动删除，非常适合用于临时存储数据或中间结果。以下是 tempfile 库的一些常见用法和示例。

1. 创建临时文件

使用 NamedTemporaryFile

NamedTemporaryFile 会创建一个带有文件名的临时文件，这个文件在关闭后会被自动删除。

import tempfile

# 创建一个临时文件

with tempfile.NamedTemporaryFile(delete=True) as temp_file:
    print(f"Temporary file created: {temp_file.name}")
    temp_file.write(b'Hello, World!')
    temp_file.seek(0)
    print(temp_file.read())

文件关闭后会自动删除

使用 TemporaryFile

TemporaryFile 创建的临时文件在关闭后也会被自动删除，但不一定有文件名（在某些操作系统上）。

import tempfile

# 创建一个临时文件

with tempfile.TemporaryFile() as temp_file:
    temp_file.write(b'Hello, World!')
    temp_file.seek(0)
    print(temp_file.read())

文件关闭后会自动删除

2. 创建临时目录

使用 TemporaryDirectory

TemporaryDirectory 创建一个临时目录，这个目录在使用完后会自动删除。

import tempfile
import os

# 创建一个临时目录

with tempfile.TemporaryDirectory() as temp_dir:
    print(f"Temporary directory created: {temp_dir}")
    temp_file_path = os.path.join(temp_dir, 'temp_file.txt')
    with open(temp_file_path, 'w') as temp_file:
        temp_file.write('Hello, World!')

目录关闭后会自动删除

3. 获取临时文件和目录的路径

使用 gettempdir

gettempdir 返回用于存储临时文件的目录路径。

import tempfile

temp_dir = tempfile.gettempdir()
print(f"Temporary directory: {temp_dir}")

4. 自定义临时文件和目录

你可以通过传递参数来自定义临时文件和目录的前缀、后缀和目录。

import tempfile

# 自定义前缀、后缀和目录

with tempfile.NamedTemporaryFile(prefix='my_prefix_', suffix='_my_suffix', dir='/tmp') as temp_file:
    print(f"Custom temporary file: {temp_file.name}")

with tempfile.TemporaryDirectory(prefix='my_prefix_', suffix='_my_suffix', dir='/tmp') as temp_dir:
    print(f"Custom temporary directory: {temp_dir}")

5. 手动管理临时文件和目录

如果你不使用上下文管理器（with 语句），你需要手动删除临时文件和目录。

import tempfile
import os

# 创建临时文件
temp_file = tempfile.NamedTemporaryFile(delete=False)
try:
    print(f"Temporary file created: {temp_file.name}")
    temp_file.write(b'Hello, World!')
    temp_file.seek(0)
    print(temp_file.read())
finally:
    temp_file.close()
    os.remove(temp_file.name)  # 手动删除临时文件

# 创建临时目录
temp_dir = tempfile.mkdtemp()
try:
    print(f"Temporary directory created: {temp_dir}")
    temp_file_path = os.path.join(temp_dir, 'temp_file.txt')
    with open(temp_file_path, 'w') as temp_file:
        temp_file.write('Hello, World!')
finally:
    os.rmdir(temp_dir)  # 手动删除临时目录

总结

tempfile 模块提供了创建和管理临时文件和目录的便捷方法，特别适合用于需要临时存储数据的场景。通过上下文管理器（with 语句），你可以确保临时文件和目录在使用完后自动删除，避免资源泄漏。