Python使用'with'在使用后删除文件

Question

我正在使用一个明确命名的文件作为临时文件。为了确保我能正确删除这个文件，我不得不为打开文件的操作创建一个包装类。

这看起来是可行的，但我有两个疑问：

A] 这样做安全吗？

B] 有没有更好的方法？

import os

string1 = """1. text line
2. text line
3. text line
4. text line
5. text line
"""
class tempOpen():
    def __init__(self, _stringArg1, _stringArg2):
        self.arg1=_stringArg1
        self.arg2=_stringArg2

    def __enter__(self):
        self.f= open(self.arg1, self.arg2)
        return self.f

    def __exit__(self, exc_type=None, exc_val=None, exc_tb=None):
        self.f.close()
        os.remove(self.arg1)

if __name__ == '__main__':

    with tempOpen('tempfile.txt', 'w+') as fileHandel:
        fileHandel.write(string1)
        fileHandel.seek(0)
        c = fileHandel.readlines()
        print c

顺便说一下：由于很多原因，我不能使用tempfile.NamedTemporaryFile。

Answer 1

我想你可以用 contextlib.contextmanager 来简化一些：

from contextlib import contextmanager

@contextmanager
def tempOpen( path, mode ):
    # if this fails there is nothing left to do anyways
    file = open(path, mode)

    try:
        yield file
    finally:
        file.close()
        os.remove(path)

你需要处理两种不同的错误：一种是创建文件时出错，另一种是写入文件时出错。