Python+Selenium开发YouTube直播聊天Bot：iframe定位失败求助

需求与现有代码 #

我想用Python+Selenium开发一个YouTube直播聊天Bot，实现选中消息输入框、输入内容并发送的功能。当前初始代码如下：

from selenium.webdriver.chrome.service import Service
from webdriver_manager.chrome import ChromeDriverManager
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.by import By
import undetected_chromedriver as uc

chrome_options = Options()
chrome_options.add_argument('--allow-profiles-outside-user-dir')
chrome_options.add_argument('--enable-profile-shortcut-manager')
chrome_options.add_argument(r'user-data-dir=\.\User')
chrome_options.add_argument('--profile-directory=Profile 3')
driver = uc.Chrome(service=Service(ChromeDriverManager().install()), options=chrome_options)
x = int(input())
...

启动Chrome后我手动进入带聊天的YouTube直播，但无法定位聊天对应的iframe（Selenium默认仅操作顶层框架），查看页面源码找不到该iframe的name、id属性，当前尝试的定位代码无效：

pis = driver.find_element(By.XPATH, '//*[@id="content-pages"]')
print(pis)
driver.switch_to.frame(pis) 

解决方案 #

1. 正确定位直播聊天iframe #

YouTube直播聊天的iframe虽然没有明确的id/name，但可以通过src属性包含live_chat的特征来定位，同时需要用显式等待确保iframe加载完成（避免页面动态加载导致元素未出现）：

首先导入显式等待所需的模块：

from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

然后替换原有的iframe定位代码：

# 等待iframe加载并定位
live_chat_iframe = WebDriverWait(driver, 10).until(
    EC.presence_of_element_located((By.XPATH, "//iframe[contains(@src, 'live_chat')]"))
)
# 切换到该iframe
driver.switch_to.frame(live_chat_iframe)

2. 定位消息输入框并发送内容 #

切换iframe后，就可以定位输入框和发送按钮：

# 导入按键模块（如果用回车发送需要）
from selenium.webdriver.common.keys import Keys

# 定位聊天输入框（YouTube直播聊天输入框的特征是textarea嵌套在id为input的div中）
chat_input = WebDriverWait(driver, 10).until(
    EC.presence_of_element_located((By.XPATH, "//div[@id='input']//textarea"))
)
# 输入内容
chat_input.send_keys("你要发送的消息内容")
# 方式一：按回车发送
chat_input.send_keys(Keys.ENTER)

# 方式二：定位发送按钮点击（备选）
# send_button = driver.find_element(By.XPATH, "//button[@aria-label='发送']")
# send_button.click()

3. 切换回顶层框架（可选） #

如果后续需要操作页面其他元素，记得切回顶层：

driver.switch_to.default_content()

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