一般来说,软件运行过程中一般会在指定位置生成临时文件,这些资源不要轻易删除,可能是过程文件。下面介绍一个非常有用的临时文件生成模块,巧妙运用,让你的代码轻松无痕运行。
一、简介
这里介绍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)
当然根据实际情况,临时资源可以直接利于内存或数据库存储。
本文分享自 Python数据分析实例 微信公众号,前往查看
如有侵权,请联系 cloudcommunity@tencent.com 删除。
本文参与 腾讯云自媒体同步曝光计划 ,欢迎热爱写作的你一起参与!