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)

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