Python中临时文件的作用什么,创建需要注意哪些?
导读:这篇文章给大家分享的是python中临时文件的相关内容,介绍了临时文件的作用,以及创建临时文件需要注意的问题和正确使用方法,对大家学习和理解Python临时文件会有用的帮助,感兴趣的朋友接下来一起跟随小编看看吧。1、前言临时文件通常用来保存...
这篇文章给大家分享的是python中临时文件的相关内容,介绍了临时文件的作用,以及创建临时文件需要注意的问题和正确使用方法,对大家学习和理解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() # 文件关闭即删除
现在大家对于Python中临时文件的作用和安全创建的方法应该都清楚了吧,上述的错误示例和正确示例都有一定的参考价值,需要的朋友可以了解看看,希望大家阅读完这篇文章能有所收获。最后,想要了解更多Python的内容,大家可以关注网络其它相关文章。
文本转载自PHP中文网
声明:本文内容由网友自发贡献,本站不承担相应法律责任。对本内容有异议或投诉,请联系2913721942#qq.com核实处理,我们将尽快回复您,谢谢合作!
若转载请注明出处: Python中临时文件的作用什么,创建需要注意哪些?
本文地址: https://pptw.com/jishu/651303.html