如何在Jupyter Notebook中用ipywidgets即时中止任务

你的核心问题是耗时任务阻塞了主线程，导致ipywidgets的事件回调（中止点击）必须等待主线程中的sleep(10)执行完毕才能触发。要实现即时响应中止，需要把耗时任务放到独立线程/异步任务中，让主线程能及时处理UI事件。

方法一：使用线程（threading） #

通过线程将耗时任务与UI线程分离，同时设置一个中止标志，让任务可以中途退出。

import ipywidgets as widgets
from IPython.display import display
from time import sleep
import threading

button = widgets.Button(description='run job')
output = widgets.Output()
abort_flag = False  # 中止标志位

def abort(event):
    global abort_flag
    with output:
        print('abort!')
        abort_flag = True
    # 恢复按钮状态
    button.description = 'run job'
    button.on_click(run_job)
    button.on_click(abort, remove=True)

def run_task():
    global abort_flag
    abort_flag = False
    with output:
        print('running job')
        # 把长耗时任务拆分为可中断的小步骤
        for _ in range(10):
            if abort_flag:
                print('job aborted!')
                break
            sleep(1)
        else:
            # 循环正常结束（未被中止）
            print('job complete!')
        # 任务结束后恢复按钮状态
        button.description = 'run job'
        button.on_click(run_job)
        button.on_click(abort, remove=True)

def run_job(event):
    # 切换按钮状态并绑定中止回调
    button.description = 'abort'
    button.on_click(run_job, remove=True)
    button.on_click(abort)
    # 启动子线程执行任务
    threading.Thread(target=run_task, daemon=True).start()

button.on_click(run_job)
display(button, output)

方法二：使用异步（asyncio） #

利用Jupyter对异步的原生支持，用异步函数执行耗时任务，避免阻塞主线程。

import ipywidgets as widgets
from IPython.display import display
import asyncio

button = widgets.Button(description='run job')
output = widgets.Output()
abort_flag = False

def abort(event):
    global abort_flag
    with output:
        print('abort!')
        abort_flag = True
    button.description = 'run job'
    button.on_click(run_job)
    button.on_click(abort, remove=True)

async def run_task():
    global abort_flag
    abort_flag = False
    with output:
        print('running job')
        for _ in range(10):
            if abort_flag:
                print('job aborted!')
                break
            await asyncio.sleep(1)
        else:
            print('job complete!')
        button.description = 'run job'
        button.on_click(run_job)
        button.on_click(abort, remove=True)

def run_job(event):
    button.description = 'abort'
    button.on_click(run_job, remove=True)
    button.on_click(abort)
    # 在Jupyter中运行异步任务
    asyncio.create_task(run_task())

button.on_click(run_job)
display(button, output)

关键说明 #

• 两种方案都通过拆分长耗时任务为小片段，并在每个片段检查中止标志，实现任务的中途退出。
• 线程方案适合CPU密集型任务，异步方案更适合IO密集型任务（比如网络请求、文件读写）。
• 任务结束/中止后，都需要恢复按钮的回调和显示状态，避免后续点击异常。

内容的提问来源于stack exchange，提问作者lane