python中ui自动化selenium方法封装分享

# -*- coding: utf-8 -*-
# @Time : 2021年02月07日 13时58分
# @Email : Gamest@foxmail.com
# @Author : BeiLe
# @File : base_class.py
# @notice ：


from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions
from Common.path_os import record_dir
from Common.log import Log
import time
import win32gui
import win32con

log = Log()


class Base:

    # 引入驱动
    def __init__(self, drivers):
        self.driver = drivers

    # 截图
    def record(self, model_name="error"):
        RecordPath = record_dir + "/{}_{}.png".format(model_name,
                                                      time.strftime("%Y-%m-%d-""%H-%M-%S", time.localtime()))
        try:
            self.driver.get_screenshot_as_file(RecordPath)
        except Exception as e:
            log.error("截图失败：{}".format(e))

    # 等待元素 - 默认等待30秒，每0.5秒找一次
    def wait_elevisible(self, locator, timeout=30, poll_frequency=0.5, model_name="model"):
        try:
            WebDriverWait(self.driver, timeout, poll_frequency).until(
                expected_conditions.visibility_of_element_located(locator))
        except Exception as e:
            self.record(model_name)
            log.error("等待超时，未找到元素：{}".format(e))

    # 查找元素
    def get_element(self, locator, model_name="model"):
        try:
            return self.driver.find_element(*locator)
        except Exception as e:
            self.record(model_name)
            log.error("未找到元素：{}".format(e))

    # 点击元素
    def click_elemet(self, locator, model_name="model"):
        ele = self.get_element(locator, model_name)
        try:
            ele.click()
        except Exception as e:
            self.record(model_name)
            log.error("点击元素失败：{}".format(e))

    # 输入框 - 输入内容
    def input_text(self, locator, value, model_name="model"):
        ele = self.get_element(locator, model_name)
        try:
            ele.send_keys(value)
        except Exception as e:
            self.record(model_name)
            log.error("输入内容失败：{}".format(e))

    # 输入框 - 清空内容
    def clear_text(self, locator, model_name="model"):
        ele = self.get_element(locator, model_name)
        try:
            ele.clear()
        except Exception as e:
            self.record(model_name)
            log.error("清空输入框内容失败：{}".format(e))

    # 获取元素的文本内容
    def get_text(self, locator, model_name="model"):
        ele = self.get_element(locator, model_name)
        try:
            return ele.text
        except Exception as e:
            self.record(model_name)
            log.error("获取元素文本内容失败：{}".format(e))

    # 获取元素属性
    def get_ele_attribute(self, locator, attribute, model_name="model"):
        ele = self.get_element(locator, model_name)
        try:
            return ele.get_attribute(attribute)
        except Exception as e:
            self.record(model_name)
            log.error("获取元素属性失败：{}".format(e))

    # 元素存在则为True，否则为False
    def is_eleExist(self, locator, model_name="model"):
        try:
            self.wait_elevisible(locator, model_name=model_name)
            return True
        except Exception as e:
            self.record(model_name)
            log.error("元素不存在：{}".format(e))
            return False

    # 切换alert弹框
    def switch_to_alert(self, action="accept", content="请输入内容", model_name="model"):
        try:
            WebDriverWait(self.driver, timeout=30, poll_frequency=0.5).until(expected_conditions.alert_is_present())
            # 切换到alert
            alert = self.driver.switch_to.alert
            # 确定按钮
            if action == "accept":
                alert.accept()
            # 输入框
            elif action == "value":
                alert.send_keys(content)
            # 取消按钮
            else:
                alert.dismiss()
        except Exception as e:
            self.record(model_name)
            log.error("alert操作失败：{}".format(e))

    # windows窗口切换
    def switch_window(self, str_wd="", index=None, model_name="mode1"):
        try:
            time.sleep(2)
            # 获取所有窗口
            windows = self.driver.window_handles
            if str_wd == "new":
                self.driver.switch_to.window(windows[-1])
            else:
                if index is not None and 0 <= int(index) < len(windows):
                    self.driver.switch_to.window(windows[int(index)])
        except Exception as e:
            self.record(model_name)
            log.error("切换浏览器窗口失败：{}".format(e))

    # iframe切换
    def switch_to_iframe(self, locator, model_name="mode1"):
        try:
            # 切过去
            ele = self.get_element(locator)
            WebDriverWait(self.driver, timeout=30, poll_frequency=0.5).until(
                expected_conditions.frame_to_be_available_and_switch_to_it(ele))
            # 执行命令后切回来
            self.driver.switch_to.default_content()
        except Exception as e:
            self.record(model_name)
            log.error("切换iframe失败：{}".format(e))

    # 上传文件  每个系统不一样，一般是这种
    def upload(self, file, browser_type="chrome", model_name="mode1"):
        try:
            if browser_type == "chrome":
                title = "打开"
            else:
                title = ""
            # 一级窗口
            dialog = win32gui.FindWindow("#32770", title)
            # 二级窗口
            ComboBoxEx32 = win32gui.FindWindowEx(dialog, 0, "ComboBoxEx32", None)
            comboBox = win32gui.FindWindowEx(ComboBoxEx32, 0, "ComboBox", None)
            # 三级窗口 - 输入框
            edit = win32gui.FindWindowEx(comboBox, 0, 'Edit', None)
            # 四级窗口 - 打开按钮
            open_button = win32gui.FindWindowEx(dialog, 0, 'Button', "打开(&O)")
            # 输入文件路径
            win32gui.SendMessage(edit, win32con.WM_SETTEXT, None, file)
            # 上传
            win32gui.SendMessage(dialog, win32con.WM_COMMAND, 1, open_button)
        except Exception as e:
            self.record(model_name)
            log.error("上传文件失败：{}".format(e))

    # 滚动条操作
    def move_to_see(self, locator, type_ele="will", model_name="mode1"):
        try:
            if type_ele == "will":
                self.driver.execute_script("arguments[0].scrollIntoView();", locator)
                time.sleep(2)
            elif type_ele == "top":
                self.driver.execute_script("window.scrollTo(0,0)")
                time.sleep(2)
            else:
                self.driver.execute_script("var q=document.documentElement.scrollTop=10000")
                time.sleep(2)
        except Exception as e:
            self.record(model_name)
            log.error("滚动条操作失败：{}".format(e))