Python+Selenium 自动化框架篇5 -- 框架内封装元素方法

韦爵社

已于 2022-06-01 15:15:30 修改

阅读量2.4k

点赞数 1

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

于 2022-04-27 15:34:48 首次发布

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

版权

本文介绍了如何在Python Selenium自动化框架中封装基类，包括元素定位和截图功能，并展示了页面对象模型（POM）的实现。通过基类，每个页面类都能继承通用操作。此外，讨论了元素定位的最佳实践，强调了保持driver一致性的关键性。测试执行后，日志和截图会被保存到相应文件夹。

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

前面文章，我们实现了框架的一部分功能，包括日志类和浏览器引擎类的封装，今天我们继续封装一个基类和介绍如何实现POM。关于基类，是这样定义的：把一些常见的页面操作的selenium封装到base_page.py这个类文件，以后每个POM中的页面类，都继承这个基类，这样每个页面类都有基类的方法，这个我们会在这篇文章实现。

在实现封装基类里，我们实现了元素八大方式的定位和截图类封装。具体项目层级结构如下图。
基类base_page.py的具体实现代码，这里就封装了几个常用方法，其他方法，你自己去练习封装下。

# coding=utf-8
import time
from selenium.common.exceptions import NoSuchElementException
from selenium.webdriver.common.by import By
import os.path
from framework.logger import Logger


# create a logger instance
logger = Logger(logger="BasePage").getlog()


class BasePage(object):
    """
    定义一个页面基类，让所有页面都继承这个类，封装一些常用的页面操作方法到这个类
    """

    def __init__(self, driver):
        self.driver = driver

    # 退出浏览器并结束测试
    def quit_browser(self):
        self.driver.quit()

    # 浏览器前进操作
    def forward(self):
        self.driver.forward()
        logger.info("在当前页面点击前进.")

    # 浏览器后退操作
    def back(self):
        self.driver.back()
        logger.info("点击返回当前页面.")

    # 隐式等待
    def wait(self, seconds):
        self.driver.implicitly_wait(seconds)
        logger.info("等待 %d 秒." % seconds)


    # 切换窗口的方法
    def switch_windows(self):
        handles = self.driver.window_handles
        self.driver.switch_to.window(handles[-1])
        logger.info("切换新的窗口")

    # 点击关闭当前窗口
    def close(self):
        try:
            self.driver.close()
            logger.info("关闭并退出浏览器.")
        except NameError

最低0.47元/天解锁文章