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

技术背景

临时文件在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!

总结概要

本文主要介绍了python中自带的tempfile库对临时文件的操作，通过tempfile库我们可以创建自动删除的或者持久化存储的临时文件，存储路径为Linux系统下的/tmp目录，而我们还可以根据不同的场景需要对产生的临时文件的后缀进行配置。

版权声明

本文首发链接为：https://www.cnblogs.com/dechinphy/p/tempfile.html

作者ID：DechinPhy

更多原著文章请参考：https://www.cnblogs.com/dechinphy/