当前位置:主页 > 软件编程 > Python代码 >

Python临时文件创建之tempfile模块介绍

时间:2022-11-28 10:29:04 | 栏目:Python代码 | 点击:

一、tempfile模块简介

1、tempfile模块应用场景

Pythontempfile模块是用来创建临时文件或者文件夹的跨平台工具。在大型数据处理项目中,有的处理结果是不需要向用户最终展示的,但是它们的应用又是贯穿项目始终的,在这种情况下,我们就需要使用tempfile模块来解决这种问题。

2、tempfile模块的方法

tempfile模块主要包括三类函数:

函数名 描述
TemporaryFile和NamedTemporaryFile 高级临时文件对象创建函数
SpooledTemporaryFile 假脱机模式高级临时文件对象创建函数
TemporaryDirectory 高级临时目录创建函数
函数名 描述
mkstemp() 底层临时文件创建函数
mkdtemp() 底层临时目录创建函数
函数名 描述
gettempdir() 以文本格式返回临时文件夹名函数
gettempdirb() 以二进制格式返回临时文件夹名函数
gettempprefix 以文本格式返回临时文件名前缀
gettempprefixb 以二进制格式返回临时文件名前缀

tempfile的所有用户可调用函数和构造函数都带有额外参数,通过这些参数可以实现对临时文件目录和位置的管理。此模块会在共享临时目录中安全地创建临时文件,并给创建临时文随机起个名。

二、tempfile模块主要函数介绍

高级临时文件对象创建函数:Temporaryfile和NamedTemporaryFile

1、Temporaryfile函数

Temporaryfile函数返回一个类文件对象用作临时存储区,它使用与mkstemp()函数相同规则的安全重建文件模式,它一关闭就会被销毁(包括针对垃圾收集对象的隐式关闭)。需要注意的是,在Unix系统环境,文件的目录要么根本不创建,要么在文件创建后立即删除。也就是说,代码不会依赖于此函数创建的临时文件,包括它们的名称,这也是这个函数与NamedTemporaryfile函数的不同之处。

TemporaryFile函数的调用格式为:

TemporaryFile(mode='w+b', buffering=-1, encoding=None,
    newline=None, suffix=None, prefix=None, *, error=None)

函数参数选项:

下面给出一个典型例子:

import tempfile

# 下面创建一个临时文件并写入一些数据
fp = tempfile.TemporaryFile()
fp.write(bb'Hello Xiaoliang!')

# 读取数据
fp.seek(0);  fp.read()

另外,该函数生成的对象可以用作上下文管理器(参见示例)。完成上下文管理或销毁文件对象后,临时文件将从文件系统中删除。

import tempfile

with tempfile.TemporaryFile() as fp:
    fp.write(b'Hello Xiaoliang!')
    fp.seek(0)
    fp.read()

2、NamedTemporaryfile函数

除了NamedTemporaryfile()函数实现了在文件系统中只有一个可见名之外,其作用与TemporaryFile()功能完全相同。执行完该函数后,我们可以从返回的类文件对象的name属性中检索文件名称。在Unix系统上,在命名的临时文件处于打开状态时,可以使用该名称打开生成的文件。

NamedTemporaryFile函数的调用格式与TemporaryFile函数的非常相似,唯一不同的是具有了一个delete参数:

NamedTemporaryFile(mode='w+b', buffering=-1, encoding=None,
    newline=None, suffix=None, prefix=None, dir=Nonde, delete=True,
    *, error=None)

delete参数选项:如果该参数为True(默认),则文件一关闭则被删除。返回的对象始终是一个类文件对象,其文件属性是底层的真实文件对象。也就是说,这个类文件对象可以在with上下文管理中使用,就像普通文件一样。

3、假脱机模式高级临时文件对象创建函数:SpooledTemporaryfile

SPOOL技术简介:

SPOOL是Simultaneous Peripheral Operation On-Line
(即外部设备联机并行操作)的缩写,它是关于慢速字符设备如何与计算机主机交换信息的一种技术,通常称为“假脱机技术”。其实,严格意义上说,SPOOL系统既不同于脱机方式,也不同于直接耦合方式,SPOOL技术实际上是一种外围设备同时联机操作技术,又称为排队转储技术。它在输入和输出之间增加了“输入井”和“输出井”的排队转储环节,以消除用户的“联机”等待时间。

SpooledTemporaryFile()函数除了数据会在内存中进行假脱机外,与TemporaryFile()完全相同。直到文件大小超过max_size或直到调用文件的fileno()方法,此时使用与TemporaryFile()一样的操作,将内容写入磁盘。

该函数同样返回一个类文件对象,其_file属性可以是io.BytesIO(二进制模式)、io.TextIOWrapper(文本模式)对象或真实的文件对象(调用roller()函数)。同样的,这个类文件对象可以在with上下文管理中使用,就像普通文件一样。

该函数还有一个额外的方法rollover(),它使创建的文件滚寻磁盘上的内容,而不管其大小。SpooledTemporaryFile函数的调用格式与TemporaryFile()函数一样,这里就不介绍了。

4、高级临时目录创建函数:TemporaryDirectory

TemporaryDirectory函数使用与mkdtemp()相同的规则安全地创建临时目录。生成的对象可以用作上下文管理器(这里给出示例)。完成上下文或销毁临时目录对象后,新创建的临时目录及其所有内容将从文件系统中删除。其调用格式如下所示:

TemporaryDirectory(suffix=None, prefix=None, dir=None)

调用该函数后,创建的目录名可以从返回对象的name属性中检索到。当返回的对象作为上下文管理器时,该名称将被分配给with语句中as子句的目标。另外,可以通过调用cleanup()方法显式清理目标。

5、底层临时文件/目录创建函数:mkstemp和mkdtemp

底层临时文件创建函数mkstemp()

mkstemp()函数以尽可能安全的方式创建临时文件,与TemporaryFile()函数不同,用户需要负责临时文件的删除。其调用格式如下所示:

mkstemp(suffix=None, prefix=None, dir=None, text=False)

函数参数选项:

mkstemp()返回一个元组,该元组包含一个用于打开文件的操作系统级别(OS-leve)句柄(有os.open()函数返回)和该文件绝对路径名。

底层临时目录创建函数mkdtemp()

mkdtemp()函数以尽可能安全的方式创建临时目录。只有通过用户ID才能读取、写入和搜索目录,同时用户负责删除临时目录及其内容。其调用格式和参数与mkstemp()相同,这里就不展开介绍了。注意,mkdtemp()函数只返回新建目录的绝对路径。

三、tempfile模块的文件/文件夹属性操作函数

1、返回临时文件夹名函数:gettempdir和gettempdirb

gettempdir()返回存放临时文件的文件夹名字,该函数的返回值是tempfile内的全局变量,也就是说,gettempdir()返回的文件夹名将作为tempfile模块中的所有函数dir参数的默认值。

Python内部搜索一个标准目录列表,用户可以在其中创建文件的目录:

Windows系统,目录为:C:\TEMP, C:\TMP, \TEMP, and \TMP;
其他系统,目录为:/tmp, /var/tmp, and /usr/tmp。

另外,还有一个与gettempdir()作用相同,只是返回的格式二进制的函数gettempdirb()。

2、返回临时文件前缀函数:gettempprefix和gettempprefix

gettempprefix以文本格式返回创建的临时文件的前缀;gettempprefixb以二进制格式返回创建的临时文件的前缀。

您可能感兴趣的文章:

相关文章