日志记录和错误处理，让脚本运行更稳定

1. 日志记录

今天我们要深入探讨一个可能拯救你不少时间和精力的话题——日志记录和错误处理。哪怕是最有经验的程序员也难免会犯错，而你的任务就是让这些错误变得可预测并可控。想象一下，你就是一名考古学家，正在挖掘遗址，而你的任务就是保护你的团队不受突如其来的塌陷或陷阱影响——日志记录和错误处理就是你在这个充满风险的过程中值得信赖的指南针和盾牌！

当我们谈论编程中的日志记录时，我们指的是记录程序运行时信息的过程。这些过程可能在你的代码内部运行，一开始不容易被察觉。通过记录日志，你就像格林童话里的韩塞尔和格蕾特在森林中留下面包屑一样，为自己留下了可以回家的指引。

设置与配置日志记录

我们从 Python 的基本工具开始。为了记录日志，我们将使用内置模块 logging。它就像是你的瑞士军刀，可以记录程序执行过程中的各种信息。

import logging

# 配置日志记录
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

# 使用示例
logging.info("脚本开始运行")
logging.error("这是一个错误")

        

日志记录有哪些级别？

• DEBUG: 最详细的信息，通常只对问题诊断有帮助。
• INFO: 确认程序按预期工作。
• WARNING: 表明某些意外事件或潜在问题，但程序仍然正常运行。
• ERROR: 严重的错误，导致某个功能无法执行。

你可以设置日志记录级别来过滤信息。例如，如果你只对错误和警告感兴趣，可以设置级别为 logging.WARNING。

Selenium 中的日志记录示例

让我们看看日志记录如何在 Selenium 项目中使用：

from selenium import webdriver

# 配置日志记录
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

try:
    logging.info("初始化 Chrome 驱动程序")
    driver = webdriver.Chrome()
    logging.info("打开 Python.org 页面")
    driver.get("https://www.python.org")
    logging.info("成功完成任务")
except Exception as e:
    logging.error(f"发生错误: {e}")
finally:
    driver.quit()

        

这个示例展示了如何将日志记录集成到您的 Selenium 脚本中以跟踪其执行情况。我们在这里使用了 try 、except 和 finally 来处理错误——下面会详细讨论。

2. Selenium 中的错误处理

在编程中，就像在现实生活中，错误是不可避免的。但是我们可以控制如何应对它们。使用 try 、except 、else 和 finally ，可以优雅地捕获和处理错误。

回顾异常处理

让我们回顾一下异常处理如何工作：

try:
    # 可能引发异常的代码
    result = 10 / 0
except ZeroDivisionError as e:
    # 如果 ZeroDivisionError 异常被引发，将执行此代码
    logging.error("无法除以零！")
else:
    # 如果未引发异常，将执行此代码
    logging.info("操作成功完成")
finally:
    # 无论是否引发异常，都会执行此代码
    logging.info("操作已完成")

        

您的代码将在 try 块中运行。如果发生 ZeroDivisionError 类型的错误，except 块中的代码将被执行，程序不会崩溃。else 块用于在未发生异常时执行代码。最后，finally 块无论是否发生错误都会执行。

Selenium 中的错误处理

Selenium 提供了一些内置异常，用于处理各种情况。以下是一些有用的异常及其使用方法。

1. NoSuchElementException — 如果找不到某个元素，将引发此异常。可用于检查元素的可用性。
   Python

   
   try:
       element = driver.find_element(By.ID, "element_id")
   except NoSuchElementException:
       logging.warning("未找到 ID 为 'element_id' 的元素。")
   
                   

2. TimeoutException — 如果页面中某个期望的元素未在规定时间内出现，将引发此异常。用于管理等待。
   Python

   
   from selenium.webdriver.support.ui import WebDriverWait
   from selenium.webdriver.support import expected_conditions as EC
   
   try:
       element = WebDriverWait(driver, 10).until(
           EC.presence_of_element_located((By.ID, "element_id"))
       )
   except TimeoutException:
       logging.error("等待 ID 为 'element_id' 的元素超时失败。")
   
                   

3. ElementClickInterceptedException — 如果一个元素被另一个元素遮挡，导致无法点击，将引发此异常。对处理被弹窗挡住的元素点击操作很有帮助。
   Python

   
   from selenium.common.exceptions import ElementClickInterceptedException
   
   try:
       driver.find_element(By.ID, "button_id").click()
   except ElementClickInterceptedException:
       logging.warning("元素被另一个元素遮挡，无法点击。")
   
                   

重试与恢复逻辑

为了提高脚本的稳定性，可以在发生错误时实现重试机制。例如，当网站暂时不可用或没有找到某个元素时，脚本可以尝试多次执行操作，然后再抛出错误。

import time

def retry_find_element(driver, by, value, retries=3):
    attempt = 0
    while attempt < retries:
        try:
            element = driver.find_element(by, value)
            logging.info(f"找到元素: {value}")
            return element
        except NoSuchElementException:
            attempt += 1
            logging.warning(f"第 {attempt} 次尝试失败，2 秒后重试...")
            time.sleep(2)
    logging.error(f"经过 {retries} 次尝试仍未找到元素。")
    return None

        

脚本稳定性与可靠性建议

• 使用显式等待: 为那些不会立即出现的元素设置显式等待。这会减少因为加载延迟导致的错误。
• 在每个阶段添加 try-except 块: 这样可以记录错误并避免因为小问题导致脚本中止。
• 关键步骤设置重试机制: 如果错误可能是暂时的，可以设置几秒间隔的重试机制。
• 记录每个步骤: 记录每个关键步骤的日志，有助于分析脚本的执行和解决隐藏的错误。
• 为关键错误设置告警: 例如，如果日志中出现 CRITICAL 级别的错误，可以发送邮件或聊天通知。

3. 高效利用日志

现在我们已经学会了记录日志和处理错误，让我们再进一步，看看如何从日志中获得最大收益。

日志存储位置与格式的选择

如果你想保存日志供以后分析，可以使用文件存储日志。这就像写日记一样，可以随时翻阅和分析，避免未来犯错。我们来设置日志记录到文件：

logging.basicConfig(filename='app.log', filemode='w', level=logging.INFO)

# 现在所有日志都会保存到 app.log 文件中
logging.info("日志记录到文件已经开始。")

        

分析日志

分析你的日志，找出常见错误或性能问题。或许你的脚本只在某些特定的日子或时间段会遇到问题，而只有通过细致分析日志才能发现。这种分析就如同一次私家侦探调查，有助于你更好地了解你的应用程序如何运行。