使用Python向Selenium WebDriver WebElements添加函数

from selenium import webdriver
from selenium.webdriver.remote.webelement import WebElement

# 创建自定义的WebDriver类
class CustomWebDriver(webdriver.Chrome):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

    # 添加自定义函数，用于获取元素的文本内容
    def get_element_text(self, element):
        return element.text

# 创建自定义的WebElement类
class CustomWebElement(WebElement):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

    # 添加自定义函数，用于点击元素并等待页面加载完成
    def click_and_wait(self):
        self.click()
        self.parent.wait_for_page_to_load()

# 创建自定义的WebDriver实例
driver = CustomWebDriver()

# 打开网页
driver.get('https://www.example.com')

# 获取元素并调用自定义函数
element = CustomWebElement(driver, driver.find_element_by_xpath('//a[@id="example-link"]'))
text = element.get_element_text()
element.click_and_wait()