base_Page:
import logging
import os
import time
from datetime import datetime
from time import sleep
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.wait import WebDriverWait
from selenium.common.exceptions import TimeoutException, NoSuchElementException
from utils.MyLog import MyLog
"""
此类封装所有操作,所有页面继承该类
"""
class BasePage(object):
def __init__(self, driver):
self.logger = MyLog().getLog()
self.driver = driver
# 等待元素可见
def wait_eleVisible(self, loc, timeout=30, poll_frequency=0.5, model=None):
"""
:param loc:元素定位表达;元组类型,表达方式(元素定位类型,元素定位方法)
:param timeout:等待的上限
:param poll_frequency:轮询频率
:param model:等待失败时,截图操作,图片文件中需要表达的功能标注
:return:None
"""
self.logger.info(f'等待"{model}"元素,定位方式:{loc}')
try:
start = datetime.now()
WebDriverWait(self.driver, timeout, poll_frequency).until(EC.visibility_of_element_located(loc))
end = datetime.now()
self.logger.info(f'等待"{model}"时长:{end - start}')
except TimeoutException:
self.logger.exception(f'等待"{model}"元素失败,定位方式:{loc}')
# 截图
self.save_webImgs(f"等待元素[{model}]出现异常")
raise
# 等待元素不可见
def wait_eleNoVisible(self, loc, timeout=30, poll_frequency=0.5, model=None):
"""
:param loc:元素定位表达;元组类型,表达方式(元素定位类型,元素定位方法)
:param timeout:等待的上限
:param poll_frequency:轮询频率
:param model:等待失败时,截图操作,图片文件中需要表达的功能标注
:return:None
"""
logging.info(f'等待"{model}"消失,元素定位:{loc}')
try:
start = datetime.now()
WebDriverWait(self.driver, timeout, poll_frequency).until_not(EC.visibility_of_element_located(loc))
end = datetime.now()
self.logger.info(f'等待"{model}"时长:{end - start}')
except TimeoutException:
self.logger.exception(f'等待"{model}"元素失败,定位方式:{loc}')
# 截图
self.save_webImgs(f"等待元素[{model}]消失异常")
raise
# 查找一个元素element
def find_element(self, loc, model=None):
self.logger.info(f'查找"{model}"元素,元素定位:{loc}')
try:
return self.driver.find_element(*loc)
except NoSuchElementException:
self.logger.exception(f'查找"{model}"元素失败,定位方式:{loc}')
# 截图
self.save_webImgs(f"查找元素[{model}]异常")
raise
# 查找元素elements
def find_elements(self, loc, model=None):
self.logger.info(f'查找"{model}"元素集,元素定位:{loc}')
try:
return self.driver.find_elements(*loc)
except NoSuchElementException:
self.logger.exception(f'查找"{model}"元素集失败,定位方式:{loc}')
# 截图
self.save_webImgs(f"查找元素集[{model}]异常")
raise
# 输入操作
def input_text(self, loc, text, model=None):
# 查找元素
ele = self.find_element(loc, model)
# 输入操作
self.logger.info(f'在"{model}"输入"{text}",元素定位:{loc}')
try:
ele.send_keys(text)
except:
self.logger.exception(f'"{model}"输入操作失败!')
# 截图
self.save_webImgs(f"[{model}]输入异常")
raise
# 清除操作
def clean_inputText(self, loc, model=None):
ele = self.find_element(loc, model)
# 清除操作
self.logger.info(f'清除"{model}",元素定位:{loc}')
try:
ele.clear()
except:
self.logger.exception(f'"{model}"清除操作失败')
# 截图
self.save_webImgs(f"[{model}]清除异常")
raise
# 点击操作
def click_element(self, loc, model=None):
# 先查找元素在点击
ele = self.find_element(loc, model)
# 点击操作
self.logger.info(f'点击"{model}",元素定位:{loc}')
try:
ele.click()
except:
self.logger.exception(f'"{model}"点击失败')
# 截图
self.save_webImgs(f"[{model}]点击异常")
raise
# 获取文本内容
def get_text(self, loc, model=None):
# 先查找元素在获取文本内容
ele = self.find_element(loc, model)
# 获取文本
self.logger.info(f'获取"{model}"元素文本内容,元素定位:{loc}')
try:
text = ele.text
self.logger.info(f'获取"{model}"元素文本内容为"{text}",元素定位:{loc}')
return text
except:
self.logger.exception(f'获取"{model}"元素文本内容失败,元素定位:{loc}')
# 截图
self.save_webImgs(f"获取[{model}]文本内容异常")
raise
# 获取属性值
def get_element_attribute(self, loc, name, model=None):
# 先查找元素在去获取属性值
ele = self.find_element(loc, model)
# 获取元素属性值
self.logger.info(f'获取"{model}"元素属性,元素定位:{loc}')
try:
ele_attribute = ele.get_attribute(name)
self.logger.info(f'获取"{model}"元素"{name}"属性集为"{ele_attribute}",元素定位:{loc}')
return ele_attribute
except:
self.logger.exception(f'获取"{model}"元素"{name}"属性失败,元素定位:{loc}')
# 截图
self.save_webImgs(f"获取[{model}]属性异常")
raise
# iframe 切换
def switch_iframe(self, frame_refer, timeout=30, poll_frequency=0.5, model=None):
# 等待 iframe 存在
self.logger.info('iframe 切换操作:')
try:
# 切换 == index\name\id\WebElement
WebDriverWait(self.driver, timeout, poll_frequency).until(
EC.frame_to_be_available_and_switch_to_it(frame_refer))
sleep(0.5)
self.logger.info('切换成功')
except:
self.logger.exception('iframe 切换失败!!!')
# 截图
self.save_webImgs(f"iframe切换异常")
raise
# 窗口切换 = 如果是切换到新窗口,new. 如果是回到默认的窗口,default
def switch_window(self, name, cur_handles=None, timeout=20, poll_frequency=0.5, model=None):
"""
调用之前要获取window_handles
:param name: new 代表最新打开的一个窗口. default 代表第一个窗口. 其他的值表示为窗口的 handles
:param cur_handles:
:param timeout:等待的上限
:param poll_frequency:轮询频率
:param model:等待失败时,截图操作,图片文件中需要表达的功能标注
:return:
"""
try:
if name == 'new':
if cur_handles is not None:
self.logger.info('切换到最新打开的窗口')
WebDriverWait(self.driver, timeout, poll_frequency).until(EC.new_window_is_opened(cur_handles))
window_handles = self.driver.window_handles
self.driver.swich_to.window(window_handles[-1])
else:
self.logger.exception('切换失败,没有要切换窗口的信息!!!')
self.save_webImgs("切换失败_没有要切换窗口的信息")
raise
elif name == 'default':
self.logger.info('切换到默认页面')
self.driver.switch_to.default()
else:
self.logger.info('切换到为 handles 的窗口')
self.driver.swich_to.window(name)
except:
self.logger.exception('切换窗口失败!!!')
# 截图
self.save_webImgs("切换失败_没有要切换窗口的信息")
raise
# 截图
def save_webImgs(self, model=None):
# filepath = 指图片保存目录/model(页面功能名称)_当前时间到秒.png
# 截图保存目录
# 拼接日志文件夹,如果不存在则自动创建
cur_path = os.path.dirname(os.path.realpath(__file__))
now_date = time.strftime('%Y-%m-%d', time.localtime(time.time()))
screenshot_path = os.path.join(os.path.dirname(cur_path), f'Screenshots\\{now_date}')
if not os.path.exists(screenshot_path):
os.mkdir(screenshot_path)
# 当前时间
dateNow = time.strftime('%Y%m%d_%H%M%S', time.localtime(time.time()))
# 路径
filePath = '{}\\{}_{}.png'.format(screenshot_path, model, dateNow)
try:
self.driver.save_screenshot(filePath)
self.logger.info(f"截屏成功,图片路径为{filePath}")
except:
self.logger.exception('截屏失败!')
# 退出
def get_driver(self):
return self.driver
日志说明:
import logging
import os
import time
class MyLog(object):
def __init__(self, logger=None):
"""
指定保存日志的文件路径,日志级别,以及调用文件
将日志存入到指定的文件中
"""
# cur_path变量获得当前文件的绝对路径。
cur_path = os.path.dirname(os.path.realpath(__file__))
# log_path变量使用os.path.join()函数将文件夹名Logs与上层文件夹路径拼接而成,如果该路径不存在,则使用os.mkdir()函数创建一个新的文件夹。
log_path = os.path.join(os.path.dirname(cur_path), f'Logs')
if not os.path.exists(log_path):
os.mkdir(log_path)
# now_date变量获得当前日期,格式为YYYY-MM-DD。phone_log_path变量使用os.path.join()函数将文件夹名与当前日期拼接而成,如果该路径不存在,则使用os.mkdir()函数创建一个新的文件夹。
now_date = time.strftime('%Y-%m-%d', time.localtime(time.time()))
phone_log_path = os.path.join(os.path.dirname(cur_path), f'Logs\\{now_date}')
if not os.path.exists(phone_log_path):
os.mkdir(phone_log_path)
# 创建一个名为logger的日志记录器,并将日志级别设置为INFO。
self.logger = logging.getLogger(logger)
self.logger.setLevel(logging.INFO)
# 创建一个文件处理器(FileHandler),用于将日志写入到文件中。now_time变量获得当前时间,格式为YYYYMMDD-HHMMSS。log_name变量使用os.path.join()函数将日志文件的完整路径与文件名拼接而成。文件处理器的日志级别被设置为INFO。
now_time = time.strftime('%Y%m%d-%H%M%S', time.localtime(time.time()))
log_name = os.path.join(phone_log_path, f'{now_time}.log')
fh = logging.FileHandler(log_name, encoding="utf-8")
fh.setLevel(logging.INFO)
# 创建一个流处理器(StreamHandler),用于将日志输出到控制台。流处理器的日志级别被设置为INFO。
ch = logging.StreamHandler()
ch.setLevel(logging.INFO)
# 定义了日志处理器的输出格式,该格式包括时间、日志级别、文件名、行号和具体日志消息等。
formatter = logging.Formatter('%(asctime)s - %(levelname)s %(filename)s [line:%(lineno)d]: %(message)s')
fh.setFormatter(formatter)
ch.setFormatter(formatter)
# 将文件处理器和流处理器添加到日志记录器中。
self.logger.addHandler(fh)
self.logger.addHandler(ch)
def getLog(self):
return self.logger
dao.py
class DBHelper:
def __init__(self, conn):
self.conn = conn
def query(self, sql):
"""
查询数据
:param sql: sql语句
:return: 查询出信息
"""
cursor = self.conn.cursor()
cursor.execute(sql)
result = cursor.fetchall()
cursor.close()
return result
def insert(self, sql):
"""
插入数据
:param sql: 插入数据sql
:return:
"""
cursor = self.conn.cursor()
cursor.execute(sql)
self.conn.commit()
cursor.close()
def update(self, sql):
"""
:param sql: 修改数据SQL
:return:
"""
cursor = self.conn.cursor()
cursor.execute(sql)
self.conn.commit()
cursor.close()
def delete(self, sql):
"""
:param sql: 删除SQL
:return:
"""
cursor = self.conn.cursor()
cursor.execute(sql)
self.conn.commit()
cursor.close()