如何让ipywidget等待用户输入后再继续执行脚本

问题背景 #

在Jupyter笔记本中，为了提升交互性，需要替换原生的input()函数——因为脚本常需要多选项变量或本地文件路径这类输入，input()不够友好。选择ipywidgets能提供更便捷的输入组件，但默认情况下ipywidgets不会暂停代码执行，脚本会直接继续运行。尝试过官方文档的前两种方法及自定义按钮方案，虽能实现等待输入，但会导致后续代码结构混乱（比如嵌套按钮的情况），希望找到一种能像input()一样阻塞等待输入后再继续执行后续脚本的方式。

解决方案：使用jupyter-ui-poll包 #

通过jupyter-ui-poll包可以实现阻塞等待ipywidgets输入的效果，无需复杂的嵌套逻辑，代码结构更清晰。

示例代码 #

import ipywidgets as widgets
from jupyter_ui_poll import ui_events

def wait_for_widget(widget):
    """阻塞等待widget的交互事件"""
    with ui_events() as poll:
        while True:
            # 处理UI事件，超时10ms
            poll(10)
            # 检查是否有用户输入完成的标志（根据widget类型自定义）
            if hasattr(widget, 'value') and widget.value not in (None, ''):
                break
    return widget.value

# 示例1：下拉选择框
dropdown = widgets.Dropdown(
    options=['选项A', '选项B', '选项C'],
    value=None,
    description='选择选项:',
)
display(dropdown)
selected_option = wait_for_widget(dropdown)
print(f"你选择了: {selected_option}")

# 示例2：文件路径输入框
file_input = widgets.Text(
    value='',
    placeholder='输入文件路径',
    description='文件路径:',
)
display(file_input)
file_path = wait_for_widget(file_input)
print(f"你输入的文件路径是: {file_path}")

代码说明 #

• wait_for_widget函数封装了阻塞逻辑，通过jupyter-ui-poll的ui_events上下文管理器处理UI事件循环，每隔10ms检查一次widget的状态。
• 根据不同的widget类型（如下拉框、文本框），可以自定义判断输入完成的条件（示例中以value不为空/None作为完成标志）。
• 调用wait_for_widget后，代码会阻塞直到用户完成输入，后续代码可以像使用input()一样线性执行，避免嵌套混乱。

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