Python_Selenium之basepage 识别元素、浏览器操作、获取属性、鼠标事件、键盘事件、弹窗、切换frame、切换句柄等封装

#coding=gbk
import os
import time
from selenium import webdriver
from selenium.webdriver.common.by import By #导入by方法
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.action_chains import ActionChains ##对鼠标事件操作
from selenium.webdriver.common.keys import Keys # 对键盘事件操作
from common import HTMLTestReportCN #测试报告
from common.log_utils import logger
from common.config_utils import config

class BasePage(object):
def __init__(self,driver):
self.driver=driver

# ------ -------------------------------------浏览器操作封装----------------------------------------------#
#打开网页
def open_url(self,url):
self.driver.get(url)
logger.info('打开URL地址%s;'%url)

def set_browser_max(self):
self.driver.maximize_window()
logger.info("设置浏览器的最大化")

def close_tab(self):
self.driver.close()
logger.info('关闭当前的tab页签')

def set_brower_min(self):
self.driver.minimize_window()
logger.info("设置浏览器的最小化")

def brower_refresh(self):
self.driver.refresh()
logger.info("浏览器的刷新操作")

def get_title(self):
value=self.driver.title
logger.info("获取网页的标题为：%s"%value)
return value

def quit_brower(self):
self.driver.quit()
logger.info("关闭浏览器")

# ------ ------------------------------------- 等待封装---------------------------------------------------#
#测试OK
def implicitly_wait(self,seconds=config.get_timeout):
self.driver.implicitly_wait(seconds)

def wait(self,seconds=config.get_timeout):
time.sleep(seconds)

# ------ ------------------------------------- 元素识别的封装----------------------------------------------#
def find_element(self,element_info):
try:
locator_type_name=element_info['locator_type']
locator_value_info=element_info['locator_value']
locator_timeout=element_info['timeout']
if locator_type_name =='id':
locator_type=By.ID
elif locator_type_name=='class':
locator_type=By.CLASS_NAME
elif locator_type_name=='xpath':
locator_type=By.XPATH
# 识别元素
element=WebDriverWait(self.driver,locator_timeout)\
.until(lambda x:x.find_element(locator_type,locator_value_info))
logger.info('[%s]元素识别成功'%element_info['element_name'])
except Exception as e:
logger.error('[%s]元素不能识别，原因是%s' % (element_info['element_name'], e.__str__()))
self.screenshot_as_file() # 失败后截屏
return element

# ------ -------------------------------------获取属性封装----------------------------------------------#
#获取元素属性封装-测试OK
def get_element_attribute(self,element_info,attribute_name):
element=self.find_element(element_info)
value=element.get_attribute(attribute_name)
logger.info('[%s]元素的%s的值为：%s'%(element_info['element_name'],attribute_name,value))
return value

#获取元素text属性封装-测试OK
def get_text(self,element_info):
text=self.find_element(element_info).text
logger.info('[%s]元素的text值为：%s'%(element_info['element_name'],text))
return text

#------ ------------------------------------- 鼠标事件封装----------------------------------------------#
#元素鼠标操作封装：-测试OK
def click(self,element_info):
element=self.find_element(element_info)
logger.info('[%s]元素进行点击操作'%element_info['element_name'])
element.click()

#元素鼠标操作：右击-测试OK
def context_click(self,element_info):
mouse = ActionChains(self.driver)
element = self.find_element(element_info)
logger.info('[%s]元素进行右击操作' % element_info['element_name'])
mouse.context_click(element).perform()

#元素鼠标操作：移动到该元素上--测试OK
def move_to_element_by_mouse(self,element_info):
element = self.find_element(element_info)
mouse = ActionChains(self.driver)
logger.info('将鼠标移动到[%s]元素上' % element_info['element_name'])
mouse.move_to_element(element).perform()

def long_press_element(self,element_info,senconds):
element = self.find_element(element_info)
logger.info('将鼠标长按到[%s]元素上后松开' % element_info['element_name'])
mouse = ActionChains(self.driver)
mouse.click_and_hold(element).pause(senconds).release(element)

def scrollIntoView(self,element_info):
element=self.find_element(element_info)
logger.info('将滚动条滚动至[%s]元素可见'%element_info['element_name'])
self.driver.execute_script('arguments[0].scrollIntoView();', element)
time.sleep(1)

# ------ ------------------------------------- 键盘事件封装-------------------------------------------#
# #键盘事件操作封装：输入内容--OK
def input(self,element_info,content):
element=self.find_element(element_info)
element.send_keys(content)
logger.info('[%s]元素输入内容：%s'%(element_info['element_name'],content))

