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

标准库中的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模块中的功能。