如何在Python Selenium中给Web元素赋值变量并规避StaleElementReference异常

StaleElementReference异常的本质是：你保存的元素引用对应的DOM节点已经被更新、移除或重新渲染，导致旧引用失效。你原代码中把page.checkbox赋值给chk后，这个变量只存储了第一次查找的静态元素实例，点击操作引发DOM变化后，旧引用自然无法再使用。

以下是几种可行的调整方案：

方案1：直接复用页面对象的属性（每次操作重新查找元素） #

不需要把元素赋值给变量，每次操作都通过页面对象的checkbox属性获取最新元素：

class PageObject:
    def __init__(self, driver):
        self.driver = driver
    @property
    def checkbox(self):
        return self.driver.find_element_by_css_selector('selector')

class TestCases:
    def test_case(self):    
        page = PageObject(self.driver)
        assert(page.checkbox.get_attribute("checked") == "false")
        page.checkbox.click()
        assert(page.checkbox.get_attribute("checked") == "true")

每次调用page.checkbox都会执行一次find_element，确保拿到的是当前DOM中的有效元素。

方案2：页面对象封装操作逻辑（隐藏元素细节） #

把元素的操作和状态判断封装成页面对象的方法，完全避免在测试代码中直接处理元素引用：

class PageObject:
    def __init__(self, driver):
        self.driver = driver
        self._checkbox_locator = ('css selector', 'selector')
    
    def _get_checkbox(self):
        return self.driver.find_element(*self._checkbox_locator)
    
    def is_checkbox_unchecked(self):
        return self._get_checkbox().get_attribute("checked") == "false"
    
    def toggle_checkbox(self):
        self._get_checkbox().click()

class TestCases:
    def test_case(self):    
        page = PageObject(self.driver)
        assert(page.is_checkbox_unchecked())
        page.toggle_checkbox()
        assert(not page.is_checkbox_unchecked())

这种方式更符合页面对象模式的设计原则，测试代码只关注业务逻辑，无需关心元素定位细节。

方案3：自定义可自动刷新的元素包装类 #

如果一定要把"元素"赋值给变量，可通过包装类实现每次操作前自动重新查找元素：

class RefreshableElement:
    def __init__(self, driver, locator):
        self.driver = driver
        self.locator = locator
    
    def _get_element(self):
        return self.driver.find_element(*self.locator)
    
    def get_attribute(self, attr_name):
        return self._get_element().get_attribute(attr_name)
    
    def click(self):
        self._get_element().click()

class PageObject:
    def __init__(self, driver):
        self.driver = driver
    
    @property
    def checkbox(self):
        return RefreshableElement(self.driver, ('css selector', 'selector'))

class TestCases:
    def test_case(self):    
        page = PageObject(self.driver)
        chk = page.checkbox
        assert(chk.get_attribute("checked") == "false")
        chk.click()
        assert(chk.get_attribute("checked") == "true")

包装类的每个方法都会重新查找元素，保证每次操作使用的都是当前DOM中的有效引用。

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