Python进阶-day 26:自动化脚本 - 使用 Selenium 实现网页搜索

学习目标

  • 理解 Selenium 的基础用法,掌握如何控制浏览器进行自动化操作。
  • 学习如何定位网页元素并与之交互(如输入文本、点击按钮)。
  • 完成一个简单的自动化任务:在百度搜索“Selenium”并查看结果。

课程内容

1. 准备工作

  • 安装 Selenium: 在你的 Python 环境中安装 Selenium:bash
  • pip install selenium
  • 下载 ChromeDriver
    • 访问 chromedriver.chromium.org,下载与你的 Chrome 浏览器版本匹配的 ChromeDriver(例如,Chrome 135.x 需要对应的 ChromeDriver)。
    • 将 ChromeDriver 放置在指定路径,例如 /Users/xxxx/Downloads/chromedriver-mac-arm64/chromedriver。
  • 验证环境: 确保你的 Python 脚本可以找到 ChromeDriver,并确认 Chrome 浏览器已安装。

2. 学习 Selenium 基础

  • 导入必要的库:python
import time
from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By
    • time:用于添加暂停以便观察结果。
    • webdriver:Selenium 的核心模块,用于控制浏览器。
    • Service:管理 ChromeDriver 的服务。
    • By:用于定位网页元素。
  • 设置 ChromeDriver 路径:python指定 ChromeDriver 的路径并创建服务对象。
driver_path = "/Users/xxxx/Downloads/chromedriver-mac-arm64/chromedriver"
service = Service(driver_path)
  • 配置 Chrome 选项:python添加选项以减少网站检测到自动化行为(例如隐藏 Selenium 特征)。
options = webdriver.ChromeOptions()
options.add_argument("--disable-blink-features=AutomationControlled")
options.add_experimental_option("excludeSwitches", ["enable-automation"])
options.add_experimental_option("useAutomationExtension", False)
  • 初始化 WebDriver:python创建 Chrome WebDriver 实例,启动浏览器。
driver = webdriver.Chrome(service=service, options=options)
  • 打开网页:python使用 driver.get() 打开百度首页。
driver.get("https://www.baidu.com")
  • 定位元素并交互
    • 定位搜索框(id="kw")并输入文本:python
driver.find_element(By.ID, 'kw').send_keys("selenium")
      • By.ID:通过元素的 id 属性定位。
      • send_keys():模拟键盘输入,输入搜索关键词“Selenium”。
    • 定位搜索按钮(id="su")并点击:python
driver.find_element(By.ID, 'su').click()
      • click():模拟鼠标点击,触发搜索。
  • 错误处理与清理
    • 使用 try-except 捕获可能出现的异常:python
try:
    # 自动化操作
except Exception as e:
    print(f"发生错误:{e}")
    time.sleep(10)  # 暂停以便检查
    • 使用 finally 确保浏览器关闭:python
finally:
    driver.quit()  # 关闭浏览器

3. 完整代码示例

以下是完整的自动化脚本,用于在百度搜索“Selenium”:

python

import time
from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By

# 设置 chromedriver 路径
driver_path = "/Users/xxxx/Downloads/chromedriver-mac-arm64/chromedriver"
service = Service(driver_path)

# 配置 Chrome 选项以减少自动化检测
options = webdriver.ChromeOptions()
options.add_argument("--disable-blink-features=AutomationControlled")
options.add_experimental_option("excludeSwitches", ["enable-automation"])
options.add_experimental_option("useAutomationExtension", False)

try:
    # 初始化 Chrome WebDriver
    driver = webdriver.Chrome(service=service, options=options)

    # 打开百度首页
    driver.get("<https://www.baidu.com>")

    # 定位搜索框并输入关键词
    driver.find_element(By.ID, 'kw').send_keys("selenium")

    # 定位搜索按钮并点击
    driver.find_element(By.ID, 'su').click()

    print("搜索成功!")

    # 暂停查看结果
    time.sleep(5)

except Exception as e:
    print(f"发生错误:{e}")
    # 暂停以便手动检查页面
    time.sleep(10)

finally:
    # 关闭浏览器
    driver.quit()

4. 运行结果

  • 运行脚本后: Chrome 浏览器会自动打开并访问 https://www.baidu.com。 搜索框中会输入“selenium”。 点击搜索按钮后,页面显示搜索结果。 脚本暂停 5 秒以便查看结果,然后关闭浏览器。
  • 如果发生错误,脚本会打印错误信息并暂停 10 秒,方便调试。

扩展练习

  1. 改进等待机制

当前脚本使用 time.sleep() 进行固定等待,可能不稳定。尝试使用 Selenium 的显式等待(WebDriverWait):python

from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

# 等待搜索框出现
search_box = WebDriverWait(driver, 10).until(
    EC.element_to_be_clickable((By.ID, "kw"))
)
search_box.send_keys("selenium")

# 等待搜索按钮可点击
search_button = WebDriverWait(driver, 10).until(
    EC.element_to_be_clickable((By.ID, "su"))
)
search_button.click()

这样可以确保元素加载完成后再操作,提高脚本稳定性。

  1. 提取搜索结果

在搜索完成后,提取第一条搜索结果的标题:python

first_result = WebDriverWait(driver, 10).until(
    EC.presence_of_element_located((By.XPATH, "//h3[contains(@class, 't')]"))
)
print(f"第一条搜索结果标题:{first_result.text}")

检查百度搜索结果页面的 HTML 结构,调整 XPath 或其他定位方式。

  1. 保持浏览器打开

如果需要在搜索后执行更多操作(例如点击搜索结果),可以移除 finally 块中的 driver.quit(),并添加交互式选项:python

print("搜索完成,浏览器将保持打开状态...")
input("按 Enter 键关闭浏览器...")
driver.quit()

注意事项

  1. ChromeDriver 版本匹配

确保 ChromeDriver 版本与 Chrome 浏览器版本匹配。如果不匹配,可能导致 driver.get() 失败。

如果遇到问题,可以使用 webdriver-manager 自动管理 ChromeDriver:python

from webdriver_manager.chrome import ChromeDriverManager
service = Service(ChromeDriverManager().install())
  1. 反自动化检测

百度可能会检测到自动化行为(尽管可能性较低)。当前脚本已添加反检测选项(如 --disable-blink-features=AutomationControlled),但如果仍被识别,可以:

添加随机延迟:time.sleep(random.uniform(1, 3))(需导入 random)。

设置 User-Agent:python

options.add_argument("user-agent=Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/135.0.0.0 Safari/537.36")
  1. macOS 权限问题

如果 macOS 阻止 ChromeDriver 运行,运行以下命令移除隔离属性:bash

xattr -d com.apple.quarantine /Users/xxxx/Downloads/chromedriver-mac-arm64/chromedriver

确保 ChromeDriver 有执行权限:bash

chmod +x /Users/xxxx/Downloads/chromedriver-mac-arm64/chromedriver

总结

  • 你学习了如何使用 Selenium 自动化控制 Chrome 浏览器,完成了在百度搜索“Selenium”的任务。
  • 掌握了基本的元素定位(By.ID)和交互方法(send_keys 和 click)。
  • 通过扩展练习,你可以进一步学习显式等待、提取页面内容等高级用法。

建议:尝试将脚本应用到其他搜索引擎(例如 Google)或更复杂的页面,练习不同的元素定位方式(By.XPATH、By.CLASS_NAME 等)。如果遇到问题(如元素未找到或反自动化检测),可以手动检查页面 HTML 或调整脚本。

如果需要更复杂的自动化任务(例如登录网站、爬取数据),请继续学习下一课!