如何让Selenium WebDriver在Bokeh文档中与Pyro远程对象协同工作？

我之前遇到过一模一样的Bokeh前后端同步坑，你的问题核心原因很清晰：Selenium直接点击DOM元素只是触发了前端视觉变化，没触发Bokeh服务器端的状态更新流程，而手动点击会自动完成前后端的WebSocket消息同步，所以Pyro能拿到正确状态。下面是具体的解决思路和代码实现：

问题根源拆解 #

• 手动点击Bokeh按钮时，前端会向Bokeh服务器发送WebSocket消息，触发Python端Toggle对象的active属性更新，同时执行相关回调逻辑。
• Selenium的普通click()操作只是模拟了DOM的点击事件，只会改变按钮的前端样式（比如颜色），但没有触发Bokeh内置的后端同步机制，导致Python端的active值始终停留在初始状态，Pyro自然拿不到更新后的值。

解决方案：调用Bokeh的JavaScript API触发完整交互 #

Bokeh为每个组件都提供了JavaScript层面的官方API，我们可以通过Selenium执行JS代码，直接调用Bokeh组件的click()方法，这样就能触发完整的前后端同步流程。

示例代码 #

from selenium import webdriver
from selenium.webdriver.common.by import By
import time

# 初始化浏览器驱动
driver = webdriver.Chrome()
driver.get("http://localhost:5006/view_server_so")
time.sleep(2)  # 等待Bokeh文档加载完成

# 定位Bokeh按钮的DOM元素（可根据按钮样式调整选择器）
btn_dom = driver.find_element(By.CSS_SELECTOR, ".bk-btn.bk-btn-default")

# 执行JS代码，触发Bokeh组件的官方点击事件
driver.execute_script("""
    // 获取当前页面的Bokeh文档
    const doc = Bokeh.documents[0];
    // 通过DOM元素的data-bk-model-id属性找到对应的Bokeh组件实例
    const btnModel = doc.get_model_by_id(arguments[0].getAttribute('data-bk-model-id'));
    // 调用组件的click方法，触发完整的前后端状态同步
    btnModel.click();
""", btn_dom)

# 短暂等待状态同步（可选，根据网络情况调整）
time.sleep(0.5)

关键细节说明 #

• 每个Bokeh组件的DOM元素都带有data-bk-model-id属性，这个ID和后端Python对象的id属性一一对应，是前后端关联的核心标识。
• 通过Bokeh.documents[0]获取当前页面的Bokeh文档（多文档场景可调整索引），再用get_model_by_id拿到JS层面的组件实例，调用它的click()方法才会触发WebSocket消息，同步更新后端的active状态。

额外排查点 #

1. 确认Pyro暴露的对象实例：确保在Bokeh服务器脚本中，你注册到Pyro的是同一个Toggle实例，而不是重新创建的副本。比如：

   # 正确写法：把已创建的Toggle对象注册到Pyro
   toggle_btn = Toggle(active=False, label="Task")
   Pyro4.expose(toggle_btn.get_btn_task_active_state)
   Pyro4.Daemon.serveSimple({toggle_btn: "bokeh.toggle"}, host="localhost", port=9090)
   

2. 查看Bokeh服务器日志：执行Selenium操作后，检查Bokeh服务器的日志，如果看到有回调函数的执行记录，说明后端状态已经更新，此时Pyro就能拿到正确值了。

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