tempfile库:Python临时文件系统对象

Posted 一天一篇Python库

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了tempfile库:Python临时文件系统对象相关的知识,希望对你有一定的参考价值。

前言

在实际的项目处理中,往往我们并不需要创建文件,仅仅用于中转而已。这个时候在系统中频繁的创建中转文件,删除中转文件,不仅浪费系统的资源,而且容易被破坏或者篡改,这个时候用临时文件反而更好。

而Python给我们提供了临时文件操作库:tempfile。本篇将详细介绍临时文件的应用与处理。

创建临时文件(TemporaryFile)

一般来说,我们通过tempfile.TemporaryFile()函数创建临时文件,具体的代码如下所示:

import tempfile

with tempfile.TemporaryFile(mode='w+t') as temp:
    temp.write("My name is Li Yuanjing")
    temp.seek(0)
    print(temp.read())
    print(temp.name)

运行之后,效果如下:
临时文件
其中,mode表示以什么方式创建并打开临时文件,这里w+表示可以读写文件,t是以文本模式打开文件。默认写完内容后,文件句柄在末尾,要重新读取文件,需要使用seek回转重定位。

命名文件

当然,我们创建临时文件是需要使用的,也就是在程序中,我们需要给它一个名字,方便我们进行操作。我们将上面代码变更一下:

import tempfile
import pathlib

with tempfile.TemporaryFile(mode='w+t') as temp:
    temp.write("My name is Li Yuanjing")
    temp.seek(0)
    print(temp.read())
    f = pathlib.Path(temp.name)
print(f.name)
f.exists()

这里,我们用f变量标记了文件,后面操作的时候,可以使用f进行操作。当我们调用f.exists()函数时,默认临时文件会被删除。

临时目录(TemporaryDirectory)

在实际的项目中,我们可能并不仅仅只是创建一个临时文件,有时候也会创建一批量的临时文件,比如上传文件时。这个时候,我们就需要创建一个临时目录,在临时目录中创建文件进行操作。操作完成之后,只要关闭临时目录,就可以批量的关闭临时文件,比单个临时文件要方便许多。

下面,我们就来创建临时的目录,代码如下:

import tempfile
import pathlib

with tempfile.TemporaryDirectory() as temp:
    f = pathlib.Path(temp)
    print(f)
    a_file = f / 'a.txt'
    a_file.write_text("111111111111")
    b_file = f / 'b.txt'
    b_file.write_text("222222222222")
    c_file = f / 'c.txt'
    c_file.write_text("333333333333")
    print(a_file.read_text())
    print(b_file.read_text())
    print(c_file.read_text())
f.exists()

运行之后,效果如下:
临时目录

假脱机文件

如果我们处理的临时文件的数据较少,其实使用SpooledTemporaryFile可能更高效,因为它使用一个io.BytesIO或io.StringIO缓冲区在内存中保存内容,直到数据超过一定的大小,才写入磁盘,然后用TemporaryFile替代缓冲区。

具体使用方式如下:

import tempfile

with tempfile.SpooledTemporaryFile(max_size=1000, mode='w+t', encoding='utf-8') as temp:
    print(temp)
    temp.write('15648497987987987')
    temp.seek(0)
    print(temp.read())

运行之后,效果如下:
假脱机文件
这里的max_size就是决定文件小于这个值写入缓冲区,大于这个值写入TemporaryFile临时文件。

当然,我们可以强制将缓冲区写入临时文件,可以调用rollover()或fileno()函数。具体代码如下所示:

import tempfile

with tempfile.SpooledTemporaryFile(max_size=1000, mode='w+t', encoding='utf-8') as temp:
    print(temp)
    temp.write('15648497987987987')
    temp.seek(0)
    temp.rollover()
    print(temp.read())

如上面代码所示,加一句temp.rollover()就行。

预测名(NamedTemporaryFile)

从上面的代码,我们可以看到,临时文件单独创建的方式,文件名是随机生成的,并不是由程序员指定的,但有时候还需要在名字中包含一些可预测的部分,以便查找和检查文件来进行调用。

这个时候,指定文件名往往更好。我们指定文件名规律用NamedTemporaryFile()函数,代码如下:

import tempfile
import pathlib

tmp_file = tempfile.NamedTemporaryFile(prefix='image_', suffix='_png')
print(tmp_file.name)

运行之后,效果如下:
预测名
可以看出来,我们的文件名等于:prefix+random+suffix。当然,其实该函数还有一个dir参数,可以指定目录。完整的文件名等于:dir+prefix+random+suffix。

gettempdir()与gettempprefix()

在创建临时文件的时候,如果我们没有指定dir,那么该临时文件的路径会跟随平台和设置变化,也就是不确定的位置。这个时候,我们可能需要获取文件的路径以及其他参数,通过gettempdir()与gettempprefix()可以做到,具体代码如下:

import tempfile

print(tempfile.gettempdir())
print(tempfile.gettempdirb())
print(tempfile.gettempprefix())
print(tempfile.gettempprefixb())

运行之后,效果如下:
临时文件位置
gettempdir:返回包含所有临时文件的默认目录

gettempprefix:返回新文件和目录名的字符串前缀。

而这个2个方法还有一个后缀b的方法,用于返回bytes类型的字符串对象。当然如果你想自己设置临时目录,并不跟随系统,可以直接使用tempfile.tempdir进行设置。

import tempfile

tempfile.tempdir=r'd:\\temp'
print(tempfile.gettempdir())

运行之后,效果如下:
设置临时目录位置

以上是关于tempfile库:Python临时文件系统对象的主要内容,如果未能解决你的问题,请参考以下文章

tempfile库:Python临时文件系统对象

Python标准库tempfile的使用总结

tempfile:临时文件系统对象

Python tempfile (临时文件)

Python ——tempfile

[python] 创建临时文件-tempfile模块