python - 仅在引发异常时清理对象

Question

如果不是必须对其执行的所有操作都成功（也就是说，如果引发异常），我需要删除一个文件。它可以像使用except:一样简单，删除文件然后重新引发异常，但在这种情况下，如果由于任何神秘原因无法在except子句中删除文件，则原始异常将丢失。

我能想到的最好的是：

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 的方法吗？

Answer 1

如果您不想丢失它，另一种选择是简单地存储异常：

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块中的子类，因为您不想在此处捕获SystemExit或捕获KeyboardInterrupt。还要报告在取消链接期间发生的任何异常，而不是仅仅删除它。