Python+Selenium 中级篇11 - 封装一个截图方法

韦爵社

已于 2022-04-24 10:02:00 修改

阅读量493

点赞数

分类专栏： Python + Selenium从基础到高级篇 python 文章标签： python selenium

于 2022-04-22 20:12:32 首次发布

本文链接：https://blog.youkuaiyun.com/aa122509969/article/details/124353908

版权

本文详细讲解如何将截图功能封装到Python的BasePage.py模块中。通过这个方法，截图会自动保存到项目根目录的 Screenshots 文件夹，并以当前系统时间为文件名，采用png作为文件格式。示例代码展示了一个测试类的应用，成功运行后，可以在指定目录找到对应的截图。

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

本文介绍把截图类方法封装到BasePage.py文件里，这个文件创建之前的二次封装方法有介绍：
我们截图类写死了把截图图片保存到根目录下的Screenshots文件夹里，图片名称是当前系统时间，图片后缀名是png。

在这里插入图片描述

# coding=utf-8
import os
import time

from test1.logger import Logger

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

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

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

    def