Python-Selenium报错截图
报错截图设计方案：
功能：截图层主要用来存放selenium运行时的报错截图信息
1. 截图路径管理
- 分层存储：在项目根目录下创建
screenshots文件夹，并按日期进一步分类（如20250601）。 - 命名规范：截图文件名包含错误类型和精确时间戳（如
error_el_message_20250601_143022.png），便于快速定位问题。 - 自动创建目录：使用
os.makedirs(exist_ok=True)确保目录存在，避免手动创建。
2. 多类型错误检测
- 多样化定位策略：支持 XPath、ID、ClassName、CSS 选择器等多种定位方式，覆盖不同框架的错误组件（如 Element UI、Ant Design）。
- 关键词扫描：若未找到特定错误组件，扫描页面源码中的错误关键词（如
error、错误、异常）。 - 自定义错误类型：为不同定位方式设置专属错误类型标签（如
el_message、ant_message），便于后续分析。
3. 智能错误捕获
-
显式等待机制：使用
WebDriverWait等待错误元素出现，避免因页面加载延迟导致的误判。 -
异常处理
：
- 捕获
TimeoutException和NoSuchElementException以跳过不存在的定位器。 - 处理页面源码扫描时的异常，并记录为
page_source_error。
- 捕获
-
多轮检测：先检查特定错误组件，再扫描页面源码，确保全面覆盖。
4. 信息完整性
- 错误信息记录：截图时保存错误文本内容（如
用户名不存在/密码错误），便于后续分析。 - 操作反馈：通过打印日志提示截图路径和错误详情，支持自定义消息。
5. 扩展性设计
- 可配置定位器：错误定位器以元组形式存储在列表中，便于添加新的定位方式。
- 错误类型分类：支持自定义错误类型标签，方便后续统计和筛选。
6. 异常安全保障
- 多层异常捕获：在不同层级捕获异常，确保即使某个检测环节失败，仍能生成通用错误截图。
- 健壮的路径生成：使用
os.path模块处理路径，确保跨平台兼容性。
示例代码：
class ErrorChecker:def __init__(self, driver):self.driver = driverdef get_screenshot_path(self, error_type="unknown"):"""获取截图保存路径"""# 获取项目根目录project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))# 在项目根目录下创建screenshots文件夹screenshot_dir = os.path.join(project_root, "screenshots")# 创建截图目录（如果不存在）os.makedirs(screenshot_dir, exist_ok=True)# 获取当前日期作为子文件夹名date_folder = time.strftime("%Y%m%d")screenshot_path_dir = os.path.join(screenshot_dir, date_folder)# 创建日期子目录（如果不存在）os.makedirs(screenshot_path_dir, exist_ok=True)# 生成带完整日期时间的截图文件名timestamp = time.strftime("%Y%m%d_%H%M%S")return os.path.join(screenshot_path_dir, f"error_{error_type}_{timestamp}.png")def save_screenshot(self, error_type="unknown", message=""):"""保存截图并打印消息"""screenshot_path = self.get_screenshot_path(error_type)self.driver.save_screenshot(screenshot_path)if message:print(message)print(f"📸 错误截图已保存至: {screenshot_path}")return Truedef check_and_capture_error(self):"""检查页面是否存在错误提示，并在发现时截图"""# 定义多种定位方式的错误提示元素error_locators = [# XPath定位(By.XPATH, "//div[contains(@class, 'el-message--error')]//p", "el_message"),(By.XPATH, "//div[contains(@class, 'ant-message-error')]", "ant_message"),(By.XPATH, "//div[contains(@class, 'error-text')]", "error_text"),(By.XPATH, "//div[contains(@class, 'toast-error')]", "toast_error"),# ID定位(By.ID, "errorMessage", "error_message_id"),(By.ID, "error-container", "error_container_id"),# Class Name定位(By.CLASS_NAME, "error-message", "error_message_class"),(By.CLASS_NAME, "alert-danger", "alert_danger_class"),# CSS选择器定位(By.CSS_SELECTOR, ".error-box .message", "error_box_css"),(By.CSS_SELECTOR, "#errorContent", "error_content_css"),# Name属性定位(By.NAME, "error-display", "error_display_name"),(By.NAME, "errorInfo", "error_info_name")]try:# 遍历所有可能的错误提示元素for locator_type, locator_value, error_type in error_locators:try:error_element = WebDriverWait(self.driver, 1).until(EC.visibility_of_element_located((locator_type, locator_value)))error_text = error_element.textreturn self.save_screenshot(error_type, f"⚠️ 捕获到错误: {error_text}")except (TimeoutException, NoSuchElementException):continue# 如果没有找到自定义错误提示,则检查页面是否存在其他错误try:# 获取页面源码page_source = self.driver.page_source# 如果页面中包含常见的错误关键词,则进行截图error_keywords = ['error', 'exception', 'failed', '错误', '异常', '失败']for keyword in error_keywords:if keyword.lower() in page_source.lower():return self.save_screenshot(f"keyword_{keyword}", f"⚠️ 捕获到错误关键词: {keyword}")print("✅ 未发现任何错误")return Falseexcept Exception as e:return self.save_screenshot("page_source_error", f"❌ 检查错误时发生异常: {str(e)}")except Exception as e:return self.save_screenshot("general_error", f"❌ 检查错误时发生异常: {str(e)}")