Python 实现Selenium录屏的一种方法(图片整合成动态图)

由于UI层自动化的不稳定性,经常会遇到执行中断或用例失败的问题,以下是一些常见的措施。

1.详细的日志

2.定位出错时截图

3.Pytest的缓存机制(可以记录成功了哪些失败了哪些)

4.自动重试机制(如pytest-rerunfailures)

5.用例录像

用例录像是最直观的一种方式,可以查看到实时的操作情况。很多云平台如Saucelabs等都提供了用例录像的功能。

然而很遗憾selenium本身没有这种能力。除了借助三方软件如ffmpeg同步录屏外,另外一种实现方法是另外启动一个线程,不断截图,最后将图片拼接成git。

操作方法如下。

实时截图

1.编写一个循环截图函数shot

def shot(dr):

i = 0

while True:

img_file = os.path.join(img_dir, f'{i}.png')

try:

dr.save_screenshot(img_file)

except:

return

i += 1

由于webdriver接口执行截图命令的速度限制,每一轮截图无需再sleep。

2.操作web时,启动一个线程

img_dir = 'img' # 临时图片目录

dr = webdriver.Chrome()

t = threading.Thread(target=shot, args=(dr, img_dir)) # 新建线程

t.start() # 启动截图线程

dr.get('https://www.baidu.com')

dr.find_element('id', 'kw').send_keys('简书 韩志超')

dr.find_element('id', 'su').click()

time.sleep(1)

dr.get('https://www.qq.com')

dr.back()

time.sleep(2)

dr.quit()

3.图片拼接成gif

需要安装pillow: pip install pillow

img_list = os.listdir(img_dir) # 列出目录所有图片

img_list.sort(key=lambda x: int(x[:-4])) # 排序

first_img = Image.open(os.path.join(img_dir, img_list[0])) # 第一张图片对象

else_imgs = [Image.open(os.path.join(img_dir, img)) for img in img_list[1:]] # 剩余图片对象

first_img.save("record.gif", append_images=else_imgs,

duration=300,

save_all=True) # 拼接保存

完整代码

from selenium import webdriver

import threading

import os

import time

from PIL import Image

def clear_dir(path):

"""创建或清空目录"""

if not os.path.isdir(path):

os.mkdir(path) # 创建目录

else: # 清空目录

os.remove(os.path.join(path, file_name)) for file_name in os.listdir(path)

def shot(dr, img_dir):

"""循环截图函数"""

i = 0

clear_dir(img_dir) # 清空目录

while True:

img_file = os.path.join(img_dir, f'{i}.png')

try:

dr.save_screenshot(img_file)

except:

return

i += 1

Selenium操作

img_dir = 'img' # 临时图片目录

dr = webdriver.Chrome()

t = threading.Thread(target=shot, args=(dr, img_dir)) # 新建线程
t.start() # 启动截图线程

dr.get('https://www.baidu.com')

dr.find_element('id', 'kw').send_keys('简书 韩志超')

dr.find_element('id', 'su').click()

time.sleep(1)

dr.get('https://www.qq.com')

dr.back()

time.sleep(2)

dr.quit()

# 图片拼接成gif
img_list = os.listdir(img_dir) # 列出目录所有图片
img_list.sort(key=lambda x: int(x[:-4])) # 排序

first_img = Image.open(os.path.join(img_dir, img_list[0])) # 第一张图片对象
else_imgs = [Image.open(os.path.join(img_dir, img)) for img in img_list[1:]] # 剩余图片对象

first_img.save("record.gif", append_images=else_imgs,
duration=300, # 每张图片的过过渡时间
save_all=True) # 拼接保存,如果想要循环播放可以加上loop=0

相关推荐
luckys.one6 分钟前
第9篇:Freqtrade量化交易之config.json 基础入门与初始化
javascript·数据库·python·mysql·算法·json·区块链
大翻哥哥2 小时前
Python 2025:量化金融与智能交易的新纪元
开发语言·python·金融
zhousenshan3 小时前
Python爬虫常用框架
开发语言·爬虫·python
IMER SIMPLE3 小时前
人工智能-python-深度学习-经典神经网络AlexNet
人工智能·python·深度学习
CodeCraft Studio3 小时前
国产化Word处理组件Spire.DOC教程:使用 Python 将 Markdown 转换为 HTML 的详细教程
python·html·word·markdown·国产化·spire.doc·文档格式转换
专注API从业者4 小时前
Python/Java 代码示例:手把手教程调用 1688 API 获取商品详情实时数据
java·linux·数据库·python
java1234_小锋4 小时前
[免费]基于Python的协同过滤电影推荐系统(Django+Vue+sqlite+爬虫)【论文+源码+SQL脚本】
python·django·电影推荐系统·协同过滤
看海天一色听风起雨落5 小时前
Python学习之装饰器
开发语言·python·学习
XiaoMu_0015 小时前
基于Python+Streamlit的旅游数据分析与预测系统:从数据可视化到机器学习预测的完整实现
python·信息可视化·旅游
THMAIL5 小时前
深度学习从入门到精通 - 生成对抗网络(GAN)实战:创造逼真图像的魔法艺术
人工智能·python·深度学习·神经网络·机器学习·生成对抗网络·cnn