如何点击ipywidgets按钮后回到原函数继续执行流程

首先得指出你原来代码里的几个关键问题：

• on_click方法只是注册点击回调函数，它不会阻塞执行流程，也不会返回回调函数的返回值——所以你后面写的if continues.on_click(clicked) == "continue"完全不会按预期工作，因为on_click返回的是None，而且这行代码会在按钮显示后立刻执行，根本不等你点击。
• 回调函数clicked的返回值不会被任何地方捕获，它只是在点击时执行打印，没办法把状态传递回index函数。

要实现“点击按钮后回到index函数暂停处继续执行”的需求，我们可以用同步阻塞的方式，让index函数在显示按钮后暂停，直到某个按钮被点击，再根据点击的按钮进入对应的分支。下面给你两种可行的实现方案：

方案1：使用threading.Event实现同步等待 #

这种方式利用线程事件来阻塞index函数，直到按钮被点击后触发事件，同时记录点击的按钮类型：

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

def index():
    print('hello')
    # 创建事件对象和存储点击结果的变量
    click_event = threading.Event()
    clicked_button = None

    def on_button_click(button):
        nonlocal clicked_button
        clicked_button = button.description
        click_event.set()  # 触发事件，解除阻塞

    continues = widgets.Button(description="continue")
    finish = widgets.Button(description="finish")
    
    continues.on_click(on_button_click)
    finish.on_click(on_button_click)
    
    display(continues, finish)
    click_event.wait()  # 暂停在这里，直到按钮被点击触发事件

    # 现在根据点击的按钮进入对应分支
    if clicked_button == "continue":
        print('here!')
    elif clicked_button == "finish":
        print('here2')

index()

方案2：使用Jupyter的异步支持（推荐） #

Jupyter原生支持异步函数，我们可以用async/await配合按钮的wait_for_click方法（这个方法会返回一个可等待对象，直到按钮被点击）：

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

async def index():
    print('hello')
    continues = widgets.Button(description="continue")
    finish = widgets.Button(description="finish")
    
    display(continues, finish)

    # 等待任意一个按钮被点击
    clicked = await widgets.Widget.wait_for_click([continues, finish])
    
    # 判断是哪个按钮被点击
    if clicked is continues:
        print('here!')
    elif clicked is finish:
        print('here2')

# 在Jupyter中运行异步函数
await index()

两种方案的说明： #

• 方案1用线程事件实现同步阻塞，兼容更多场景，不需要异步语法；
• 方案2是Jupyter推荐的异步写法，代码更简洁，符合交互式环境的设计思路。

这两种方案都能实现你要的效果：index函数在显示按钮后暂停，直到你点击其中一个按钮，然后回到暂停处，根据点击的按钮进入对应的条件分支执行后续代码。

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