# coding=utf-8
import time
from selenium.common.exceptions import NoSuchElementException
from selenium.webdriver import ActionChains
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.select import Select
from selenium.webdriver.support.wait import WebDriverWait
from Common.LoggerBase import Logger # 自己封装的logger类
from pathlib import *
'''--create a logger instance--'''
Logger = Logger()
class BasePage(object):
"""
定义一个页面基类，让所有页面都继承这个类，封装一些常用的页面操作方法到这个类
"""
'''--初始化--'''
def __init__(self, driver):
self.driver = driver
self.timeout = 10
self.poll_frequency = 0.5
'''--定义open方法--'''
def open(self, url):
self.driver.maximize_window()
self.driver.get(url)
'''--quit browser and end testing--'''
def quit_browser(self):
self.driver.quit()
'''--浏览器前进操作--'''
def forward(self):
self.driver.forward()
Logger.info("Click forward on current page.")
'''--浏览器后退操作--'''
def back(self):
self.driver.back()
Logger.info("Click back on current page.")
'''--隐式等待--'''
def wait(self, seconds):
self.driver.implicitly_wait(seconds)
Logger.info("wait for %d seconds." % seconds)
'''--点击关闭当前窗口--'''
def close(self):
try:
self.driver.close()
Logger.info("Closing and quit the browser.")
except NameError as e:
Logger.error("Failed to quit the browser with %s" % e)
'''--保存图片--'''
def get_windows_img(self):
"""
在这里我们把file_path这个参数写死，直接保存到我们项目根目录的一个文件夹.\Screenshots下
"""
file_path = Path(Path.cwd().parent) / 'Screenshots'
rq = time.strftime('%Y%m%d%H%M', time.localtime(time.time()))
screen_name = Path.joinpath(file_path, rq + '.png')
try:
self.driver.get_screenshot_as_file(str(screen_name))
Logger.info("Had take screenshot and save to folder : /Screenshots")
except NameError as e:
Logger.error("Failed to take screenshot! %s" % e)
self.get_windows_img()
'''--重写定位元素方法--'''
def find_element(self, selector):
"""
这个地方为什么是根据=>来切割字符串，请看页面里定位元素的方法
submit_btn = "id=>su"
login_lnk = "xpath => //*[@id='u1']/a[7]" # 百度首页登录链接定位
如果采用等号，结果很多xpath表达式中包含一个=，这样会造成切割不准确，影响元素定位
:param selector:
:return: element
"""
element = ''
if '=>' not in selector:
Logger.info("Not found '=>', default is ID")
return self.driver.find_element_by_id(selector)
selector_by = selector.split('=>')[0]
selector_value = selector.split('=>')[1]
if selector_by == "i" or selector_by == 'id':
try:
element = self.driver.find_element_by_id(selector_value)
Logger.info("Had find the element \" %s \" successful "
"by %s via value: %s " % (element.text, selector_by, selector_value))
except NoSuchElementException as e:
Logger.error("NoSuchElementException: %s" % e)
self.get_windows_img() # take screenshot
elif selector_by == "n" or selector_by == 'name':
try:
element = self.driver.find_element_by_name(selector_value)
Logger.info("Had find the element \" %s \" successful "
"by %s via value: %s " % (element.text, selector_by, selector_value))
except NoSuchElementException as e:
Logger.error("NoSuchElementException: %s" % e)
self.get_windows_img()
elif selector_by == "c" or selector_by == 'class_name':
try:
element = self.driver.find_element_by_class_name(selector_value)
Logger.info("Had find the element \" %s \" successful "
"by %s via value: %s " % (element.text, selector_by, selector_value))
except NoSuchElementException as e:
Logger.error("NoSuchElementException: %s" % e)
elif selector_by == "l" or selector_by == 'link_text':
try:
element = self.driver.find_element_by_link_text(selector_value)
Logger.info("Had find the element \" %s \" successful "
"by %s via value: %s " % (element.text, selector_by, selector_value))
except NoSuchElementException as e:
Logger.error("NoSuchElementException: %s" % e)
elif selector_by == "p" or selector_by == 'partial_link_text':
try:
element = self.driver.find_element_by_partial_link_text(selector_value)
Logger.info("Had find the element \" %s \" successful "
"by %s via value: %s " % (element.text, selector_by, selector_value))
except NoSuchElementException as e:
Logger.error("NoSuchElementException: %s" % e)
elif selector_by == "t" or selector_by == 'tag_name':
try:
element = self.driver.find_element_by_tag_name(selector_value)
Logger.info("Had find the element \" %s \" successful "
"by %s via value: %s " % (element.text, selector_by, selector_value))
except NoSuchElementException as e:
Logger.error("NoSuchElementException: %s" % e)
elif selector_by == "x" or selector_by == 'xpath':
try:
element = self.driver.find_element_by_xpath(selector_value)
Logger.info("Had find the element \" %s \" successful "
"by %s via value: %s " % (element.text, selector_by, selector_value))
except NoSuchElementException as e:
Logger.error("NoSuchElementException: %s" % e)
self.get_windows_img()
elif selector_by == "s" or selector_by == 'selector_selector':
try:
element = self.driver.find_element_by_css_selector(selector_value)
Logger.info("Had find the element \" %s \" successful "
"by %s via value: %s " % (element.text, selector_by, selector_value))
except NoSuchElementException as e:
Logger.error("NoSuchElementException: %s" % e)
else:
raise NameError("Please enter a valid type of targeting elements.")
return element
'''--输入--'''
def input_text(self, selector, text):
el = self.find_element(selector)
try:
el.send_keys(text)
Logger.info("Had type \" %s \" in inputBox" % text)
except NameError as e:
Logger.error("Failed to type in input box with %s" % e)
self.get_windows_img()
'''--清除文本框--'''
def clear(self, selector):
el = self.find_element(selector)
try:
el.clear()
Logger.info('Clear text in input box before typing')
except NameError as e:
Logger.error("Failed to clear in input box with %s" % e)
self.get_windows_img()
'''--点击元素--'''
def click(self, selector):
el = self.find_element(selector)
try:
el.click()
Logger.info("The element \" %s \" was clicked." % el)
except NameError as e:
Logger.error("Failed to click the element with %s" % e)
'''--获取网页标题--'''
def get_page_title(self):
Logger.info("Current page title is %s" % self.driver.title)
return self.driver.title
def send_key(self, locator, value):
"""
@param locator: 定位器
@param value: value
"""
elem = self.find_element(locator)
try:
elem.send_keys(value)
Logger.info("元素对象输入值成功，值为：{}".format(value))
except Exception as e:
Logger.error("元素对象输入值失败，错误信息为：{}".format(e))
def get_text(self, locator):
"""
@param locator:定位器
@return:元素文本值
"""
elem_text = None
elem = self.find_element(locator)
try:
elem_text = elem.text
except Exception as e:
Logger.error("元素text获取失败，错误信息为：{}".format(e))
Logger.info("元素text值：{}".format(elem_text))
return elem_text
'''--重写跳转switch_to_frame方法'''
def switch_to_frame(self, selector):
"""
:param selector: 传入定位器参数
:return:
"""
elem = self.find_element(selector)
try:
self.driver.switch_to.frame()
Logger.info("frame切换成功")
except Exception as e:
Logger.error("frame切换失败，错误信息为：{}".format(e))
'''--切回父级frame--'''
def turn_back_frame(self):
self.driver.switch_to.default_content()
def switch_to_handle(self, index):
"""
切换窗口句柄
"""
# 获取当前所有窗口句柄
global handles
try:
handles = self.driver.window_handles
Logger.info("获取当前所有窗口句柄成功，句柄对象列表为：{}".format(handles))
except Exception as e:
Logger.error("获取当前所有窗口句柄失败，错误信息为：{}".format(e))
# 切换到新窗口句柄
try:
self.driver.switch_to.window(handles[index])
Logger.info("切换新窗口句柄成功，切换窗口的索引index为：{}".format(index))
except Exception as e:
Logger.error("切换新窗口句柄失败，错误信息为：{}".format(e))
"""--多选、单选元素获取方法--"""
def select_by_text(self, selectByText):
if selectByText is not "":
select_by_text = self.find_element("xpath=>//*[text()=\"%s\"]" % selectByText)
select_by_text.click()
else:
print("Not find the selectByText, get the default value!")
pass
def move_mouse_to_element(self, locator):
"""
移动鼠标到某个元素上面
@param locator:
@return:
"""
elem = self.find_element(locator)
action = ActionChains(self.driver)
action.move_to_element(elem).perform()
def clear_input_value(self, locator):
"""
清除输入框中的内容
@param locator:
@return:
"""
elem = self.find_element(locator)
elem.send_keys(Keys.CONTROL, "a")
elem.send_keys(Keys.DELETE)
def get_value(self, locator):
"""
获取输入框的value
@param locator:
@return:
"""
elem = self.find_element(locator)
return elem.get_attribute("value")
def double_click_elem(self, locator):
"""
双击元素
@param locator:
@return:
"""
elem = self.find_element(locator)
ActionChains(self.driver).double_click(elem).perform()
def elem_is_display(self, locator):
"""
判断元素在页面是否显示，显示返回True,不显示返回false
@param locator:
@return:
"""
elem = self.find_element(locator)
return elem.is_displayed()
def elem_is_selected(self, locator):
"""
判断元素是否被选中，用于多选框，如果多选框被选中状态，返回True，否则返回False
@param locator:
@return:
"""
elem = self.find_element(locator)
return elem.is_selected()
def elem_is_enable(self, locator):
"""
判断页面元素是否可用
@param locator:
@return:
"""
elem = self.find_element(locator)
Logger.info("按钮的点击状态，是否可点击：{}".format(elem.is_enabled()))
return elem.is_enabled()
def choose_select_by_value(self, locator, value):
"""
根据内置属性value值,选择下拉输入框
@param locator:
@param value:
@return:
"""
elem = self.find_element(locator)
Select(elem).select_by_value(value)
def choose_select_by_index(self, locator, index):
"""
根据索引选择下拉框
@param locator:
@param index:
@return:
"""
elem = self.find_element(locator)
Select(elem).select_by_index(index)
def choose_select_by_visible_value(self, locator, value):
"""
根据下拉选项的文本值选择下拉框
@param locator:
@param value:
@return:
"""
elem = self.find_element(locator)
Select(elem).select_by_visible_text(value)
def elem_object_click(self, elem):
"""
元素点击，传入参数为元素对象
@param elem:
@return:
"""
elem.click()
def find_elements(self, selector):
"""
:param selector: 传入定位器参数locator => 标识切割
:return: 返回元素对象列表
"""
Logger.info("输出定位器信息：{}".format(selector))
element = ''
if '=>' not in selector:
Logger.info("Not found '=>', default is ID")
return self.driver.find_element_by_id(selector)
selector_by = selector.split('=>')[0]
selector_value = selector.split('=>')[1]
if selector_by is None and selector_value is None:
Logger.error('find_elements：selector_by and selector_value is empty:，Please pass the correct value)')
else:
Logger.info("find_elements：正在定位元素信息：定位方式->%s,value值->%s" % (selector_by, selector_value))
try:
time.sleep(1)
elems = WebDriverWait(self.driver, self.timeout, self.poll_frequency).until(
lambda x: x.find_elements(*selector))
Logger.info("元素对象为：{}".format(elems))
return elems
except Exception as e:
Logger.error("定位不到元素，错误信息为:{}".format(e))
return False
def choose_elem_by_visible_value(self, locator, value):
"""
根据一组元素对象中某一个元素对象的文本值确定，哪一个元素对象
@param locator:
@param value:
@return: 单个元素对象
"""
elems = self.find_elements(locator)
for item in elems:
if self.elem_object_get_text(item) == value:
return item
def elem_object_get_text(self, elem):
"""
元素对象获取text值，传入元素对象
@param elem:
@return:
"""
elem_text = elem.text
return elem_text
def get_text_by_elements(self, locator, index):
"""
@param locator: 定位器
@return: 返回定位对象组的第一个元素的值
"""
global elem_text
elem = self.find_elements(locator)
try:
elem_text = elem[index].text
Logger.info("获取元素组对象，索引位置{}的值成功，值为：{}".format(index, elem_text))
except Exception as e:
Logger.error("获取元素组对象，索引位置{}的值失败，失败信息为: {}".format(e, e))
return elem_text
'''--静态方法--'''
@staticmethod
def sleep(seconds):
time.sleep(seconds)
Logger.info("Sleep for %d seconds" % seconds)
