問(wèn)題

你需要在程序執(zhí)行時(shí)創(chuàng)建一個(gè)臨時(shí)文件或目錄，并希望使用完之后可以自動(dòng)銷毀掉。

解決方案

tempfile 模塊中有很多的函數(shù)可以完成這任務(wù)。 為了創(chuàng)建一個(gè)匿名的臨時(shí)文件，可以使用 tempfile.TemporaryFile ：

from tempfile import TemporaryFilewith TemporaryFile(’w+t’) as f: # Read/write to the file f.write(’Hello Worldn’) f.write(’Testingn’) # Seek back to beginning and read the data f.seek(0) data = f.read()# Temporary file is destroyed

或者，如果你喜歡，你還可以像這樣使用臨時(shí)文件：

f = TemporaryFile(’w+t’)# Use the temporary file...f.close()# File is destroyed

TemporaryFile() 的第一個(gè)參數(shù)是文件模式，通常來(lái)講文本模式使用 w+t ，二進(jìn)制模式使用 w+b 。 這個(gè)模式同時(shí)支持讀和寫操作，在這里是很有用的，因?yàn)楫?dāng)你關(guān)閉文件去改變模式的時(shí)候，文件實(shí)際上已經(jīng)不存在了。 TemporaryFile() 另外還支持跟內(nèi)置的 open() 函數(shù)一樣的參數(shù)。比如：

with TemporaryFile(’w+t’, encoding=’utf-8’, errors=’ignore’) as f: ...

在大多數(shù)Unix系統(tǒng)上，通過(guò) TemporaryFile() 創(chuàng)建的文件都是匿名的，甚至連目錄都沒(méi)有。 如果你想打破這個(gè)限制，可以使用 NamedTemporaryFile() 來(lái)代替。比如：

from tempfile import NamedTemporaryFilewith NamedTemporaryFile(’w+t’) as f: print(’filename is:’, f.name) ...# File automatically destroyed

這里，被打開(kāi)文件的 f.name 屬性包含了該臨時(shí)文件的文件名。 當(dāng)你需要將文件名傳遞給其他代碼來(lái)打開(kāi)這個(gè)文件的時(shí)候，這個(gè)就很有用了。 和 TemporaryFile() 一樣，結(jié)果文件關(guān)閉時(shí)會(huì)被自動(dòng)刪除掉。 如果你不想這么做，可以傳遞一個(gè)關(guān)鍵字參數(shù) delete=False 即可。比如：

with NamedTemporaryFile(’w+t’, delete=False) as f: print(’filename is:’, f.name) ...

為了創(chuàng)建一個(gè)臨時(shí)目錄，可以使用 tempfile.TemporaryDirectory() 。比如：

from tempfile import TemporaryDirectorywith TemporaryDirectory() as dirname: print(’dirname is:’, dirname) # Use the directory ...# Directory and all contents destroyed

討論

TemporaryFile() 、NamedTemporaryFile() 和 TemporaryDirectory() 函數(shù) 應(yīng)該是處理臨時(shí)文件目錄的最簡(jiǎn)單的方式了，因?yàn)樗鼈儠?huì)自動(dòng)處理所有的創(chuàng)建和清理步驟。 在一個(gè)更低的級(jí)別，你可以使用 mkstemp() 和 mkdtemp() 來(lái)創(chuàng)建臨時(shí)文件和目錄。比如：

>>> import tempfile>>> tempfile.mkstemp()(3, ’/var/folders/7W/7WZl5sfZEF0pljrEB1UMWE+++TI/-Tmp-/tmp7fefhv’)>>> tempfile.mkdtemp()’/var/folders/7W/7WZl5sfZEF0pljrEB1UMWE+++TI/-Tmp-/tmp5wvcv6’>>>

但是，這些函數(shù)并不會(huì)做進(jìn)一步的管理了。 例如，函數(shù) mkstemp() 僅僅就返回一個(gè)原始的OS文件描述符，你需要自己將它轉(zhuǎn)換為一個(gè)真正的文件對(duì)象。 同樣你還需要自己清理這些文件。

通常來(lái)講，臨時(shí)文件在系統(tǒng)默認(rèn)的位置被創(chuàng)建，比如 /var/tmp 或類似的地方。 為了獲取真實(shí)的位置，可以使用 tempfile.gettempdir() 函數(shù)。比如：

>>> tempfile.gettempdir()’/var/folders/7W/7WZl5sfZEF0pljrEB1UMWE+++TI/-Tmp-’>>>

所有和臨時(shí)文件相關(guān)的函數(shù)都允許你通過(guò)使用關(guān)鍵字參數(shù) prefix 、suffix 和 dir 來(lái)自定義目錄以及命名規(guī)則。比如：

>>> f = NamedTemporaryFile(prefix=’mytemp’, suffix=’.txt’, dir=’/tmp’)>>> f.name’/tmp/mytemp8ee899.txt’>>>

最后還有一點(diǎn)，盡可能以最安全的方式使用 tempfile 模塊來(lái)創(chuàng)建臨時(shí)文件。 包括僅給當(dāng)前用戶授權(quán)訪問(wèn)以及在文件創(chuàng)建過(guò)程中采取措施避免競(jìng)態(tài)條件。 要注意的是不同的平臺(tái)可能會(huì)不一樣。因此你最好閱讀 官方文檔 來(lái)了解更多的細(xì)節(jié)。

以上就是Python創(chuàng)建臨時(shí)文件和文件夾的詳細(xì)內(nèi)容，更多關(guān)于Python臨時(shí)文件和文件夾的資料請(qǐng)關(guān)注好吧啦網(wǎng)其它相關(guān)文章！