快捷導(dǎo)航

正確的使用Python臨時(shí)文件

更新時(shí)間：2021年03月26日 15:42:01 作者：技術(shù)火炬手

這篇文章主要介紹了正確的使用Python臨時(shí)文件，幫助大家更好的理解和學(xué)習(xí)使用python，感興趣的朋友可以了解下

1、前言

臨時(shí)文件通常用來保存無法保存在內(nèi)存中的數(shù)據(jù)，或者傳遞給必須從文件讀取的外部程序。一般我們會(huì)在/tmp目錄下生成唯一的文件名，但是安全的創(chuàng)建臨時(shí)文件并不是那么簡單，需要遵守許多規(guī)則。永遠(yuǎn)不要自己去嘗試做這件事，而是要借助庫函數(shù)實(shí)現(xiàn)。而且也要小心清理臨時(shí)文件。

臨時(shí)文件引起的最大問題就是，可以預(yù)測文件名，導(dǎo)致惡意用戶可以預(yù)測臨時(shí)文件名，從而創(chuàng)建軟鏈接劫持臨時(shí)文件。

2. tempfile模塊介紹

創(chuàng)建臨時(shí)文件一般使用的模塊就是tempfile，此模塊庫函數(shù)常用的有以下幾個(gè)：

tempfile.mktemp # 不安全，禁止使用
tempfile.mkstemp # 隨機(jī)創(chuàng)建tmp文件,默認(rèn)創(chuàng)建的文件在/tmp目錄，當(dāng)然也可以指定（可以使用）
tempfile.TemporaryFile # 內(nèi)存中創(chuàng)建文件，文件不會(huì)存儲(chǔ)在磁盤，關(guān)閉后即刪除（可以使用）
tempfile.NamedTemporaryFile(delete=True) 當(dāng)delete=True時(shí)，作用跟上面一樣，當(dāng)是False時(shí)，會(huì)存儲(chǔ)在磁盤（可以使用）

3. 示例介紹

以下幾種方式分別介紹了安全的創(chuàng)建臨時(shí)文件及不安全的方式。

3.1 不正確示例：

不正確1：

import os
import tempfile
 
# This will most certainly put you at risk
tmp = os.path.join(tempfile.gettempdir(), filename)
if not os.path.exists(tmp):
 with open(tmp, "w") file:
  file.write("defaults")

不正確2：

import os
import tempfile
 
open(tempfile.mktemp(), "w")

不正確3：

filename = "{}/{}.tmp".format(tempfile.gettempdir(), os.getpid())
open(filename, "w")

3.2 正確示例

正確1：

fd, path = tempfile.mkstemp()
try:
 with os.fdopen(fd, 'w') as tmp:
  # do stuff with temp file
  tmp.write('stuff')
finally:
 os.remove(path)

正確2：

# 句柄關(guān)閉，文件即刪除
with tempfile.TemporaryFile() as tmp:
 # Do stuff with tmp
 tmp.write('stuff')

正確3：

tmp = tempfile.NamedTemporaryFile(delete=True)
try:
 # do stuff with temp
 tmp.write('stuff')
finally:
 tmp.close() # 文件關(guān)閉即刪除

以上就是正確的使用Python臨時(shí)文件的詳細(xì)內(nèi)容，更多關(guān)于使用Python臨時(shí)文件的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章:

相關(guān)文章

Python判斷變量是否是None寫法代碼實(shí)例
這篇文章主要介紹了Python判斷變量是否是None寫法代碼實(shí)例,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下
2020-10-10
python微信公眾號(hào)之關(guān)注公眾號(hào)自動(dòng)回復(fù)
這篇文章主要為大家詳細(xì)介紹了python微信公眾號(hào)之關(guān)注公眾號(hào)自動(dòng)回復(fù)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下
2018-10-10
python實(shí)現(xiàn)代理服務(wù)功能實(shí)例
本文分析了代理服務(wù)的原理，并使用PYTHON實(shí)現(xiàn)了一個(gè)簡單的代理服務(wù)功能供大家參考使用
2013-11-11
python通過http下載文件的方法詳解
這篇文章主要介紹了python通過http下載文件的方法詳解,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下
2019-07-07
從0開始的Python學(xué)習(xí)014面向?qū)ο缶幊蹋ㄍ扑]）
這篇文章主要介紹了Python面向?qū)ο缶幊?，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2019-04-04
win10下安裝Anaconda的教程（python環(huán)境+jupyter_notebook)
Anaconda指的是一個(gè)開源的Python發(fā)行版本，其包含了conda、Python等180多個(gè)科學(xué)包及其依賴項(xiàng)。這篇文章主要介紹了win10下安裝Anaconda（python環(huán)境+jupyter_notebook),需要的朋友可以參考下
2019-10-10
python開發(fā)之函數(shù)定義實(shí)例分析
這篇文章主要介紹了python開發(fā)之函數(shù)定義方法,以實(shí)例形式較為詳細(xì)的分析了Python中函數(shù)的定義與使用技巧,需要的朋友可以參考下
2015-11-11
Python函數(shù)進(jìn)階與文件操作詳情
這篇文章主要介紹了Python函數(shù)進(jìn)階與文件操作詳情，文章為榮啊主題展開詳細(xì)的內(nèi)容介紹，具有一定的參考價(jià)值，需要的小伙伴可以參考一下
2022-08-08
Linux CentOS Python開發(fā)環(huán)境搭建教程
這篇文章主要介紹了Linux CentOS Python開發(fā)環(huán)境搭建方法，非常不錯(cuò)，具有一定的參考借鑒價(jià)值,需要的朋友可以參考下
2018-11-11
Numpy中對向量、矩陣的使用詳解
這篇文章主要介紹了Numpy中對向量、矩陣的使用詳解，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2019-10-10