web自动化basepage封装

# !/user/bin/env python3
# -*- coding:utf-8 -*-

from selenium.webdriver.remote.webdriver import WebElement
from selenium.webdriver.remote.webdriver import WebDriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.action_chains import ActionChains
import logging
import time
import os
import datetime
import select
import random

from common.dir_config import screenshot_dir


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

    # 等待元素可见
    def wait_ele_visible(self, loc, img_desc, timeout=30, frequency=0.5):
        """

        :param loc: 元素定位
        :param img_desc: 图片描述名称
        :param timeout: 时间
        :param frequency:
        :return:
        """
        start = datetime.datetime.now()
        try:
            WebDriverWait(self.driver, timeout, frequency).until(EC.visibility_of_element_located(loc))
        except:
            logging.exception("等待元素{}可见，失败！".format(loc))
            self.save_web_screenshot(img_desc)
            raise
        else:
            end = datetime.datetime.now()
            logging.info("等待 {}  元素  {} 可见成功。耗时：{}".format(img_desc, loc, end - start))

    # 等待元素存在
    def wait_ele_exist(self, loc, img_desc, timeout=30, frequency=0.5):
        start = datetime.datetime.now()
        try:
            WebDriverWait(self.driver, timeout, frequency).until(EC.presence_of_element_located(loc))
        except:
            logging.exception("等待元素{}可见，失败！".format(loc))
            self.save_web_screenshot(img_desc)
            raise
        else:
            end = datetime.datetime.now()
            logging.info("等待 {}  元素  {} 可见成功。耗时：{}".format(img_desc, loc, end - start))

    # 实现网页截图操作
    def save_web_screenshot(self, img_description):
        #  页面_功能_时间.png
        now = time.strftime("%Y-%m-%d %H_%M_%S")
        filepath = "{}_{}.png".format(img_description, now)
        try:
            self.driver.save_screenshot(screenshot_dir + "/" + filepath)
            logging.info("网页截图成功。图片存储在：{}".format(screenshot_dir + "/" + filepath))
        except:
            logging.exception("网页截屏失败！")
            raise

    # 
    def del_readonly(self, ele):
        eles = self.driver.find_element(*ele)
        self.driver.execute_script('arguments[0].removeAttribute("readonly")', eles)  # 删除元素d下面 readonly属性

    # 查找元素
    def get_element(self, loc, img_desc, find_all=False):
        """

        :param loc: 元组，元素表达式
        :param img_desc: 截图命名
        :param find_all: 是否匹配全部
        :return: WebElement对象。当find_all为true，返回列表
        """
        try:
            if find_all is True:
                ele = self.driver.find_elements(*loc)
            else:
                ele = self.driver.find_element(*loc)
        except:
            logging.exception("查找  {} 元素 {} 失败！".format(img_desc, loc))
            self.save_web_screenshot(img_desc)
            raise
        else:
            return ele

    # 元素可以是通过元素定位查找，也可以是直接是webElement对象。
    def _deal_element(self, loc, img_desc, timeout=30, frequency=0.5, wait_type="visible"):
        # 先等待可见,再查找元素
        if isinstance(loc, tuple):  # 元素定位表达式类型
            if wait_type == "visible":  # 等待元素可见
                self.wait_ele_visible(loc, img_desc, timeout, frequency)
            else:  # 等待元素存在
                self.wait_ele_exist(loc, img_desc, timeout, frequency)
            return self.get_element(loc, img_desc)
        elif isinstance(loc, WebElement):  # WebElement对象
            return loc
        else:
            logging.error("参数loc: {} 即不是元组，也不是WebElement对象，无法根据此参数找到元素。".format(loc))
            raise

    # 点击元素
    def click_element(self, loc, img_desc, timeout=30, frequency=0.5):
        ele = self._deal_element(loc, img_desc, timeout, frequency)
        time.sleep(1)
        try:
            ele.click()
        except:
            logging.exception("点击  {} 元素 {} 失败！".format(img_desc, loc))
            self.save_web_screenshot(img_desc)
            raise

    # 双击元素
    def double_click_element(self, loc, img_desc, timeout=30, frequency=0.5):
        ele = self._deal_element(loc, img_desc, timeout, frequency)
        try:
            ActionChains(driver=self.driver).double_click(ele).perform()
        except:
            logging.exception("双击  {} 元素 {} 失败！".format(img_desc, loc))
            self.save_web_screenshot(img_desc)
            raise

    # 清除文本元素
    def clean_element_text(self, loc, img_desc):
        ele = self.driver.find_element(*loc)
        try:
            ele.send_keys(Keys.CONTROL, "a")
            ele.send_keys(Keys.DELETE)
        except:
            # 日志
            logging.exception("清除  {} 元素 的文本 {} 失败！".format(img_desc, loc))
            # 截图
            self.save_web_screenshot(img_desc)
            raise

    # 文本输入
    def input_text(self, loc, value, img_desc, timeout=30, frequency=0.5):
        ele = self._deal_element(loc, img_desc, timeout, frequency)
        try:
            ele.send_keys(value)
        except:
            logging.exception("在 {} 元素 {} 上输入文本值：{} 失败！".format(img_desc, loc, value))
            self.save_web_screenshot(img_desc)
            raise

    # 下拉菜单
    def select_element(self, loc, img_desc):
        # ele = self._deal_element(loc, img_desc, timeout, frequency)
        ele = self.driver.find_element(*loc)
        # 操作
        try:
            ActionChains(self.driver).move_to_element(ele).click(ele).perform()

        except:
            logging.exception("滑动到  {} 元素 {} 失败！".format(img_desc, loc))
            self.save_web_screenshot(img_desc)
            raise

    def scroll_up_down(self, img_desc):
        time.sleep(1)
        # ele = self._deal_element(loc, img_desc, timeout, frequency)
        # 操作
        try:
            # self.driver.execute_script("window.scrollTo(0, document.body.scrollHeight); ")
            self.driver.execute_script("window.scrollTo(0, 650); ")
            logging.info("滑动到  {} 元素 成功！".format(img_desc))
        except:
            logging.exception("滑动到  {} 元素 失败！".format(img_desc))
            self.save_web_screenshot(img_desc)
            raise
        time.sleep(1)

    def scroll_down(self, img_desc):
        # ele = self._deal_element(loc, img_desc, timeout, frequency)
        try:
            self.driver.execute_script("window.scrollTo(0, document.body.scrollHeight); ")
            logging.info("滑动到  {} 元素 成功！".format(img_desc))
        except:
            # 日志
            logging.exception("滑动到  {} 元素 失败！".format(img_desc))
            # 截图
            self.save_web_screenshot(img_desc)
            raise  # 抛出异常，让用例识别到异常将用例状态为失败。
        time.sleep(1)

    def scroll_mouse_to_location(self,loc):
        logging.info("正在准备滚动屏幕到目标元素{}".format(loc))
        try:
            self.driver.execute_script("arguments[0].scrollIntoView();", self.driver.find_element(*loc))
            # logging.info("滚动到目标元素{}成功".format(loc))
        except:
            self.save_web_screenshot("滚动到{}截图".format(loc))
            logging.exception("滚动到目标元素{}失败".format(loc))
            raise

    # 获取元素的属性值
    def get_element_attribute(self, loc, attr_name, img_desc, timeout=30, frequency=0.5):
        ele = self._deal_element(loc, img_desc, timeout, frequency, wait_type="precence")
        try:
            attr_value = ele.get_attribute(attr_name)
        except:
            logging.exception("获取 {} 元素 {} 的属性 {} 失败！".format(img_desc, loc, attr_name))
            self.save_web_screenshot(img_desc)
            raise
        else:
            return attr_value

    # 获取元素的文本值。
    def get_text(self, loc, img_desc, timeout=30, frequency=0.5,wait_type="visible"):
        ele = self._deal_element(loc, img_desc, timeout, frequency, wait_type)
        try:
            text = ele.text
            return text
        except:
            logging.exception("获取 {} 元素 {} 的文本失败！".format(img_desc, loc))
            self.save_web_screenshot(img_desc)
            raise

    # 获取隐藏元素的文本值。
    def get_hide_text(self, loc, img_desc, timeout=30, frequency=0.5):
        ele = self._deal_element(loc, img_desc, timeout, frequency, wait_type="precence")
        try:
            text = ele.get_attribute("textContent")
        except:
            logging.exception("获取 {} 元素 {} 的文本失败！".format(img_desc, loc))
            self.save_web_screenshot(img_desc)
            raise
        else:
            logging.info("获取 {} 元素 {} 的文本值为:{}".format(img_desc, loc, text))
        return text

    # Alert是否存在
    def alert_is_present(self, timeout=30, frequency=1):
        logging.info("正在准备判断 alert 是否出现")
        try:
            WebDriverWait(self.driver, timeout, frequency).until(EC.alert_is_present())
            logging.info("alert 弹框已经出现")
        except:
            logging.exception("alert弹框没有出现")

    # Alert处理
    def deal_alter(self, value="1"):  # driver.switch_to.alert.accept()
        logging.info("正在准备处理alert弹框")
        try:
            if value == "1":
                text = self.driver.switch_to.alert.text
                self.driver.switch_to.alert.accept()
                logging.info("alert接受成功")
                return text
            else:
                self.driver.switch_to.alert.dismiss()
                logging.info("alert 取消接收")
        except:
            logging.exception("alert弹框处理失败")
            raise

    # def get_select_options_list(self,loc):
    #     select = self.driver.find_elements(*loc)
    #     try:
    #         option_list = select.options
    #         print(option_list)
    #         index = random.randint(1,len(option_list)-1)
    #         print("随机数字为",index)
    #         # loc = '//label[text()="地址"]/following-sibling::div[1]/div[1]/div[2]/p[{}]'.format(index)
    #         # return loc
    #     except:
    #         logging.exception("获取随机元素对象失败！")
    #         raise