仅在引发异常时清理对象

时间:2011-10-07 12:08:46

标签: python exception try-catch-finally

如果不是必须对其执行的所有操作都成功(即,如果引发异常),则需要删除文件。它可以像使用一样简单,除了:,删除文件然后重新引发异常,但在这种情况下,如果无法在中删除文件,原始异常将会丢失除了条款,无论出于什么神秘的原因。

我能想到的最好的是:

try:
    file_path = "whatever.jpg"
    # do stuff with file
except:
    exception_raised = True
    raise
finally:
    try:
        if exception_raised:
            os.unlink(file_path)
    except:
        pass

return file_path   # everything OK

有人知道更好,更Pythonic的方法吗?

1 个答案:

答案 0 :(得分:5)

另一个选择是如果你不想丢失它,只需存储例外:

Python 3.x版本:

try:
    file_path = "whatever.jpg"
    # do stuff with file
except BaseException as e:
    try:
        os.unlink(file_path)
    except Exception:
        traceback.print_exc()
    raise e

Python 2.x版本稍微复杂一些,因为您需要手动存储完整的异常信息(否则您将丢失回溯):

try:
    file_path = "whatever.jpg"
    # do stuff with file
except:
    e = sys.exc_info()
    try:
        os.unlink(file_path)
    except Exception:
        traceback.print_exc()
    raise e[0], e[1], e[2]

修改:仅捕获内部Exception块中try的子类,因为您不希望捕获SystemExitKeyboardInterrupt这里。同时报告在取消链接期间发生的任何重复,而不是仅丢弃它。