用Python装饰器降低selenium代码

在Python中，Selenium装饰器是一种强大的工具，可以帮助我们编写更简洁、易维护的代码，特别是当涉及到频繁的初始化、登录、会话管理等重复性任务时。装饰器本质上是一个函数，它接收另一个函数作为输入并返回一个新的函数，而新的函数通常会在原函数的基础上增加一些额外的功能。

以下是如何使用装饰器简化Selenium代码的一个例子：

import time
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

# 创建一个装饰器
def wait_and_click(element_locator):
    def wrapper(driver, element):
        WebDriverWait(driver, 10).until(
            EC.presence_of_element_located((By.XPATH, element_locator))
        )
        driver.find_element_by_xpath(element_locator).click()
        time.sleep(1)  # 添加延时，模拟用户交互
    return wrapper

# 使用装饰器
@wait_and_click("//button[@id='myButton']")
def perform_action(driver):
    driver.get("https://example.com")
    do_something()  # 这里是你具体的动作

perform_action(driver)  # 调用时，按钮会被自动点击

在这个例子中，`wait_and_click`装饰器负责查找并点击指定元素，这使得每次需要点击某个元素时，我们只需要传入元素定位器即可，无需在每个需要点击的地方重复写相同的等待和点击代码。