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

tempfile 模塊專門用於創建臨時文件和臨時目錄，它既可以在 UNIX 平台上運行良好，也可以在 Windows 平台上運行良好。

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

表 1 tempfile 模塊常用函數及功能

tempfile 模塊函數	功能描述
tempfile.TemporaryFile(mode='w+b', buffering=None, encoding=None, newline=None, suffix=None, prefix=None, dir=None)	創建臨時文件。該函數返回一個類文件對象，也就是支持文件 I/O。
tempfile.NamedTemporaryFile(mode='w+b', buffering=None, encoding=None, newline=None, suffix=None, prefix=None, dir=None, delete=True)	創建臨時文件。該函數的功能與上一個函數的功能大致相同，只是它生成的臨時文件在文件系統中有文件名。
tempfile.SpooledTemporaryFile(max_size=0, mode='w+b', buffering=None, encoding=None, newline=None, suffix=None, prefix=None, dir=None)	創建臨時文件。與 TemporaryFile 函數相比，當程序向該臨時文件輸出數據時，會先輸出到內存中，直到超過 max_size 才會真正輸出到物理磁盤中。
tempfile.TemporaryDirectory(suffix=None, prefix=None, dir=None)	生成臨時目錄。
tempfile.gettempdir()	獲取系統的臨時目錄。
tempfile.gettempdirb()	與 gettempdir() 相同，只是該函數返回字節串。
tempfile.gettempprefix()	返回用於生成臨時文件的前綴名。
tempfile.gettempprefixb()	與 gettempprefix() 相同，只是該函數返回字節串。

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

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

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

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

1. import tempfile
3. # 創建臨時文件
4. fp = tempfile.TemporaryFile()
5. print(fp.name)
6. fp.write('兩情若是久長時，'.encode('utf-8'))
7. fp.write('又豈在朝朝暮暮。'.encode('utf-8'))
8. # 將文件指針移到開始處，准備讀取文件
9. fp.seek(0)
10. print(fp.read().decode('utf-8')) # 輸出剛才寫入的內容
11. # 關閉文件，該文件將會被自動刪除
12. fp.close()
14. # 通過with語句創建臨時文件，with會自動關閉臨時文件
15. with tempfile.TemporaryFile() as fp:
16. # 寫入內容
17. fp.write(b'I Love Python!')
18. # 將文件指針移到開始處，准備讀取文件
19. fp.seek(0)
20. # 讀取文件內容
21. print(fp.read()) # b'I Love Python!'
23. # 通過with語句創建臨時目錄
24. with tempfile.TemporaryDirectory() as tmpdirname:
25. print('創建臨時目錄', tmpdirname)

上面程序以兩種方式來創建臨時文件：

1. 第一種方式是手動創建臨時文件，讀寫臨時文件后需要主動關閉它，當程序關閉該臨時文件時，該文件會被自動刪除。
2. 第二種方式則是使用 with 語句創建臨時文件，這樣 with 語句會自動關閉臨時文件。

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

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

C:\Users\admin\AppData\Local\Temp\tmphvehw9z1
兩情若是久長時，又豈在朝朝暮暮。
b'I Love Python!'
創建臨時目錄C:\Users\admin\AppData\Local\Temp\tmp3sjbnwob

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