善用tempfile庫建立python程序中的臨時檔案
阿新 • • 發佈：2021-01-28
# 技術背景
臨時檔案在python專案中時常會被使用到，其作用在於隨機化的建立不重名的檔案，路徑一般都是放在Linux系統下的`/tmp`目錄。如果專案中並不需要持久化的儲存一個檔案，就可以採用臨時檔案的形式進行儲存和讀取，在使用之後可以自行決定是刪除還是保留。
# tempfile庫的使用
`tempfile`一般是python內建的一個函式庫，不需要單獨安裝，這裡我們直接介紹一下其常規使用方法：
```python
# tempfile_test.py
import tempfile
file = tempfile.NamedTemporaryFile()
name = str(file.name)
file.write('This is the first tmp file!'.encode('utf-8'))
file.close()
print (name)
```
上述程式碼執行的任務為：使用`tempfile.NamedTemporaryFile`建立一個臨時檔案，其檔名採用的是隨機化的字串格式，作為`name`這樣的一個屬性來呼叫。通過執行這個任務，我們可以檢視一般是生成什麼樣格式的臨時檔案：
```bash
[dechin@dechin-manjaro tmp_file]$ python3 tempfile_test.py
/tmp/tmppetcksa8
[dechin@dechin-manjaro tmp_file]$ ll
總用量 4
-rw-r--r-- 1 dechin dechin 181 1月 27 21:39 tempfile_test.py
[dechin@dechin-manjaro tmp_file]$ cat /tmp/tmppetcksa8
cat: /tmp/tmppetcksa8: 沒有那個檔案或目錄
```
在這個python程式碼的執行過程中，產生了`tmppetcksa8`這樣的一個檔案，我們可以向這個檔案中直接`write`一些字串。這個臨時檔案被儲存在`tmp`目錄下，與當前的執行路徑無關。同時執行結束之後我們發現，產生的這個臨時檔案被刪除了，這是`NamedTemporaryFile`自帶的一個`delete`的屬性，預設配置是關閉臨時檔案後直接刪除。
# 持久化儲存臨時檔案
需要持久化儲存臨時檔案是非常容易的，只需要將上述章節中的`delete`屬性設定為`False`即可：
```python
# tempfile_test.py
import tempfile
file = tempfile.NamedTemporaryFile(delete=False)
name = str(file.name)
file.write('This is the first tmp file!'.encode('utf-8'))
file.close()
print (name)
```
這裡我們唯一的變動，只是在括號中加上了`delete=True`這一設定，這個設定可以允許我們持久化的儲存臨時檔案：
```bash
[dechin@dechin-manjaro tmp_file]$ python3 tempfile_test.py
/tmp/tmpwlt27ryk
[dechin@dechin-manjaro tmp_file]$ cat /tmp/tmpwlt27ryk
This is the first tmp file!
```
# 設定臨時檔案字尾
在有些場景下對於臨時檔案的儲存有一定的格式要求，比如字尾等，這裡我們將臨時檔案的字尾設定為常用的`txt`格式，同樣的，只需要在`NamedTemporaryFile`的引數中進行配置即可：
```python
# tempfile_test.py
import tempfile
file = tempfile.NamedTemporaryFile(delete=False, suffix='.txt')
name = str(file.name)
file.write('This is the first tmp file!'.encode('utf-8'))
file.close()
print (name)
```
由於還是設定了`delete=True`引數，因此該臨時`txt`檔案被持久化的儲存在系統中的`/tmp`目錄下：
```bash
[dechin@dechin-manjaro tmp_file]$ python3 tempfile_test.py
/tmp/tmpk0ct_kzs.txt
[dechin@dechin-manjaro tmp_file]$ cat /tmp/tmpk0ct_kzs.txt
This is the first tmp file!
```
# 總結概要
本文主要介紹了python中自帶的tempfile庫對臨時檔案的操作，通過tempfile庫我們可以建立自動刪除的或者持久化儲存的臨時檔案，儲存路徑為Linux系統下的`/tmp`目錄，而我們還可以根據不同的場景需要對產生的臨時檔案的字尾進行配置。
# 版權宣告
本文首發連結為：https://www.cnblogs.com/dechinphy/p/tempfile.html
作者ID：DechinPhy
更多原著文章請參考：https://www.cnblogs.com/de