善用tempfile库创建python进程中的临时文件

DechinPhy

发布于 2021-05-21 15:00:27

1.2K0

发布于 2021-05-21 15:00:27

技术背景

临时文件在python项目中时常会被使用到，其作用在于随机化的创建不重名的文件，路径一般都是放在Linux系统下的/tmp目录。如果项目中并不需要持久化的存储一个文件，就可以采用临时文件的形式进行存储和读取，在使用之后可以自行决定是删除还是保留。

tempfile库的使用

tempfile一般是python内置的一个函数库，不需要单独安装，这里我们直接介绍一下其常规使用方法：

# tempfile_test.py

import tempfile

file = tempfile.NamedTemporaryFile()
name = str(file.name)
file.write('This is the first tmp file!'.encode('utf-8'))
file.close()

print (name)

上述代码执行的任务为：使用tempfile.NamedTemporaryFile创建一个临时文件，其文件名采用的是随机化的字符串格式，作为name这样的一个属性来调用。通过执行这个任务，我们可以查看一般是生成什么样格式的临时文件：

[dechin@dechin-manjaro tmp_file]$ python3 tempfile_test.py 
/tmp/tmppetcksa8
[dechin@dechin-manjaro tmp_file]$ ll
总用量 4
-rw-r--r-- 1 dechin dechin 181  1月 27 21:39 tempfile_test.py
[dechin@dechin-manjaro tmp_file]$ cat /tmp/tmppetcksa8
cat: /tmp/tmppetcksa8: 没有那个文件或目录

在这个python代码的执行过程中，产生了tmppetcksa8这样的一个文件，我们可以向这个文件中直接write一些字符串。这个临时文件被存储在tmp目录下，与当前的执行路径无关。同时执行结束之后我们发现，产生的这个临时文件被删除了，这是NamedTemporaryFile自带的一个delete的属性，默认配置是关闭临时文件后直接删除。

持久化保存临时文件

需要持久化保存临时文件是非常容易的，只需要将上述章节中的delete属性设置为False即可：

# tempfile_test.py

import tempfile

file = tempfile.NamedTemporaryFile(delete=False)
name = str(file.name)
file.write('This is the first tmp file!'.encode('utf-8'))
file.close()

print (name)

这里我们唯一的变动，只是在括号中加上了delete=True这一设定，这个设定可以允许我们持久化的存储临时文件：

[dechin@dechin-manjaro tmp_file]$ python3 tempfile_test.py 
/tmp/tmpwlt27ryk
[dechin@dechin-manjaro tmp_file]$ cat /tmp/tmpwlt27ryk
This is the first tmp file!

设置临时文件后缀

在有些场景下对于临时文件的存储有一定的格式要求，比如后缀等，这里我们将临时文件的后缀设置为常用的txt格式，同样的，只需要在NamedTemporaryFile的参数中进行配置即可：

# tempfile_test.py

import tempfile

file = tempfile.NamedTemporaryFile(delete=False, suffix='.txt')
name = str(file.name)
file.write('This is the first tmp file!'.encode('utf-8'))
file.close()

print (name)

由于还是设置了delete=True参数，因此该临时txt文件被持久化的保存在系统中的/tmp目录下：

[dechin@dechin-manjaro tmp_file]$ python3 tempfile_test.py 
/tmp/tmpk0ct_kzs.txt
[dechin@dechin-manjaro tmp_file]$ cat /tmp/tmpk0ct_kzs.txt
This is the first tmp file!