改进Page Object公共方法封装（base_page）
import os from time import sleep from selenium import webdriver from selenium.webdriver.common.by import By from selenium.webdriver import ActionChains from selenium.webdriver.common.keys import Keys from common.log_utils import logger from common.config_utils import ConfigUtils from selenium.webdriver.support.wait import WebDriverWait class BasePage: def __init__(self, driver): self.driver = driver def open_url(self, url): """ 浏览器操作封装 """ self.driver.get(url) logger.info('打开url地址%s' % url) def quit_browser(self): """ 关闭浏览器 """ sleep(2) self.driver.quit() logger.info("关闭浏览器") def set_browser_max(self): """ 浏览器最大化 """ self.driver.maximize_window() logger.info("设置浏览器最大化") def set_browser_min(self): """ 浏览器最小化 """ self.driver.minimize_window() logger.info("设置浏览器最小化") def refresh(self): """ 刷新浏览器 """ self.driver.refresh() logger.info("浏览器刷新操作") def get_title(self): """ 获取网页标题 """ value = self.driver.title logger.info("获取网页标题，标题是%s" % value) return value def set_driver(self): self.set_browser_max() self.open_url(ConfigUtils().test_url) def find_element(self, element_info): """ 元素定位方法 """ locator_element_name = element_info['element_name'] 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 == 'name': locator_type = By.NAME elif locator_type_name == 'xpath': locator_type = By.XPATH elif locator_type_name =='link_text': locator_type = By.LINK_TEXT elif locator_type_name == 'css_selector': locator_type = By.CSS_SELECTOR element = WebDriverWait(self.driver, int(locator_timeout))\ .until(lambda x: x.find_element(locator_type, locator_value_info)) logger.info("%s元素识别成功" % locator_element_name) # element = WebDriverWait(self.driver, locator_timeout)\ # .until(EC.presence_of_element_located(locator_type, locator_value_info)) return element def click(self, element_info): """ 元素点击操作 """ self.find_element(element_info).click() logger.info('%s点击操作成功' % element_info['element_name']) def input(self, element_info, content): """ 元素输入操作 """ self.find_element(element_info).send_keys(content) logger.info("%s 输入内容【%s】" % (element_info['element_name'], content)) def target_locator(self): """ 下拉至底部 """ sleep(2) js = "var q=document.documentElement.scrollTop=10000" self.driver.execute_script(js) logger.info("下拉至底部") # 表单切换 def frame_switch(self, frame_name=None): """ 表单切换 """ if frame_name is None: print("Please enter the form element to jump to") logger.info("没有输入页面元素") else: frame_name = self.find_element(frame_name) self.driver.switch_to.frame(frame_name) sleep(2) logger.info("跳转至%s表单" % frame_name['element_name']) def frame_default_content(self): """ 切回初始表单 """ self.driver.switch_to.default_content() sleep(2) # 句柄切换 def get_current_handle(self): """ 获取当前句柄 """ current_windows = self.driver.current_window_handle title = self.driver.title logger.info('获取%s句柄' % title) return str(current_windows) def go_other_handle(self, current_handles): """ 前往别的表单 """ for handle in self.driver.window_handles: if handle != current_handles: self.driver.switch_to.window(handle) title = self.driver.title logger.info('前往%s' % title) def go_beginning_handle(self, current_handle): """ 返回初始表单 """ self.driver.switch_to.window(current_handle) title = self.driver.title logger.info('返回%s' % title) # 警告框 def get_alert(self): alert = self.driver.switch_to.alert sleep(2) alert_text = alert.text alert.accept() logger.info('接受警告框，警告框内容为%s' % alert_text) # 鼠标操作 def mouse_operation(self, element_info, mouse_operate): """ 鼠标操作 """ above = self.find_element(element_info) if mouse_operate == 'context_click': ActionChains(self.driver).context_click(above).perform() elif mouse_operate == 'double_click': ActionChains(self.driver).double_click(above).perform() elif mouse_operate == 'move_to_element': ActionChains(self.driver).move_to_element(above).perform() logger.info("正在进行%s" % mouse_operate) def drag_element(self, element_info, element_info2): """ 鼠标拖动操作 """ drag = self.find_element(element_info) drop = self.find_element(element_info2) ActionChains(self.driver).drag_and_drop(drag, drop).perform() logger.info("正在将%s拖动至%s" % (element_info['element_name'], element_info2['element_name'])) # 键盘操作 def key_enter(self, element_info): """ 回车 """ self.find_element(element_info).send_keys(Keys.ENTER) logger.info("对%s进行回车" % element_info['element_name']) def key_ctrl_a(self, element_info): """ 全选 """ self.find_element(element_info).send_keys(Keys.CONTROL, 'a') logger.info("对%s全选" % element_info['element_name']) def key_ctrl_c(self, element_info): """ 复制 """ self.find_element(element_info).send_keys(Keys.CONTROL, 'c') logger.info("对%s复制" % element_info['element_name']) def key_ctrl_v(self, element_info): """ 粘贴 """ self.find_element(element_info).send_keys(Keys.CONTROL, 'v') logger.info("对%s粘贴" % element_info['element_name']) def key_ctrl_x(self, element_info): """ 剪切 """ self.find_element(element_info).send_keys(Keys.CONTROL, 'x') logger.info("对%s剪切" % element_info['element_name'])
创建 BasePage类作为所有 Page 类的基类，在 BasePage 类中封装一些方法，这些方法是我们在做自动化时经常用到的。
- open_url()方法用于打开网页，它接收一个 url 参数；
- quit_browser()方法是当我们执行完测试用例后，可以使用这个方法进行浏览器关闭；
- set_browser_max()设置浏览器最大化
- set_browser_min()设置浏览器最小化
- refresh()刷新浏览器
...
当然，后续可能还会更加优化这个类，使功能越来越强大
logger.info()是日志封装后的方法；
import os import logging current_path = os.path.dirname(__file__) # 获取文件当前路径 log_path = os.path.join(current_path, '../logs/log.txt') class LogUtils: def __init__(self, log_file_path=log_path): self.log_file_path = log_file_path self.logger = logging.getLogger(__name__) self.logger.setLevel(level=logging.INFO) console = logging.StreamHandler() formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s") console.setFormatter(formatter) file_log = logging.FileHandler(log_path) file_log.setFormatter(formatter) self.logger.addHandler(console) self.logger.addHandler(file_log) def info(self, message): self.logger.info(message) def error(self, message): self.logger.error(message) logger = LogUtils() if __name__ == '__main__': # log_utils = LogUtils() logger.info('hello, word')
