Python 使用tempfile包轻松无痕的运行代码

2024-04-02 19:55

短信预约 -IT技能 免费直播动态提醒

一、简介

这里介绍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)

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