如何用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這樣的一個屬性來調用。通過執行這個任務,我們可以查看一般是生成什麼樣格式的臨時文件:

[[email protected] tmp_file]$ python3 tempfile_test.py 
/tmp/tmppetcksa8
[[email protected] tmp_file]$ ll
總用量 4
-rw-r--r-- 1 dechin dechin 181 1月 27 21:39 tempfile_test.py
[[email protected] 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這一設定,這個設定可以允許我們持久化的存儲臨時文件:

[[email protected] tmp_file]$ python3 tempfile_test.py 
/tmp/tmpwlt27ryk
[[email protected] 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目錄下:

[[email protected] tmp_file]$ python3 tempfile_test.py 
/tmp/tmpk0ct_kzs.txt
[[email protected] 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

以上就是如何用tempfile庫創建python進程中的臨時文件的詳細內容,更多關於tempfile庫創建臨時文件的資料請關註WalkonNet其它相關文章!