selenium自动化--截图封装

最新推荐文章于 2025-04-17 14:38:49 发布

等待小王子的花

最新推荐文章于 2025-04-17 14:38:49 发布

阅读量637

点赞数

分类专栏： selenium自动化

本文链接：https://blog.youkuaiyun.com/Robert_Hero/article/details/106636037

版权

selenium自动化专栏收录该内容

21 篇文章

订阅专栏

本文介绍了如何在Selenium自动化测试中进行截图功能的封装。首先，将基础方法如截图操作等封装到basepage.py文件，接着展示了一个具体的测试用例，详细说明了如何在测试过程中实现自动截图，以便于调试和问题定位。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

先将基本方法封装写进basepage.py文件中

# coding=utf-8
import os
import time

from unittest_demo.logger import Logger

mylog = Logger(logger='BasePage').getlog()


class BasePage(object):
    """
    主要是把常用的几个Selenium方法封装到BasePage这个类，我们这里演示以下几个方法
    back()
    forward()
    get()
    quit()
    """

    def __init__(self, driver):
        """
        写一个构造函数，有一个参数driver
        :param driver:
        """
        self.driver = driver

    def back(self):
        """
        浏览器后退按钮
        :param none:
        """
        self.driver.back()

    def forward(self):
        """
        浏览器前进按钮
        :param none:
        """
        self.driver.forward()

    def open_url(self, url):
        """
        打开url站点
        :param url:
        """
        self.driver.get(url)

    def quit_browser(self):
        """
        关闭并停止浏览器服务
        :param none:
        """
        self.driver.quit()

    def take_screenshot(self):
        """
        截图并保存在根目录下的Screenshots文件夹下
        :param none:
        """
        file_path = os.path.dirname(os.getcwd()) + '/Screenshots'
        rq = time.strftime('%Y%m%d%H%M%S', time.localtime(time.time()))
        screen_name = file_path + rq + '.png'
        try:
            self.driver.get_screenshot_as_file(screen_name)
            mylog.info("开始截图并保存")

        except Exception as e:
            mylog.error("出现异常", format(e))

再写一个测试截图的测试用例：

# coding=utf-8
import time
from selenium import webdriver

from testFile.basepage import BasePage


class TestScreenshot(object):
    driver = webdriver.Chrome()
    driver.maximize_window()
    driver.implicitly_wait(10)
    basepage = BasePage(driver)

    def test_take_screen(self):
        self.basepage.open_url("https://www.baidu.com")
        time.sleep(1)
        self.basepage.take_screenshot()
        self.basepage.quit_browser()


test = TestScreenshot()
test.test_take_screen()