如何使用Python實現(xiàn)創(chuàng)建一個臨時文件或文件夾

這篇文章運用簡單易懂的例子給大家介紹如何使用Python實現(xiàn)創(chuàng)建一個臨時文件或文件夾，內(nèi)容非常詳細，感興趣的小伙伴們可以參考借鑒，希望對大家能有所幫助。

問題

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

解決方案

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

from tempfile import TemporaryFile

with TemporaryFile('w+t') as f:
  # Read/write to the file
  f.write('Hello World\n')
  f.write('Testing\n')

  # Seek back to beginning and read the data
  f.seek(0)
  data = f.read()

# Temporary file is destroyed

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

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

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

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

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

from tempfile import NamedTemporaryFile

with NamedTemporaryFile('w+t') as f:
  print('filename is:', f.name)
  ...

# File automatically destroyed

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

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

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

from tempfile import TemporaryDirectory

with TemporaryDirectory() as dirname:
  print('dirname is:', dirname)
  # Use the directory
  ...
# Directory and all contents destroyed

討論

TemporaryFile() 、NamedTemporaryFile() 和 TemporaryDirectory() 函數(shù) 應(yīng)該是處理臨時文件目錄的最簡單的方式了，因為它們會自動處理所有的創(chuàng)建和清理步驟。 在一個更低的級別，你可以使用 mkstemp() 和 mkdtemp() 來創(chuàng)建臨時文件和目錄。比如：

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

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

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

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

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

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

最后還有一點，盡可能以最安全的方式使用 tempfile 模塊來創(chuàng)建臨時文件。 包括僅給當前用戶授權(quán)訪問以及在文件創(chuàng)建過程中采取措施避免競態(tài)條件。 要注意的是不同的平臺可能會不一樣。

關(guān)于如何使用Python實現(xiàn)創(chuàng)建一個臨時文件或文件夾就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。