Python tempfile模塊：生成臨時文件和臨時目錄

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

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

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

tempfile 模塊函數(shù)

功能描述

tempfile.TemporaryFile(mode='w+b', buffering=None, encoding=None, newline=None, suffix=None, prefix=None, dir=None)

創(chuàng)建臨時文件。該函數(shù)返回一個類文件對象，也就是支持文件 I/O。

tempfile.NamedTemporaryFile(mode='w+b', buffering=None, encoding=None, newline=None, suffix=None, prefix=None, dir=None, delete=True)

創(chuàng)建臨時文件。該函數(shù)的功能與上一個函數(shù)的功能大致相同，只是它生成的臨時文件在文件系統(tǒng)中有文件名。

tempfile.SpooledTemporaryFile(max_size=0, mode='w+b', buffering=None, encoding=None, newline=None, suffix=None, prefix=None, dir=None)

創(chuàng)建臨時文件。與 TemporaryFile 函數(shù)相比，當(dāng)程序向該臨時文件輸出數(shù)據(jù)時，會先輸出到內(nèi)存中，直到超過 max_size 才會真正輸出到物理磁盤中。

tempfile.TemporaryDirectory(suffix=None, prefix=None, dir=None)

生成臨時目錄。

tempfile.gettempdir()

獲取系統(tǒng)的臨時目錄。

tempfile.gettempdirb()

與 gettempdir() 相同，只是該函數(shù)返回字節(jié)串。

tempfile.gettempprefix()

返回用于生成臨時文件的前綴名。

tempfile.gettempprefixb()

與 gettempprefix() 相同，只是該函數(shù)返回字節(jié)串。

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

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

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

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

import tempfile

# 創(chuàng)建臨時文件

fp = tempfile.TemporaryFile()

print(fp.name)

fp.write('兩情若是久長時，'.encode('utf-8'))

fp.write('又豈在朝朝暮暮。'.encode('utf-8'))

# 將文件指針移到開始處，準(zhǔn)備讀取文件

fp.seek(0)

print(fp.read().decode('utf-8')) # 輸出剛才寫入的內(nèi)容

# 關(guān)閉文件，該文件將會被自動刪除

fp.close()

# 通過with語句創(chuàng)建臨時文件，with會自動關(guān)閉臨時文件

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)建臨時目錄

with tempfile.TemporaryDirectory() as tmpdirname:

print('創(chuàng)建臨時目錄', tmpdirname)

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

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

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

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

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

C:\Users\admin\AppData\Local\Temp\tmphvehw9z1

兩情若是久長時，又豈在朝朝暮暮。

b'I Love Python!'

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

上面第一行輸出結(jié)果就是程序生成的臨時文件的文件名，最后一行輸出結(jié)果就是程序生成的臨時目錄的目錄名。需要注意的是，不要去找臨時文件或臨時文件夾，因為程序退出時該臨時文件和臨時文件夾都會被刪除。

Python tempfile模塊：生成臨時文件和臨時目錄相關(guān)教程

總結(jié)

以上是生活随笔為你收集整理的python tempfile cleanup_Python tempfile模块：生成临时文件和临时目录的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。