使用nodriver处理SeatGeek验证码时遇AttributeError：无法访问iframe

问题场景 #

使用nodriver处理SeatGeek网站初始验证码时，尝试通过select_all('#captcha__audio__button', include_frames=True)查找音频按钮，触发AttributeError: 'NoneType' object has no attribute 'node_id'。报错显示识别到了iframe，但该iframe无实际内容、文档对象未初始化，无法正常访问其node_id属性。

报错堆栈 #

File "/home/pooh/work/xxx/./seatgeek.py", line 324, in get_result_nd
    res=await self.page.select_all('#captcha__audio__button', include_frames=True) # , best_match=True
        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/pooh/venv12/lib/python3.12/site-packages/nodriver/core/tab.py", line 307, in select_all
    items.extend(await fr.query_selector_all(selector))
                 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/pooh/venv12/lib/python3.12/site-packages/nodriver/core/element.py", line 797, in query_selector_all
    return await self.tab.query_selector_all(selector, _node=self)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/pooh/venv12/lib/python3.12/site-packages/nodriver/core/tab.py", line 377, in query_selector_all
    cdp.dom.query_selector_all(doc.node_id, selector)
                               ^^^^^^^^^^^
AttributeError: 'NoneType' object has no attribute 'node_id'

解决方案 #

• 单独等待iframe加载完成
  不要直接通过select_all包含iframe查询，先定位iframe元素并等待其内容加载完成，再切换上下文查找目标按钮：

  # 定位验证码相关iframe
  iframe = await self.page.select_one('iframe[src*="captcha"]')
  # 等待iframe内容加载完成
  await iframe.wait_for_load()
  # 在iframe内查找音频按钮
  audio_button = await iframe.select_one('#captcha__audio__button')
  if audio_button:
      await audio_button.click()
  

• 提前校验iframe有效性
  遍历页面iframe时，先检查其文档对象是否初始化，避免NoneType报错：

  frames = await self.page.select_all('iframe')
  for fr in frames:
      doc = await fr.get_document()
      if doc is not None:
          audio_button = await fr.select_one('#captcha__audio__button')
          if audio_button:
              await audio_button.click()
              break
  

• 增加页面等待时长
  验证码iframe加载速度可能较慢，在页面加载后增加等待时间，确保iframe完全初始化：

  await self.page.wait_for_timeout(3000)  # 等待3秒
  res = await self.page.select_all('#captcha__audio__button', include_frames=True)
  

• 通过CDP直接操作元素
  绕过nodriver的iframe封装，直接调用CDP接口查找并点击按钮：

  # 调用CDP查询元素
  elements = await self.page.tab.execute_cdp(
      "DOM.querySelectorAll",
      {"selector": "#captcha__audio__button"}
  )
  if elements:
      # 调用CDP执行点击
      await self.page.tab.execute_cdp(
          "DOM.click",
          {"nodeId": elements[0]["nodeId"]}
      )
  

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