atomicloop 1.0.0

入门

pip3 install atomicloop

以下示例循环运行 3 次迭代，每次持续 3 秒。您可以尝试通过按Ctrl+C（将 SIGINT发送到进程）来中断其中一个步骤。您将看到信号反馈，但程序将等待当前迭代完成并在循环迭代完成后退出。

此外，您可以看到上下文管理器知道它是否由于中断而完成。

import time
from atomicloop import AtomicLoop


# Optional callback handlers ----------------------------------------
def on_signal(signal, frame):
print(f'  Received signal: {signal}')
print('  Finishing current step before terminating')


def on_exit(was_interrupted):
print('  Ended training')
print(f'  Interrupted? {was_interrupted}')


# Running the loop --------------------------------------------------
goal_steps = 3

with AtomicLoop(on_signal, on_exit) as loop:
        step_no = 0
        while loop.run and step_no < goal_steps:
                print(f'Starting training step #{step_no}')
                time.sleep(3) # simulating some code execution
                print(f'Finished training step #{step_no}')
                print('----------------------------------\n')
                step_no += 1

文档

atomicloop.AtomicLoop(on_signal=None, on_end=None)

创建所需上下文管理器的主要且唯一的类。

on_signal是在传入的 SIGINT或SIGTERM上调用的可调用对象。它使用 2 个参数调用：信号编号和当前堆栈帧。有关参数的详细信息，请参阅官方 signal.signal 文档中的处理程序部分 。

on_end是在退出上下文后（在with块之后）被调用的可调用对象。它使用单个布尔参数调用，说明执行是否被中断。

AtomicLoop()返回一个新实例。要访问布尔标志以进行流量控制，您必须将该标志作为实例的属性读取。

.run布尔标志表明程序是否应该继续执行。默认情况下 为True ，并在收到中断信号时切换为False 。

您还可以使用许多同义属性之一：

• 循环运行

• 循环.loop

• 循环移动

• 循环.cont

• 循环保持

• loop.keep_going

• 循环不间断