如何用tempfile庫創(chuàng)建python進程中的臨時文件

更新時間：2021年01月28日 11:11:32 作者：DechinPhy

這篇文章主要介紹了如何用tempfile庫創(chuàng)建python進程中的臨時文件，幫助大家更好的理解和使用python，感興趣的朋友可以了解下

技術背景

臨時文件在python項目中時常會被使用到，其作用在于隨機化的創(chuàng)建不重名的文件，路徑一般都是放在Linux系統(tǒng)下的/tmp目錄。如果項目中并不需要持久化的存儲一個文件，就可以采用臨時文件的形式進行存儲和讀取，在使用之后可以自行決定是刪除還是保留。

tempfile庫的使用

tempfile一般是python內置的一個函數庫，不需要單獨安裝，這里我們直接介紹一下其常規(guī)使用方法：

# 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)

上述代碼執(zhí)行的任務為：使用tempfile.NamedTemporaryFile創(chuàng)建一個臨時文件，其文件名采用的是隨機化的字符串格式，作為name這樣的一個屬性來調用。通過執(zhí)行這個任務，我們可以查看一般是生成什么樣格式的臨時文件：

[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代碼的執(zhí)行過程中，產生了tmppetcksa8這樣的一個文件，我們可以向這個文件中直接write一些字符串。這個臨時文件被存儲在tmp目錄下，與當前的執(zhí)行路徑無關。同時執(zhí)行結束之后我們發(fā)現，產生的這個臨時文件被刪除了，這是NamedTemporaryFile自帶的一個delete的屬性，默認配置是關閉臨時文件后直接刪除。

持久化保存臨時文件

需要持久化保存臨時文件是非常容易的，只需要將上述章節(jié)中的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文件被持久化的保存在系統(tǒng)中的/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!