Selenium 自动化截取网页指定区域

软件测试资料领取：[内部资源] 想拿年薪40W+的软件测试人员，这份资料必须领取~

软件测试面试刷题工具：软件测试面试刷题【800道面试题+答案免费刷】

一. 需求

在本篇博客中，我将分享一段使用Python编写的自动化网页截图代码，该代码基于Selenium和PIL库，可用于截取网页中指定区域的截图。这样的功能对于需要定期监控特定网页内容或进行网页数据采集的任务非常有用。

二. 代码解析

首先，我们使用Selenium库启动了一个Chrome浏览器，并通过设置选项隐藏了浏览器自动化控制提示。代码还实现了全屏效果，类似于按下F11键。

import time

from selenium import webdriver

from PIL import Image


# 启动浏览器

chrome_options = webdriver.ChromeOptions()

# 隐藏浏览器自动化控制提示

chrome_options.add_experimental_option('excludeSwitches', ['enable-automation'])


# 添加谷歌浏览器驱动位置

# chrome_options.binary_location = r"E:\应用\谷歌浏览器插件\chrome-win64\chrome.exe"


# 加载启动项页面全屏效果，相当于F11。

chrome_options.add_argument("--kiosk")


# 配置初始化

driver = webdriver.Chrome(options=chrome_options)

然后，我们打开了指定的网页，并获取了页面的宽度和高度。

# 打开页面

url = 'https://baidu.com'# 请替换为你要打开的页面

driver.get(url)


# 获取页面宽度与高度

hight,width = 'return document.body.clientHeight','return document.body.clientWidth'

h,w = driver.execute_script(hight),driver.execute_script(width)

print(w,h)

接下来，我们指定了要截取的区域的坐标，并使用Selenium的save_screenshot方法保存整个页面的截图。

# 指定四个点的坐标，这里以左上角和右下角的坐标为例

# x1起始宽度到 x2最大宽度的距离

# y1起始高度到 y2最大高度的距离

x1, y1 = 0, 0# 左上角坐标

x2, y2 = w, h-600# 右下角坐标


# 获取整个页面截图

screenshot_path = 'screenshot.png'

driver.save_screenshot(screenshot_path)

随后，我们使用PIL库打开整个页面的截图，并通过crop方法截取指定区域。最后，保存截取的区域截图及关闭浏览器

# 截取指定区域

im = Image.open(screenshot_path)

region = im.crop((x1, y1, x2, y2))


# 保存截图

cropped_path = 'result_screenshot.png'

region.save(cropped_path)


# 关闭浏览器

driver.close()

三. 完整代码

将代码中的网页链接和截取区域坐标调整为你想要的值。

运行代码，即可生成指定区域的网页截图。

import time


from selenium import webdriver

from PIL import Image


# 启动浏览器

# 隐藏浏览器自动化控制提示

chrome_options = webdriver.ChromeOptions()

chrome_options.add_experimental_option('excludeSwitches', ['enable-automation'])


# 添加谷歌浏览器驱动位置

# chrome_options.binary_location = r"E:\应用\谷歌浏览器插件\chrome-win64\chrome.exe"


# 加载启动项页面全屏效果，相当于F11。

chrome_options.add_argument("--kiosk")


# 配置初始化

driver = webdriver.Chrome(options=chrome_options)


# 打开页面

url = 'https://baidu.com'# 请替换为你要打开的页面

driver.get(url)



# 获取页面宽度与高度

hight,width = 'return document.body.clientHeight','return document.body.clientWidth'

h,w = driver.execute_script(hight),driver.execute_script(width)

print(w,h)


# 指定四个点的坐标，这里以左上角和右下角的坐标为例

# x1起始宽度到 x2最大宽度的距离

# y1起始高度到 y2最大高度的距离

x1, y1 = 0, 0# 左上角坐标

x2, y2 = w, h-600# 右下角坐标


# 获取整个页面截图

screenshot_path = 'screenshot.png'

driver.save_screenshot(screenshot_path)


time.sleep(3)


# 截取指定区域

im = Image.open(screenshot_path)

region = im.crop((x1, y1, x2, y2))


# 保存截图

cropped_path = 'result_screenshot.png'

region.save(cropped_path)


# 关闭浏览器

driver.close()

既然看到这里，在收藏的同时，也请不吝啬的点个赞呗！期待 ~

最后感谢每一个认真阅读我文章的人，下方这份完整的软件测试教程已经整理上传完成，需要的朋友们可以文末自行领取：【保证100%免费】

​

这些资料，对于【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴上万个测试工程师们走过最艰难的路程，希望也能帮助到你！