最终确保一些代码以原子方式运行,无论如何？

Question

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

但是,我无法将所有关键操作(如文件写入)放入catch块中,因为它依赖于局部变量并确保后续Ctrl+ C无论如何都不会破坏它.

使用带有empty(pass)try部分的try-catch块以及部件中的所有代码finally将这个片段定义为"原子,中断安全代码"并且可能不会在中途中断时,它是否有效并且是一种好习惯？

例:

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不保证这一点; 所有这一切都保证finally在try套件完成后或者如果套件中出现异常,执行将切换到套件try.A try只能处理在其范围内而不是在其范围之外引发的异常,并且finally超出该范围.

因此,在声明中使用是没有意义try的pass.传球是一个无操作,它不会被打断,但finally套房很容易被打断.

你需要选择一种不同的技术.您可以写入单独的文件,并在成功完成后将其移动到位; 例如,操作系统保证文件移动是原子的.或者记录上次成功写入位置,并在下一次写入中断时将文件截断到该点.或者在文件中写入标记成功记录的标记,以便读取知道要忽略的内容.

Answer 2

在你的情况下,没有问题,因为文件写入是原子的,但如果你有一些文件对象实现,那就更复杂了,你try-except的位置错误.您必须围绕写入进行异常处理:

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