最终确保某些代码以原子方式运行，无论发生什么

Question

假设我要编写一个捕获KeyboardInterrupt异常的Python脚本，以便能够由用户使用Ctrl+C安全地终止

然而，我无法将所有关键操作（如文件写入）放入catch块，因为它依赖于局部变量，并确保后续的Ctrl+C不会破坏它。

使用一个带有空（pass）try部分和finally部分内的所有代码的try-catch块来将这个代码段定义为"原子的、中断安全的代码"，可能不会在中途中断，这是否可行，也是一种好的做法？

示例：

try:
    with open("file.txt", "w") as f:
        for i in range(1000000):
            # imagine something useful that takes very long instead
            data = str(data ** (data ** data))
            try:
                pass
            finally:
                # ensure that this code is not interrupted to prevent file corruption:
                f.write(data)
except KeyboardInterrupt:
        print("User aborted, data created so far saved in file.txt")
        exit(0)

在这个例子中，我不关心当前生成的数据字符串，也就是说，创建可能会被中断，并且不会触发写入。但一旦开始写入，就必须完成，这就是我想要确保的。此外，如果在finally子句中执行写入时发生异常（或KeyboardInterrupt），会发生什么？

Answer 1

finally中的代码也可以被中断。Python对此没有任何保证；它所保证的只是在try套件完成之后或者如果try套件中出现异常，执行将切换到finally套件。try只能处理在其范围内引发的异常，而不能处理其范围外的异常，并且finally在该范围外。

因此，在pass语句上使用try是没有意义的。通行证是不可操作的，它永远不会被中断，但finally套件仍然可以很容易地被中断。

你需要选择一种不同的技术。您可以写入一个单独的文件，并在成功完成后将其移动到位；操作系统保证文件移动是原子的。或者记录上一次成功写入的位置，如果下一次写入中断，则将文件截断到该位置。或者在文件中写入标记，表示记录成功，这样读取就知道要忽略什么。

Answer 2

在您的情况下，没有问题，因为文件写入是原子的，但如果您有一些更复杂的文件对象实现，则try-except位于错误的位置。您必须将异常处理放在write:周围

try:
    f.write(data)
except:
    #do some action to restore file integrity
    raise

例如，如果您编写二进制数据，您可以执行以下操作：

filepos = f.tell()
try:
    f.write(data)
except:
    # remove the already written data
    f.seek(filepos)
    f.truncate()
    raise