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

 

　　"""

 

　　Selenium封装基类

 

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

 

　　"""

 

　　fromHTMLReportimportlogger

 

　　fromseleniumimportwebdriver

 

　　fromselenium.webdriver.remote.webelementimportWebElement

 

　　classBase(object):

 

　　def__init__(self,driver:webdriver.Remote=None):

 

　　self.driver=driver

 

　　self.logger=logger()

 

　　deffind_element(self,locator:tuple)->WebElement:

 

　　"""查找单个元素

 

　　:paramlocator:定位器（by=By.ID,value=None）

 

　　:return:WebElement

 

　　"""

 

　　self.logger.info(f"查找元素：{locator}")

 

　　try:

 

　　element=self.driver.find_element(*locator)

 

　　exceptException:

 

　　self.logger.info("查找元素失败")

 

　　raise

 

　　else:

 

　　self.logger.info("查找元素成功")

 

　　returnelement

 

　　deffind_elements(self,locator:tuple):

 

　　"""查找元素集合

 

　　:paramlocator:定位器（by=By.ID,value=None）

 

　　:return:元素对象集合

 

　　:rtype:listofWebElement

 

　　"""

 

　　self.logger.info(f"查找元素集合：{locator}")

 

　　elements=self.driver.find_elements(*locator)

 

　　self.logger.info(f"查找到{len(elements)}个元素")

 

　　returnelements