善用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
目錄,而我們還可以根據不同的場景需要對產生的臨時文件的後綴進行配置。
版權聲明
本文首發鏈接為://www.cnblogs.com/dechinphy/p/tempfile.html
作者ID:DechinPhy
更多原著文章請參考://www.cnblogs.com/dechinphy/