Python：使用 tempfile 生成临时文件和目录

古城微笑少年丶 2023-09-26 23:23 78阅读 0赞

Python 拥有丰富的标准库集合来执行各种任务。在 Python 中，有一个名为tempfile 的模块允许我们创建和操作临时文件和目录。

我们可以使用tempfile创建临时文件和目录，用于在程序执行过程中存放临时数据。该模块具有允许我们创建、读取和操作临时文件和目录的功能。

在本文中，我们将回顾 tempfile 模块的基础知识，包括创建、读取和操作临时文件和目录，以及在我们完成它们后的清理。

创建临时文件  
tempfile 模块包含一个名为的函数TemporaryFile()，它允许我们创建一个临时文件用作临时存储。只需输入以下代码即可创建一个临时文件。

import tempfile
    
    # Creating a temporary file
    temp_file = tempfile.TemporaryFile()
    
    print(f'Tempfile object: {temp_file}')
    print(f'Tempfile name: {temp_file.name}')
    
    # Closing the file 
    temp_file.close()

首先，我们导入了模块，因为它是标准 Python 库的一部分，所以我们不需要安装它。

然后我们使用该TemporaryFile()函数创建一个临时文件，我们将其保存在temp\_file变量中。然后我们添加打印语句来显示临时文件对象和临时文件名。

最后，我们使用close()函数关闭文件，就像我们关闭任何其他文件一样。这将自动清理并删除文件。

输出

Tempfile object: <tempfile._TemporaryFileWrapper object at 0x0000015E54545AE0>
    Tempfile name: C:\Users\SACHIN\AppData\Local\Temp\tmp5jyjavv2

当我们打印对象时，我们得到了对象在内存中的位置，当我们打印临时文件的名称时，我们得到了带有整个路径的temp\_file随机名称。tmp5jyjavv2

我们可以指定dir参数在指定目录下创建一个临时文件。

import tempfile
    
    # Creating a temporary file in the cwd
    named_file = tempfile.TemporaryFile(
        dir='./'
    )
    
    print('File created in the cwd')
    print(named_file.name)
    
    # Closing the file
    named_file.close()

该文件将在当前工作目录中创建。

File created in the cwd
    D:\SACHIN\Pycharm\tempfile_module\tmptbg9u6wk

编写文本并阅读  
该TemporaryFile()函数将模式参数设置’w+b’为默认值，以二进制模式读写文件。我们还可以使用’w+t’模式将文本数据写入临时文件。

import tempfile
    
    # Creating a temporary file
    file = tempfile.TemporaryFile()
    
    try:
        # Writing into the temporary file
        file.write(b"Welcome to GeekPython.")
        # Position to start reading the file
        file.seek(0)  # Reading will start from beginning
        # Reading the file
        data = file.read()
        print(data)
    finally:
        # Closing the file
        file.close()
    
    ----------
    b'Welcome to GeekPython.'

在上面的代码中，我们创建了一个临时文件，然后使用函数write()将数据写入其中，以字节格式传递字符串（如’b’内容前的前缀所示）。

然后我们使用该函数读取内容，但首先，我们使用（从头开始）函数read()指定开始读取内容的位置，最后，我们关闭文件。seek(0)

使用上下文管理器  
我们还可以使用关键字等上下文管理器创建临时文件with。此外，以下示例将向我们展示如何将文本数据写入和读取到临时文件中。

import tempfile
    
    # Using with statement creating a temporary file
    with tempfile.TemporaryFile