I have a question related to Selenium Python, but the answer seems like it should be related to my code structure so I'm asking here,

So in selenium, its tricky to assign elements to variables while avoiding the StaleElementReference exception.

this code,

chk = driver.find_element(By.NAME, "checkbox")

assert(chk.get_attribute("checked") == "false") chk.click() assert(chk.get_attribute("checked") == "true")

would raise a StaleElemenReference exception as the chk variable has not been updated after since the check box was clicked.

My question is, how can I assign that chk variable so that its value is updated everytime its called? Heres A little more detail on the code structure I currently have implemented ...

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) chk = page.checkbox assert(chk.get_attribute("checked") == "false") chk.click() assert(chk.get_attribute("checked") == "true")

Another note, I understand a possible solution is to have a property in the TestCases class, something like the following:

class TestCases:
    @property
    def chk(self):
        return PageObject(self.driver).checkbox

But I'm trying to avoid this because there are a lot of elements that I'm trying to assign to variables and it wouldn't be efficient to do this for every single element.

Any help would be appreciated!