一. 需求
在本篇博客中,我将分享一段使用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()
现在我也找了很多测试的朋友,做了一个分享技术的交流群,共享了很多我们收集的技术文档和视频教程。
如果你不想再体验自学时找不到资源,没人解答问题,坚持几天便放弃的感受
可以加入我们一起交流。而且还有很多在自动化,性能,安全,测试开发等等方面有一定建树的技术大牛
分享他们的经验,还会分享很多直播讲座和技术沙龙
可以免费学习!划重点!开源的!!!
qq群号:691998057【暗号:csdn999】
三. 完整代码
将代码中的网页链接和截取区域坐标调整为你想要的值。
运行代码,即可生成指定区域的网页截图。
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()
下面是配套资料,对于做【软件测试】的朋友来说应该是最全面最完整的备战仓库,这个仓库也陪伴我走过了最艰难的路程,希望也能帮助到你!
最后: 可以在公众号:自动化测试老司机 ! 免费领取一份216页软件测试工程师面试宝典文档资料。以及相对应的视频学习教程免费分享!,其中包括了有基础知识、Linux必备、Shell、互联网程序原理、Mysql数据库、抓包工具专题、接口测试工具、测试进阶-Python编程、Web自动化测试、APP自动化测试、接口自动化测试、测试高级持续集成、测试架构开发测试框架、性能测试、安全测试等。
如果我的博客对你有帮助、如果你喜欢我的博客内容,请 “点赞” “评论” “收藏” 一键三连哦!