Selenium异常处理优化：如何更优雅实现元素操作的异常捕获？

当然有更优雅的实现方式，以下几种方案可以有效消除冗余代码，提升可读性和维护性：

方案一：封装通用元素操作函数 #

核心思路是将元素查找、操作执行和异常逻辑封装成一个复用函数，避免重复编写try-catch块。

from selenium.common.exceptions import NoSuchElementException
from selenium.webdriver.common.by import By

def perform_element_action(driver, by, locator, action, param=None, error_msg="Element not found"):
    try:
        element = driver.find_element(by, locator)
        if action == "click":
            element.click()
        elif action == "send_keys":
            if param is None:
                raise ValueError("Parameter required for send_keys action")
            element.send_keys(param)
        # 可扩展支持clear、submit等其他操作
    except NoSuchElementException as error:
        print(error_msg)
        raise error  # 直接抛出原异常，保留堆栈信息

# 调用示例
for link in link_list:
    web_driver.get(link)
    
    perform_element_action(
        web_driver,
        By.XPATH,
        XPath.BUTTON_EXPAND_LOGIN_BY_PASSWORD.value,
        "click",
        error_msg='Element not found button "Login by password"'
    )
    
    perform_element_action(
        web_driver,
        By.XPATH,
        XPath.INPUT_LOGIN.value,
        "send_keys",
        param=LOGIN,
        error_msg='Login field not found.'
    )
    
    perform_element_action(
        web_driver,
        By.XPATH,
        XPath.INPUT_PASSWORD.value,
        "send_keys",
        param=PASSWORD,
        error_msg='Password field not found.'
    )

这个方案的优势是集中处理异常逻辑，后续修改异常处理规则只需调整函数本身，代码简洁度大幅提升。

方案二：使用装饰器统一处理异常 #

如果希望每个元素操作的职责更单一，可以用装饰器将异常处理逻辑与业务操作分离。

from selenium.common.exceptions import NoSuchElementException
from functools import wraps
from selenium.webdriver.common.by import By

def handle_element_exception(error_msg):
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            try:
                return func(*args, **kwargs)
            except NoSuchElementException as error:
                print(error_msg)
                raise error
        return wrapper
    return decorator

# 定义带异常处理的操作函数
@handle_element_exception('Element not found button "Login by password"')
def click_expand_login_button(driver):
    driver.find_element(By.XPATH, XPath.BUTTON_EXPAND_LOGIN_BY_PASSWORD.value).click()

@handle_element_exception('Login field not found.')
def enter_login(driver, login):
    driver.find_element(By.XPATH, XPath.INPUT_LOGIN.value).send_keys(login)

@handle_element_exception('Password field not found.')
def enter_password(driver, password):
    driver.find_element(By.XPATH, XPath.INPUT_PASSWORD.value).send_keys(password)

# 调用示例
for link in link_list:
    web_driver.get(link)
    
    click_expand_login_button(web_driver)
    enter_login(web_driver, LOGIN)
    enter_password(web_driver, PASSWORD)

装饰器让每个操作函数只关注自身业务，异常处理逻辑被抽离出来，代码可读性和可维护性更强。

方案三：采用Page Object模式（适合大型项目） #

如果是规模较大的自动化测试项目，Page Object模式是行业标准实践，它将页面元素和操作封装成独立类，实现页面结构与业务逻辑的分离。

from selenium.common.exceptions import NoSuchElementException
from selenium.webdriver.common.by import By

class LoginPage:
    def __init__(self, driver):
        self.driver = driver
        # 定义页面元素定位器
        self.expand_login_btn = (By.XPATH, XPath.BUTTON_EXPAND_LOGIN_BY_PASSWORD.value)
        self.login_input = (By.XPATH, XPath.INPUT_LOGIN.value)
        self.password_input = (By.XPATH, XPath.INPUT_PASSWORD.value)
    
    def _safe_find_element(self, locator, error_msg):
        """内部复用的元素查找方法，统一处理异常"""
        try:
            return self.driver.find_element(*locator)
        except NoSuchElementException as error:
            print(error_msg)
            raise error
    
    def click_expand_login_button(self):
        """点击密码登录展开按钮"""
        element = self._safe_find_element(self.expand_login_btn, 'Element not found button "Login by password"')
        element.click()
    
    def enter_login(self, login):
        """输入登录账号"""
        element = self._safe_find_element(self.login_input, 'Login field not found.')
        element.send_keys(login)
    
    def enter_password(self, password):
        """输入密码"""
        element = self._safe_find_element(self.password_input, 'Password field not found.')
        element.send_keys(password)
    
    def complete_login(self, login, password):
        """完整登录流程"""
        self.click_expand_login_button()
        self.enter_login(login)
        self.enter_password(password)

# 调用示例
for link in link_list:
    web_driver.get(link)
    login_page = LoginPage(web_driver)
    login_page.complete_login(LOGIN, PASSWORD)

这种模式的优势在于：

• 元素定位集中管理，后续页面变更只需修改Page类
• 操作逻辑复用性强，多个测试用例可以共享同一个Page类
• 代码结构清晰，业务逻辑与页面操作分离

方案选择建议 #

• 小型脚本或简单场景：优先使用通用操作函数，快速减少冗余
• 中等规模项目：装饰器或通用函数均可，视代码风格偏好选择
• 大型自动化测试项目：必须采用Page Object模式，保障代码可维护性

内容的提问来源于stack exchange，提问作者Mop