正确的使用Python临时文件

更新时间：2021年03月26日 15:42:01 作者：技术火炬手

这篇文章主要介绍了正确的使用Python临时文件，帮助大家更好的理解和学习使用python，感兴趣的朋友可以了解下

1、前言

临时文件通常用来保存无法保存在内存中的数据，或者传递给必须从文件读取的外部程序。一般我们会在/tmp目录下生成唯一的文件名，但是安全的创建临时文件并不是那么简单，需要遵守许多规则。永远不要自己去尝试做这件事，而是要借助库函数实现。而且也要小心清理临时文件。

临时文件引起的最大问题就是，可以预测文件名，导致恶意用户可以预测临时文件名，从而创建软链接劫持临时文件。

2. tempfile模块介绍

创建临时文件一般使用的模块就是tempfile，此模块库函数常用的有以下几个：

tempfile.mktemp # 不安全，禁止使用
tempfile.mkstemp # 随机创建tmp文件,默认创建的文件在/tmp目录，当然也可以指定（可以使用）
tempfile.TemporaryFile # 内存中创建文件，文件不会存储在磁盘，关闭后即删除（可以使用）
tempfile.NamedTemporaryFile(delete=True) 当delete=True时，作用跟上面一样，当是False时，会存储在磁盘（可以使用）

3. 示例介绍

以下几种方式分别介绍了安全的创建临时文件及不安全的方式。

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：

# 句柄关闭，文件即删除
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() # 文件关闭即删除

以上就是正确的使用Python临时文件的详细内容，更多关于使用Python临时文件的资料请关注脚本之家其它相关文章！

您可能感兴趣的文章:

Python判断变量是否是None写法代码实例
这篇文章主要介绍了Python判断变量是否是None写法代码实例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2020-10-10
python微信公众号之关注公众号自动回复
这篇文章主要为大家详细介绍了python微信公众号之关注公众号自动回复，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2018-10-10
python实现代理服务功能实例
本文分析了代理服务的原理，并使用PYTHON实现了一个简单的代理服务功能供大家参考使用
2013-11-11
python通过http下载文件的方法详解
这篇文章主要介绍了python通过http下载文件的方法详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-07-07
从0开始的Python学习014面向对象编程（推荐）
这篇文章主要介绍了Python面向对象编程，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2019-04-04
win10下安装Anaconda的教程（python环境+jupyter_notebook)
Anaconda指的是一个开源的Python发行版本，其包含了conda、Python等180多个科学包及其依赖项。这篇文章主要介绍了win10下安装Anaconda（python环境+jupyter_notebook),需要的朋友可以参考下
2019-10-10
python开发之函数定义实例分析
这篇文章主要介绍了python开发之函数定义方法,以实例形式较为详细的分析了Python中函数的定义与使用技巧,需要的朋友可以参考下
2015-11-11
Python函数进阶与文件操作详情
这篇文章主要介绍了Python函数进阶与文件操作详情，文章为荣啊主题展开详细的内容介绍，具有一定的参考价值，需要的小伙伴可以参考一下
2022-08-08
Linux CentOS Python开发环境搭建教程
这篇文章主要介绍了Linux CentOS Python开发环境搭建方法，非常不错，具有一定的参考借鉴价值,需要的朋友可以参考下
2018-11-11
Numpy中对向量、矩阵的使用详解
这篇文章主要介绍了Numpy中对向量、矩阵的使用详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2019-10-10