快捷導(dǎo)航

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

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

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

技術(shù)背景

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

tempfile庫的使用

tempfile一般是python內(nèi)置的一個函數(shù)庫，不需要單獨安裝，這里我們直接介紹一下其常規(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í)行的任務(wù)為：使用tempfile.NamedTemporaryFile創(chuàng)建一個臨時文件，其文件名采用的是隨機化的字符串格式，作為name這樣的一個屬性來調(diào)用。通過執(zhí)行這個任務(wù)，我們可以查看一般是生成什么樣格式的臨時文件：

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

持久化保存臨時文件

需要持久化保存臨時文件是非常容易的，只需要將上述章節(jié)中的delete屬性設(shè)置為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這一設(shè)定，這個設(shè)定可以允許我們持久化的存儲臨時文件：

[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!

設(shè)置臨時文件后綴

在有些場景下對于臨時文件的存儲有一定的格式要求，比如后綴等，這里我們將臨時文件的后綴設(shè)置為常用的txt格式，同樣的，只需要在NamedTemporaryFile的參數(shù)中進行配置即可：

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

由于還是設(shè)置了delete=True參數(shù)，因此該臨時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!