臨時(shí)文件通常用來保存無法保存在內(nèi)存中的數(shù)據(jù),或者傳遞給必須從文件讀取的外部程序。一般我們會(huì)在/tmp目錄下生成唯一的文件名,但是安全的創(chuàng)建臨時(shí)文件并不是那么簡單,需要遵守許多規(guī)則。永遠(yuǎn)不要自己去嘗試做這件事,而是要借助庫函數(shù)實(shí)現(xiàn)。而且也要小心清理臨時(shí)文件。
臨時(shí)文件引起的最大問題就是,可以預(yù)測(cè)文件名,導(dǎo)致惡意用戶可以預(yù)測(cè)臨時(shí)文件名,從而創(chuàng)建軟鏈接劫持臨時(shí)文件。
創(chuàng)建臨時(shí)文件一般使用的模塊就是tempfile,此模塊庫函數(shù)常用的有以下幾個(gè):
tempfile.mktemp # 不安全,禁止使用
tempfile.mkstemp # 隨機(jī)創(chuàng)建tmp文件,默認(rèn)創(chuàng)建的文件在/tmp目錄,當(dāng)然也可以指定(可以使用)
tempfile.TemporaryFile # 內(nèi)存中創(chuàng)建文件,文件不會(huì)存儲(chǔ)在磁盤,關(guān)閉后即刪除(可以使用)
tempfile.NamedTemporaryFile(delete=True) 當(dāng)delete=True時(shí),作用跟上面一樣,當(dāng)是False時(shí),會(huì)存儲(chǔ)在磁盤(可以使用)
以下幾種方式分別介紹了安全的創(chuàng)建臨時(shí)文件及不安全的方式。
不正確1:
import os import tempfile # This will most certainly put you at risk tmp = os.path.join(tempfile.gettempdir(), filename) if not os.path.exists(tmp): with open(tmp, "w") file: file.write("defaults")
不正確2:
import os import tempfile open(tempfile.mktemp(), "w")
不正確3:
filename = "{}/{}.tmp".format(tempfile.gettempdir(), os.getpid()) open(filename, "w")
正確1:
fd, path = tempfile.mkstemp() try: with os.fdopen(fd, 'w') as tmp: # do stuff with temp file tmp.write('stuff') finally: os.remove(path)
正確2:
# 句柄關(guān)閉,文件即刪除 with tempfile.TemporaryFile() as tmp: # Do stuff with tmp tmp.write('stuff')
正確3:
tmp = tempfile.NamedTemporaryFile(delete=True) try: # do stuff with temp tmp.write('stuff') finally: tmp.close() # 文件關(guān)閉即刪除
以上就是正確的使用Python臨時(shí)文件的詳細(xì)內(nèi)容,更多關(guān)于使用Python臨時(shí)文件的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章!
標(biāo)簽:西安 錦州 天水 安慶 隨州 股票 日照 白城
巨人網(wǎng)絡(luò)通訊聲明:本文標(biāo)題《正確的使用Python臨時(shí)文件》,本文關(guān)鍵詞 正確,的,使用,Python,臨時(shí),;如發(fā)現(xiàn)本文內(nèi)容存在版權(quán)問題,煩請(qǐng)?zhí)峁┫嚓P(guān)信息告之我們,我們將及時(shí)溝通與處理。本站內(nèi)容系統(tǒng)采集于網(wǎng)絡(luò),涉及言論、版權(quán)與本站無關(guān)。