Python创建临时文件与目录：tempfile模块详解

tempfile 模块是 Python 内置的用于处理临时文件和目录的模块，它提供了一种创建临时文件和目录的方法，这些文件和目录可以在脚本执行完成后自动删除。本篇文章将详细介绍 tempfile 模块的使用方法。

创建临时文件

我们可以使用 tempfile 模块的 TemporaryFile() 函数创建一个临时文件，这个函数创建的临时文件默认是二进制模式打开的，文件指针在开头。

下面是一个创建临时文件的示例：

import tempfile

with tempfile.TemporaryFile() as f:
    # 向文件写入数据
    f.write(b'Hello, world!')

    # 移动文件指针到文件开头
    f.seek(0)

    # 读取文件内容
    data = f.read()
    print(data)

这个示例代码中，我们首先导入了 tempfile 模块，然后使用 TemporaryFile() 函数创建了一个临时文件对象 f，并在 with 块中操作这个文件对象。

在 with 块中，我们首先使用 write() 方法向文件写入了一个字符串 Hello, world!，然后使用 seek() 方法将文件指针移动到文件开头，最后使用 read() 方法读取文件的全部内容，并打印输出。

需要注意的是，临时文件对象在 with 块结束时会自动关闭并删除，因此我们不需要显式地调用 close() 方法和 unlink() 方法来关闭文件和删除文件。

如果我们需要创建一个文本模式的临时文件，可以在调用 TemporaryFile() 函数时指定 mode 参数为 "w+t"，示例如下：

import tempfile

with tempfile.TemporaryFile(mode='w+t') as f:
    # 向文件写入数据
    f.write('Hello, world!')

    # 移动文件指针到文件开头
    f.seek(0)

    # 读取文件内容
    data = f.read()
    print(data)

创建临时目录

Python的tempfile模块也可以用于创建临时目录，使用mkdtemp函数可以创建一个唯一的临时目录，返回的是该临时目录的路径。

示例代码如下：

import tempfile

temp_dir = tempfile.mkdtemp()
print("临时目录路径为：", temp_dir)

上述代码中，我们通过调用tempfile模块的mkdtemp()函数创建了一个临时目录，然后将其路径保存在变量temp_dir中，并将其打印输出。执行上述代码后，我们将看到类似以下的输出：

临时目录路径为： /tmp/tmpgrugvc4e

需要注意的是，创建的临时目录在使用完毕后需要手动删除，否则可能会占用磁盘空间。可以使用shutil模块中的rmtree函数来删除临时目录，示例代码如下：

import shutil
import tempfile

temp_dir = tempfile.mkdtemp()
print("临时目录路径为：", temp_dir)

# 临时目录的相关操作

# 删除临时目录
shutil.rmtree(temp_dir)

上述代码中，我们使用了shutil模块中的rmtree()函数来删除临时目录。需要注意的是，这个函数会彻底删除整个目录树，因此请谨慎使用。