使用Selenium Python时遇AttributeError：'MainPage'对象无'find_element'属性

错误原因 #

你遇到的AttributeError: 'MainPage' object has no attribute 'find_element'，核心问题是：

• find_element是Selenium WebDriver实例的内置方法，但你的BasePage类既没有持有WebDriver对象的引用，也没有自己实现find_element方法
• 当MainPage继承BasePage后，调用self.find_element时找不到对应的属性/方法，因此抛出错误

解决方案 #

要实现元素可见性断言并解决这个错误，需要从以下几个方面修改代码：

1. 让BasePage持有WebDriver实例 #

修改BasePage的初始化逻辑，接收并保存WebDriver对象，同时封装通用方法：

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

class BasePage:
    def __init__(self, driver):
        self.driver = driver  # 保存WebDriver实例，供所有页面方法调用

    def find_element(self, by, value):
        # 封装元素查找方法，统一调用WebDriver的find_element
        return self.driver.find_element(by, value)

    def assert_button_visible(self, by, value):
        # 先通过显式等待确保元素加载完成，提升稳定性
        WebDriverWait(self.driver, 10).until(
            EC.presence_of_element_located((by, value))
        )
        element = self.find_element(by, value)
        # 断言元素是否显示
        assert element.is_displayed(), f"元素[{by}:{value}]未显示"

    def click(self, by, value):
        # 封装点击方法，复用元素查找逻辑
        self.find_element(by, value).click()

2. 优化MainPage的代码逻辑 #

统一管理元素定位符，避免硬编码重复的XPath：

from selenium.webdriver.common.by import By
from .base_page import BasePage

class MainPage(BasePage):
    # 把元素定位符作为类属性统一管理，便于维护
    ACCOUNT_LINK_LOCATOR = (By.XPATH, "//li[3]/a/span")

    def click_account(self):
        # 调用BasePage的断言方法，传入定位符
        self.assert_button_visible(*self.ACCOUNT_LINK_LOCATOR)
        # 调用BasePage的点击方法
        self.click(*self.ACCOUNT_LINK_LOCATOR)

3. 确保测试fixture正确传入WebDriver #

修改测试用例的fixture，保证main_page初始化时能拿到WebDriver实例：

import pytest
from selenium import webdriver
from pages.main_page import MainPage

@pytest.fixture
def driver():
    # 初始化WebDriver，这里以Chrome为例，可根据实际情况修改
    driver = webdriver.Chrome()
    yield driver
    # 测试结束后关闭浏览器
    driver.quit()

@pytest.fixture
def main_page(driver):
    # 初始化MainPage时传入WebDriver实例
    return MainPage(driver)

def test_click_account(main_page):
    main_page.click_account()

额外优化建议 #

• 使用显式等待替代直接查找元素：WebDriverWait能确保元素加载完成后再操作，比直接调用is_displayed()更稳定，减少因页面加载慢导致的测试失败
• 统一管理元素定位符：把所有页面的元素定位符都作为类属性保存，避免代码中重复出现相同的XPath/CSS选择器，后续修改时只需改一处
• 封装更多通用方法：如输入文本、获取元素文本等，提升代码复用性

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