python 創建臨時文件和文件夾

----需要在程序執行時創建一個臨時文件或目錄，並希望使用完之后可以自動銷毀掉。

tempfile 模塊中有很多的函數可以完成這任務。為了創建一個匿名的臨時文件，可以使用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() 的第一個參數是文件模式，通常來講文本模式使用w+t ，二進制模式使用w+b 。這個模式同時支持讀和寫操作，在這里是很有用的，因為當你關閉文件去改變模式的時候，文件實際上已經不存在了。TemporaryFile() 另外還支持跟內置的open() 函數一樣的參數。比如：

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

 

 

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

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

 

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

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

 

----為了創建一個臨時目錄，可以使用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() 函數應該是處理臨時文件目錄的最簡單的方式了，因為它們會自動處理所有的創建和清理步驟。在一個更低的級別，你可以使用mkstemp() 和mkdtemp() 來創建臨時文件和目錄。比如：

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

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

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

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

 

----所有和臨時文件相關的函數都允許你通過使用關鍵字參數prefix 、suffix 和dir來自定義目錄以及命名規則。

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

 最后還有一點，盡可能以最安全的方式使用tempfile 模塊來創建臨時文件。包括僅給當前用戶授權訪問以及在文件創建過程中采取措施避免競態條件。要注意的是不同的平台可能會不一樣。因此你最好閱讀官方文檔來了解更多的細節。