乐趣区

关于python:如何正确使用Python临时文件

jiezi

摘要： 临时文件通常用来保留无奈保留在内存中的数据，或者传递给必须从文件读取的内部程序。个别咱们会在 /tmp 目录下生成惟一的文件名，然而平安的创立临时文件并不是那么简略，须要恪守许多规定。

本文分享自华为云社区《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()  # 文件敞开即删除

点击关注，第一工夫理解华为云陈腐技术~

退出移动版