tempfile：让你 Python 代码轻松无痕的运行

Posted 2021-12-13 Python学习与数据挖掘

大家好，我们知道软件运行过程中一般会在指定位置生成临时文件，这些资源不要轻易删除，可能是过程文件，定时清理是必要的，今天给大家分享一款工具：tempfile，喜欢本文点赞支持，欢迎收藏学习，文末提供技术交流群。

一、简介

这里介绍python中临时文件及文件夹使用。

使用的是tempfile包：

pip install tempfile

https://docs.python.org/3/library/tempfile.html

二、临时文件夹

2.1 获取临时文件夹

# 获取临时文件夹
tmpdir = tempfile.gettempdir()
print(tmpdir) #/tmp

2.2 生成临时文件夹

# 方式一：生成默认临时文件夹
tmpdir = tempfile.mkdtemp()
print(tmpdir) #/tmp/tmpui77cgud

# 方式二：生成自定义临时文件夹(指定前缀、后缀、目录，可指定其中一部分),suffix：后缀, prefix：前缀, dir：目录

tmpdir = tempfile.mkdtemp(suffix='_txt', prefix='tp_dir_', dir='/home/tmp/py_rs_file')

print(tmpdir) # /home/tmp/py_rs_file/tp_dir_06l_o2dm_txt

三、临时文件

3.1 生成不自动删除（关闭时）的临时文件

# 方式一：生成默认临时文件,默认为二进制文件

tmpfile = tempfile.mkstemp()[1]
print(tempfile) #/tmp/tmp75kazf_8
# 数据写入
with open(tmpfile, 'w+') as t_f:
    t_f.writelines('hello world')

# 方式二：生成自定义临时文件(指定前缀、后缀、目录、文件类型参数，可指定其中一部分),suffix：后缀, prefix：前缀, dir：目录, text：文件类型，True为文本，false为二进制

tmpfile = tempfile.mkstemp(suffix='.txt', prefix='tp_', dir='/home/tmp/py_rs_file', text=True)[1]
print(tempfile) # /home/tmp/py_rs_file/tp_pn2973g0.txt

# 数据写入
with open(tmpfile, 'w+') as t_f:
    t_f.writelines('hello world')

3.2 生成自动删除的临时文件

# 方式一：创建临时文件，文件关闭时自动删除
tmpfile = tempfile.TemporaryFile(mode='w+t')
tmpfile.write('hello world') ##数据写入
tmpfile.seek(0)
tmpTxt = tmpfile.read() #数据读取
print(tmpTxt)
tmpfile.close() #关闭时文件自动删除

# 方式二：创建临时文件，文件关闭时根据delete参数确定是否自动删除， True:删除  False:不删除
with tempfile.NamedTemporaryFile(delete=False) as tmpfile:
    file_name = tmpfile.name
    print(file_name) #/tmp/tmp73zl8gmn
    tmpfile.write('hello world'.encode())
    tmpfile.seek(0)
    tmpTxt = tmpfile.read().decode()
    print(tmpTxt)

# 方式三：创建自定义临时文件，文件关闭时可根据delete参数确定是否自动删除， True:删除  False:不删除
# 其他配置参数有，mode:文件模式(w+b为二进制模式(默认)，w+t为文本模式)，suffix：后缀, prefix：前缀, dir：目录
with tempfile.NamedTemporaryFile(mode='w+t', suffix='.txt', prefix='tp_', dir='/home/tmp/py_rs_file',delete=False) as tmpfile:
    file_name = tmpfile.name
    print(file_name) #/home/tmp/py_rs_file/tp_fcwpmh3l.txt
    tmpfile.write('hello world')
    tmpfile.seek(0)
    tmpTxt = tmpfile.read()
    print(tmpTxt)

根据具体情况，临时资源可以直接调用内存或数据库存储。

---

技术交流

欢迎转载、收藏、有所收获点赞支持一下！

目前开通了技术交流群，群友已超过2000人，添加时最好的备注方式为：来源+兴趣方向，方便找到志同道合的朋友

• 方式①、发送如下图片至微信，长按识别，后台回复：加群；
• 方式②、添加微信号：dkl88191，备注：来自CSDN
• 方式③、微信搜索公众号：Python学习与数据挖掘，后台回复：加群