使用Python Selenium webdriver无法按class属性选择元素问题求助

核心错误原因 #

• 索引取值越界：find_elements返回的匹配元素列表长度小于2，直接取索引1会触发索引错误。先打印len(element)确认匹配到的元素总数，若目标是第一个匹配项，索引应改为0。
• 元素未完成加载：执行查找逻辑时页面还未渲染完成，匹配不到足够元素，需添加显式等待保证元素加载完成后再操作。
• 元素不在可视区：截图前目标元素未滚动到可视窗口内，获取的坐标偏移，裁剪结果不符合预期。
• 语法废弃：Selenium 4.0及以上版本已废弃find_elements_by_css_selector写法，需改用标准find_elements接口。

修复后可运行代码 #

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from PIL import Image

driver = webdriver.Firefox(executable_path=r'path/geckodriver.exe')
driver.get(href)

# 显式等待10秒，直到所有匹配的表格加载完成
wait = WebDriverWait(driver, 10)
elements = wait.until(EC.presence_of_all_elements_located((By.CSS_SELECTOR, "table.add_stat_table")))
# 打印匹配数量确认索引是否有效
print(f"匹配到的元素总数：{len(elements)}")
# 确认索引存在后再取值，目标为第一个元素时索引改为0
target_ele = elements[1]

# 滚动到目标元素位置，保证坐标准确
driver.execute_script("arguments[0].scrollIntoViewIfNeeded();", target_ele)

location = target_ele.location
size = target_ele.size
print("元素坐标：", location, "元素尺寸：", size)

driver.save_screenshot("path/full_screenshot.png")
x = location['x']
y = location['y']
right = x + size['width']
bottom = y + size['height']

im = Image.open("path/full_screenshot.png")
cropped = im.crop((int(x), int(y), int(right), int(bottom)))
cropped.save("path/cropped_target.png")

driver.quit()

特殊场景适配 #

如果使用高DPI屏幕或者开启了浏览器缩放，会出现坐标和截图像素不匹配的问题，可获取设备像素比后对坐标做乘法修正，或者启动浏览器时禁用缩放参数。

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