element.py

#element.py

# Import necessary modules
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.common.by import By

# Define a class for web elements
class BasePageElement:
    # Function to set a value for any element on a web page
    def __set__(self, obj, value):
        driver = obj.driver
        WebDriverWait(driver, 10).until(
            lambda driver: driver.find_element(By.ID, self.locator)
        )
        driver.find_element(By.ID, self.locator).clear()
        driver.find_element(By.ID, self.locator).send_keys(value)
    
    # Function to get a value from any element on a web page
    def __get__(self, obj, owner):
        driver = obj.driver
        WebDriverWait(driver, 10).until(
            lambda driver: driver.find_element(By.ID, self.locator)
        )
        element = driver.find_element(By.ID, self.locator)
        return element.get_attribute("value")