如果脚本可能被中断，清理临时文件的最佳方法

Question

命令行脚本需要在存在之前清除它为使用而创建的临时文件。假设脚本在执行期间也可以被中断。 你应该在脚本中做以下哪一项？在

a:）使用atexit模块

b:）使用os.tempnam公司（）以创建临时文件

c:）定义一个函数

d:）以上都不是

Answer 1

您可以在此处定义自己的上下文管理器：

import os
class create_temp_file(object):
    def __enter__(self):
        """Define entry point actions here"""

        self.filename = os.tempnam()
        self.file_obj = open(self.filename, 'w')
        return self.file_obj

    def __exit__(self, ex_type, ex_value, ex_traceback):
        """define cleanup actions here"""

        self.file_obj.close()
        os.remove(self.filename)

现在使用with语句，这是python做try finally的方式

^{pr2}$

但是os.tempnam不安全，最好使用tempfile模块来做这些事情。在

RuntimeWarning: tempnam is a potential security risk to your program

^{3}$

如果delete为true（默认值），则文件一关闭即被删除（或当文件对象被垃圾回收时）