Python tempfile模塊生成臨時(shí)文件和臨時(shí)目錄

tempfile 模塊專門用于創(chuàng)建臨時(shí)文件和臨時(shí)目錄，它既可以在 UNIX 平臺(tái)上運(yùn)行良好，也可以在 Windows 平臺(tái)上運(yùn)行良好。

tempfile 模塊中常用的函數(shù)，如表 1 所示。

表 1 tempfile 模塊常用函數(shù)及功能

提示：表中有些函數(shù)包含很多參數(shù)，但這些參數(shù)都具有自己的默認(rèn)值，因此如果沒有特殊要求，可以不對其傳參。

tempfile 模塊還提供了 tempfile.mkstemp() 和 tempfile.mkdtemp() 兩個(gè)低級(jí)別的函數(shù)。上面介紹的 4 個(gè)用于創(chuàng)建臨時(shí)文件和臨時(shí)目錄的函數(shù)都是高級(jí)別的函數(shù)，高級(jí)別的函數(shù)支持自動(dòng)清理，而且可以與 with 語句一起使用，而這兩個(gè)低級(jí)別的函數(shù)則不支持，因此一般推薦使用高級(jí)別的函數(shù)來創(chuàng)建臨時(shí)文件和臨時(shí)目錄。

此外，tempfile 模塊還提供了 tempfile.tempdir 屬性，通過對該屬性賦值可以改變系統(tǒng)的臨時(shí)目錄。

下面程序示范了如何使用臨時(shí)文件和臨時(shí)目錄：

import tempfile
# 創(chuàng)建臨時(shí)文件
fp = tempfile.TemporaryFile()
print(fp.name)
fp.write('兩情若是久長時(shí)，'.encode('utf-8'))
fp.write('又豈在朝朝暮暮。'.encode('utf-8'))
# 將文件指針移到開始處，準(zhǔn)備讀取文件
fp.seek(0)
print(fp.read().decode('utf-8')) # 輸出剛才寫入的內(nèi)容
# 關(guān)閉文件，該文件將會(huì)被自動(dòng)刪除
fp.close()
# 通過with語句創(chuàng)建臨時(shí)文件，with會(huì)自動(dòng)關(guān)閉臨時(shí)文件
with tempfile.TemporaryFile() as fp:
# 寫入內(nèi)容
fp.write(b'I Love Python!')
# 將文件指針移到開始處，準(zhǔn)備讀取文件
fp.seek(0)
# 讀取文件內(nèi)容
print(fp.read()) # b'I Love Python!'
# 通過with語句創(chuàng)建臨時(shí)目錄
with tempfile.TemporaryDirectory() as tmpdirname:
print('創(chuàng)建臨時(shí)目錄', tmpdirname)

上面程序以兩種方式來創(chuàng)建臨時(shí)文件：

第一種方式是手動(dòng)創(chuàng)建臨時(shí)文件，讀寫臨時(shí)文件后需要主動(dòng)關(guān)閉它，當(dāng)程序關(guān)閉該臨時(shí)文件時(shí)，該文件會(huì)被自動(dòng)刪除。

第二種方式則是使用 with 語句創(chuàng)建臨時(shí)文件，這樣 with 語句會(huì)自動(dòng)關(guān)閉臨時(shí)文件。

上面程序最后還創(chuàng)建了臨時(shí)目錄。由于程序使用 with 語句來管理臨時(shí)目錄，因此程序也會(huì)自動(dòng)刪除該臨時(shí)目錄。

運(yùn)行上面程序，可以看到如下輸出結(jié)果：

C:\Users\admin\AppData\Local\Temp\tmphvehw9z1
兩情若是久長時(shí)，又豈在朝朝暮暮。
b'I Love Python!'

創(chuàng)建臨時(shí)目錄C:\Users\admin\AppData\Local\Temp\tmp3sjbnwob

上面第一行輸出結(jié)果就是程序生成的臨時(shí)文件的文件名，最后一行輸出結(jié)果就是程序生成的臨時(shí)目錄的目錄名。需要注意的是，不要去找臨時(shí)文件或臨時(shí)文件夾，因?yàn)槌绦蛲顺鰰r(shí)該臨時(shí)文件和臨時(shí)文件夾都會(huì)被刪除。

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持億速云。