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

2024-04-25 17:45:04 105

标准库中的tempfile模块定义用于创建临时文件和目录的函数。它们在操作系统文件系统定义的特殊临时目录中创建。例如，在Windows下，临时文件夹位于profile/AppData/Local/Temp中，而在Linux中，临时文件位于/tmp目录中。

在tempfile模块中定义了以下功能

TemporaryFile()

该函数在temp目录中创建一个临时文件，并返回一个文件对象，类似于内置open()函数。默认情况下，该文件以wb+模式打开，这意味着它可以同时用于读取/写入其中的二进制数据。重要的是，关闭文件对象后，将删除temp文件夹中的文件条目。以下代码显示了TemporaryFile()函数的用法。

>>> import tempfile
>>> f = tempfile.TemporaryFile()
>>> f.write(b'Welcome to nhooo')
>>> import os
>>> f.seek(os.SEEK_SET)
>>> f.read()
b'Welcome to nhooo'
>>> f.close()

下面的示例以w+模式打开TemporaryFile对象，以写入然后读取文本数据而不是二进制数据。

>>> ff = tempfile.TemporaryFile(mode = 'w+')
>>> ff.write('hello world')
>>> ff.seek(0)
>>> ff.read()
'hello world'
>>> ff.close()

命名为TemporaryFile()

此功能类似于TemporaryFile()方法。唯一的不同是，在操作系统的指定临时文件夹中可见带有随机文件名的文件。可以通过文件对象的名称属性来检索名称。该文件也将在关闭后立即删除。

>>> fo = tempfile.NamedTemporaryFile()
>>> fo.name
'C:\\Users\\acer\\AppData\\Local\\Temp\\tmpipreok8q'
>>> fo.close()

TemporaryDirectory()

此函数创建一个临时目录。您可以通过提及dir参数来选择此临时目录的位置。以下语句将在C：\\python36文件夹中创建一个临时目录。

>>> f = tempfile.TemporaryDirectory(dir = "C:/python36")
<TemporaryDirectory 'C:/python36\\ tmp9wrjtxc_'>

创建的目录出现在dir1文件夹中。通过调用cleanup()目录对象上的函数将其删除。

>>> f.name
'C:/python36\\tmp9wrjtxc_'
>>> f.cleanup()

mkstemp()

该功能还会创建一个类似于TemporaryFile()方法的临时文件。此外，可以在创建临时文件后添加后缀和前缀参数。与情况不同TemporaryFile()，创建的文件不会自动删除。应该手动将其删除。

>>> f = tempfile.mkstemp(suffix = '.tp')
C:\Users\acer\AppData\Local\Temp\tmpbljk6ku8.tp

mkdtemp()

此函数还在操作系统的temp文件夹中创建一个临时目录，并返回其绝对路径名。要明确定义其创建位置，请使用dir参数。该文件夹也不会自动删除。

>>> tempfile.mkdtemp(dir = "c:/python36")
'c:/python36\\tmpruxmm66u'

gettempdir()

此函数返回目录名称以存储临时文件。此名称通常从tempdir环境变量获得。在Windows平台上，通常为user/AppData/Local/Temp或Windowsdir/temp或systemdrive/temp。在Linux上，通常是/tmp。该目录用作dir参数的默认值。

>>> tempfile.gettempdir()
'C:\\Users\\acer\\AppData\\Local\\Temp'

在本文中，已经解释了tempfile模块中的功能。