#删除内容-测试OK
def back_space(self,element_info):
element=self.find_element(element_info)
logger.info('[%s]元素操作back_space' % (element_info['element_name']))
element.send_keys(Keys.BACK_SPACE)

#清空内容--测试OK
def clear_input(self,element_info):
element=self.find_element(element_info)
logger.info('[%s]元素输入框操作清空' % (element_info['element_name']))
element.clear()

#按回车--测试OK
def enter(self,element_info):
element=self.find_element(element_info)
logger.info('[%s]元素进行回车键操作' % (element_info['element_name']))
element.send_keys(Keys.ENTER)

#全选 Ctrl+a--测试OK
def ctrl_a(self,element_info):
element=self.find_element(element_info)
logger.info('[%s]元素输入框内容进行全选操作' % (element_info['element_name']))
element.send_keys(Keys.CONTROL, 'a')

#粘贴 Ctrl +x--测试OK
def ctrl_x(self,element_info):
element=self.find_element(element_info)
logger.info('[%s]元素输入框内容进行剪切操作' % (element_info['element_name']))
element.send_keys(Keys.CONTROL, 'x')

#粘贴 Ctrl +v--测试OK
def ctrl_v(self,element_info):
element=self.find_element(element_info)
logger.info('[%s]元素输入框内容进行粘贴操作' % (element_info['element_name']))
element.send_keys(Keys.CONTROL, 'v')

# ------ ------------------------------------- 弹窗封装----------------------------------------------#
#弹窗处理封装-测试OK
def get_alert_content(self,action='accept',timeout=config.get_timeout):
WebDriverWait(self.driver, timeout).until(EC.alert_is_present())
alter = self.driver.switch_to.alert
value = alter.text
logger.info('当前弹窗的内容为：%s' % value)
if action == 'accept':
alter.accept()
elif action == 'dismiss':
alter.dismiss()
return value

# ------ -------------------------------------frame封装-----------------------------------------------#
#frame封装frame == > id/name frame元素对象
#思路一
def switch_to_frame(self,element_info):
element = self.find_element(element_info)
self.driver.switch_to.frame(element)
logger.info('[%s]frame元素切换成功' %(element_info["element_name"]))

#思路二
def switch_to_frame_id_or_name(self,id_or_name):
self.driver.switch_to.frame(id_or_name)
def switch_to_frame_by_element(self,element_info):
element = self.find_element(element_info)
self.driver.switch_to.frame(element)
logger.info('[%s]frame元素切换成功' % (element_info["element_name"]))

#------------------------------------selenium执行js封装-----------------------------------------------#
def execute_script(self,js_str,element_info=None):
if element_info:
self.driver.execute_script(js_str)
else:
self.driver.execute_script(js_str,None)

def delete_element_attribute(self,element_info,attribute_name):
element = self.find_element(element_info)
self.driver.execute_script('arguments[0].removeAttribute("%s");'%attribute_name,element)

def update_element_attribute(self, element_info, attribute_name,attribute_value):
element = self.find_element(element_info)
self.driver.execute_script('arguments[0].setAttribute("%s","%s");' %(attribute_name,attribute_value), element)

# ------ -------------------------------------切换句柄封装-------------------------------------------#
##方式一--测试OK
def switch_window_by_title(self,title):
for handle in self.driver.window_handles:
self.driver.switch_to.window(handle)
if self.driver.title.__contains__(title):
break
logger.info('切换到网页标题为[%s]的句柄成功!'%title)

#方式二--测试OK
def switch_window_by_url(self,url):
for handle in self.driver.window_handles:
self.driver.switch_to.window(handle)
if self.driver.current_url.__contains__(url):
break
logger.info('切换到URL为[%s]的句柄成功!' % url)

#----------------------------截图封装----------------------------------------------------------#
#测试OK
# def screenshot_as_file_old(self, *screenshot_path):
# current_dir = os.path.dirname(__file__)
# if len(screenshot_path) == 0:
# screenshot_filepath = config.screenshot_path
# else:
# screenshot_filepath = screenshot_path[0]
# now = time.strftime('%Y_%m_%d_%H_%M_%S')
# screenshot_filepath_finally = os.path.join(current_dir, '../' ,screenshot_filepath, 'UITest_%s.png' % now)
# self.driver.get_screenshot_as_file(screenshot_filepath_finally)
# logger.info('图片截取成功')

#如果要在测试报告中截图，需要使用这个截图封装
def screenshot_as_file(self):
report_path = os.path.join( os.path.abspath(os.path.dirname(__file__)) , '..', config.report_path)
report_dir = HTMLTestReportCN.ReportDirectory(report_path)
report_dir.get_screenshot( self.driver )