封装查找元素以及集成日志输出,Base模块

2019-07-24 09:16:11来源:博客园 阅读 ()

新老客户大回馈,云服务器低至5折

关于查找元素封装,Base模块

"""
Selenium 封装基类

封装查找元素以及集成日志输出
"""

from HTMLReport import logger
from selenium import webdriver
from selenium.webdriver.remote.webelement import WebElement


class Base(object):
    def __init__(self, driver: webdriver.Remote = None):
        self.driver = driver
        self.logger = logger()

    def find_element(self, locator: tuple) -> WebElement:
        """查找单个元素

        :param locator: 定位器(by=By.ID, value=None)
        :return: WebElement
        """
        self.logger.info(f"查找元素:{locator}")
        try:
            element = self.driver.find_element(*locator)
        except Exception:
            self.logger.info("查找元素失败")
            raise
        else:
            self.logger.info("查找元素成功")
        return element

    def find_elements(self, locator: tuple):
        """查找元素集合

        :param locator: 定位器(by=By.ID, value=None)
        :return: 元素对象集合
        :rtype: list of WebElement
        """
        self.logger.info(f"查找元素集合:{locator}")
        elements = self.driver.find_elements(*locator)
        self.logger.info(f"查找到{len(elements)}个元素")
        return elements

 


原文链接:https://www.cnblogs.com/CesareZhang/p/11140084.html
如有疑问请与原作者联系

标签:

版权申明:本站文章部分自网络,如有侵权,请联系:west999com@outlook.com
特别注意:本站所有转载文章言论不代表本站观点,本站所提供的摄影照片,插画,设计作品,如需使用,请与原作者联系,版权归原作者所有

上一篇:python 基础学习笔记(3)--列表与元组

下一篇:coco数据集标注图转为二值图python(附代码)