selenium2 頁面元素

?element.py?類大致是這樣的：

from selenium.webdriver.support.ui import WebDriverWait


class BasePageElement(object):
    """初始化每個頁面對象類的基本頁類"""

    def __set__(self, obj, value):
        """用給定的值設置文本"""
        driver = obj.driver
        WebDriverWait(driver, 100).until(
            lambda driver: driver.find_element_by_name(self.locator))
        driver.find_element_by_name(self.locator).send_keys(value)

    def __get__(self, obj, owner):
        """從具體的對象里獲取文本"""
        driver = obj.driver
        WebDriverWait(driver, 100).until(
            lambda driver: driver.find_element_by_name(self.locator))
        element = driver.find_element_by_name(self.locator)
        return element.get_attribute("value")