本文用于记录自己二次封装的Selenium中常用的一些页面方法。
import time
from selenium.common.exceptions import NoSuchElementException
import logging
class Basepage(object):
url = None
''''
定义一个基类,封装常用的页面方法
'''
def __init__(self,driver):
self.driver = driver
#关闭浏览器
def quit_browser(self):
self.driver.quit()
#浏览器前进
def forward(self):
self.driver.forward()
#浏览器后退
def back(self):
self.driver.back()
# 隐式等待
def wait(self,seconds):
self.driver.implicitly_wait(seconds)
# 关闭当前窗口
def close(self):
try:
self.driver.close()
except NameError as e:
print('Failed to close')
# 保存图片
def get_windows_img(self):
self.logger = logging.getLogger(__name__)
file_path = 'D:/python/pom_adv/report/screenshots/'
rq = time.strftime('%Y%m%d%H%M%S', time.localtime(time.time()))
screen_name = file_path + rq + '.png'
print(screen_name)
try:
self.driver.get_screenshot_as_file(screen_name)
self.logger.info("Had take screenshot and save to folder : /screenshots")
except NameError as e:
self.logger.error("Failed to take screenshot! %s" % e)
self.get_windows_img()
# 定位元素方法
def find_element(self,selector):
self.logger = logging.getLogger(__name__)
element = ''
if '=>' not in selector:
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)
self.logger.info("Had find the element \' %s \' successful "
"by %s via value: %s " % (element.text, selector_by, selector_value))
except NoSuchElementException as e:
self.logger.error("NoSuchElementException: %s" % e)
self.get_windows_img() # take screenshot
elif selector_by == "n" or selector_by == 'name':
element = self.driver.find_element_by_name(selector_value)
elif selector_by == "c" or selector_by == 'class_name':
element = self.driver.find_element_by_class_name(selector_value)
elif selector_by == "l" or selector_by == 'link_text':
element = self.driver.find_element_by_link_text(selector_value)
elif selector_by == "p" or selector_by == 'partial_link_text':
element = self.driver.find_element_by_partial_link_text(selector_value)
elif selector_by == "t" or selector_by == 'tag_name':
element = self.driver.find_element_by_tag_name(selector_value)
elif selector_by == "x" or selector_by == 'xpath':
try:
element = self.driver.find_element_by_xpath(selector_value)
self.logger.info("Had find the element \' %s \' successful "
"by %s via value: %s " % (element.text, selector_by, selector_value))
except NoSuchElementException as e:
self.logger.error("NoSuchElementException: %s" % e)
self.get_windows_img()
elif selector_by == "s" or selector_by == 'css_selector':
element = self.driver.find_element_by_css_selector(selector_value)
else:
raise NameError("Please enter a valid type of targeting elements.")
return element
# 文本框输入
def set_value(self,element,text):
self.logger = logging.getLogger(__name__)
element.clear()
try:
element.send_keys(text)
self.logger.info("Had type \' %s \' in inputBox" % text)
except NameError as e:
self.logger.error("Failed to type in input box with %s" % e)
self.get_windows_img()
# 文本框清除
def clear(self,selector):
self.logger = logging.getLogger(__name__)
el = self.find_element(selector)
try:
el.clear()
self.logger.info("Clear text in input box before typing.")
except NameError as e:
self.logger.error("Failed to clear in input box with %s" % e)
self.get_windows_img()
# 点击元素
def click_sel(self, selector):
self.logger = logging.getLogger(__name__)
el = self.find_element(selector)
try:
self.logger.info("The element \' %s \' was clicked." % el.text)
el.click()
except NameError as e:
self.logger.error("Failed to click the element with %s" % e)
def click_ele(self,element):
self.logger = logging.getLogger(__name__)
el = element
try:
self.logger.info("The element \' %s \' was clicked." % el.text)
el.click()
except NameError as e:
self.logger.error("Failed to click the element with %s" % e)
# 或者网页标题
def get_page_title(self):
self.logger = logging.getLogger(__name__)
self.logger.info("Current page title is %s" % self.driver.title)
return self.driver.title
@staticmethod
def sleep(self,seconds):
self.logger = logging.getLogger(__name__)
time.sleep(seconds)
self.logger.info("Sleep for %d seconds" % seconds)
def switch_alert(self):
alert = self.driver.switch_to_alert()
alert.accept()