python selenium基于显示等待封装的一些常用方法

import os
import time
from PIL import Image
from selenium import webdriver
from appium import webdriver as app
from selenium.common.exceptions import *
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.select import Select
from selenium.webdriver.support.wait import WebDriverWait
from common.logger import Log
from common import read_config


def open_browser(browser='chrome'):
    """打开浏览器函数。"firefox"、"chrome"、"ie",'phantomjs'"""

    # 驱动路径
    driver_path = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
    try:
        if browser == 'firefox':
            executable_path = os.path.join(driver_path, 'driver\geckodriver.exe')
            # executable_path = os.path.join(driver_path, 'driver/geckodriver')
            driver = webdriver.Firefox(executable_path=executable_path)
            return driver
        elif browser == 'chrome':
            # 加启动配置,忽略 Chrome正在受到自动软件的控制 提示
            option = webdriver.ChromeOptions()
            option.add_argument('disable-infobars')
            # chrome启动静默模式;默认显示浏览器界面
            if read_config.chrome_interface != 'True':
                option.add_argument('headless')
            executable_path = os.path.join(driver_path, 'driver\chromedriver.exe')
            # executable_path = os.path.join(driver_path, 'driver/chromedriver')
            driver = webdriver.Chrome(chrome_options=option, executable_path=executable_path)
            return driver
        elif browser == 'ie':
            driver = webdriver.Ie()
            return driver
        elif browser == 'js':
            driver = webdriver.PhantomJS()
            return driver
        else:
            Log().warning('额，暂不支持此浏览器诶。先试试firefox、chrome、ie、phantomJS浏览器吧。')
            return
    except Exception as msg:
        Log().error('浏览器出错了呀！%s' % msg)
        return


def open_app():
    try:
        desired_caps = {
            'platformName': read_config.platform_name,

            'deviceName': read_config.device_name,

            'platformVersion': read_config.platform_version,

            'appPackage': read_config.app_package,

            'appActivity': read_config.app_activity,

            'noReset': True,

            # 隐藏手机默认键盘
            'unicodeKeyboard': True,

            'resetKeyboard': True
        }
        # 关联appium
        driver = app.Remote('http://127.0.0.1:4723/wd/hub', desired_caps)
        return driver
    except Exception as e:
        raise Exception('连接 Appium 出错：{}'.format(e))


class Crazy:
    """基于原生的selenium框架做二次封装"""

    def __init__(self, driver):
        """启动浏览器参数化，默认启动chrome"""
        self.driver = driver
        self.action = ActionChains(self.driver)
        self.timeout = 5  # 显示等待超时时间
        self.t = 1
        self.log = Log()

    def open(self, url, t=''):
        """get url，最大化浏览器，判断title"""
        self.driver.get(url)
        self.driver.implicitly_wait(10)
        # 是否最大化浏览器
        if read_config.maximize != 'True':
            self.driver.maximize_window()
        try:
            WebDriverWait(self.driver, self.timeout, self.t).until(EC.title_contains(t))
            self.log.info('打开网页成功！')
        except TimeoutException:
            self.log.error('打开%s title错误，超时' % url)
        except Exception as msg:
            self.log.error('打开网页产生的其他错误：%s' % msg)

    def find_element(self, locator):
        """重写元素定位方法"""
        if not isinstance(locator, tuple):
            self.log.error('locator参数必须是元组类型，而不是：{}'.format(type(locator)))
            return ""
        else:
            try:
                element = WebDriverWait(self.driver, self.timeout, self.t).until(
                    EC.presence_of_element_located(locator))
                if element.is_displayed():
                    return element
            except:
                self.log.info('%s页面中未能找到元素%s' % (self, locator))
                return ""

    def find_elements(self, locator):
        """定位一组元素"""
        if not isinstance(locator, tuple):
            self.log.error('locator参数必须是元组类型，而不是：{}'.format(type(locator)))
            return ""
        else:
            try:
                elements = WebDriverWait(self.driver, self.timeout, self.t).until(
                    EC.presence_of_all_elements_located(locator))
                return elements
            except:
                self.log.info('%s页面中未能找到元素%s' % (self, locator))
                return ""

    def click_coordinate(self, coordinate, timeout=10):
        """点击坐标"""
        self.driver.tap(coordinate, timeout)

    def clicks(self, locator, n):
        """点击一组元组中的一个"""
        element = self.find_elements(locator)[n]
        element.click()

    def click(self, locator):
        """点击操作"""
        element = self.find_element(locator)
        element.click()

    def double_click(self, locator):
        """双击操作"""
        element = self.find_element(locator)
        self.action.double_click(element).perform()

    def send_keys(self, locator, text):
        """发送文本，清空后输入"""
        element = self.find_element(locator)
        element.clear()
        element.send_keys(text)

    def sends_keys(self, locator, n, text):
        """选中一组元素中的一个，发送文本，清空后输入"""
        element = self.find_elements(locator)[n]
        element.clear()
        element.send_keys(text)

    def send_keys_enter(self):
        """敲enter"""
        self.action.send_keys(Keys.ENTER).perform()

    def send_keys_down(self):
        """敲向下键"""
        self.action.send_keys(Keys.DOWN).perform()

    def send_keys_arrow_down(self):
        self.action.send_keys(Keys.ARROW_DOWN).perform()

    def send_keys_arrow_right(self):
        self.action.send_keys(Keys.ARROW_RIGHT).perform()

    def is_text_in_element(self, locator, text):
        """判断文本在元素里，没定位到元素返回False，定位到返回判断结果布尔值"""
        try:
            result = WebDriverWait(self.driver, self.timeout, self.t).until(
                EC.text_to_be_present_in_element(locator, text))
            self.log.info('is_text_in_element     成功')
            return result
        except TimeoutException:
            self.log.error('%s元素没有定位到' % str(locator))
            return False

    def is_text_in_value(self, locator, value):
        """判断元素的value值，没有定位到返回False，定位到返回判断结果布尔值"""
        try:
            result = WebDriverWait(self.driver, self.timeout, self.t).until(
                EC.text_to_be_present_in_element_value(locator, value))
        except TimeoutException:
            self.log.error('元素没有定位到：%s' % str(locator))
            return False
        else:
            return result

    def is_title(self, title):
        """判断title完全等于"""
        result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.title_is(title))
        return result

    def is_title_contains(self, title):
        """判断title包含"""
        result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.title_contains(title))
        return result

    def is_selected(self, locator):
        """判断元素被选中，返回布尔值， 一般用在下拉框"""
        result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.element_located_to_be_selected(locator))
        return result

    def is_selected_be(self, locator, selected=True):
        """判断元素的状态，selected是期望的参数True/False，返回布尔值"""
        result = WebDriverWait(self.driver, self.timeout, self.t).until(
            EC.element_located_selection_state_to_be(locator, selected))
        return result

    def is_alert_present(self):
        """判断页面是否有alert，有返回alert，没有返回False"""
        result = WebDriverWait(self.driver, self.timeout, self.t).until((EC.alert_is_present()))
        text = EC.alert_is_present()(self.driver)
        if text:
            self.log.info('alert弹框显示文本是：%s' % text.text)
        else:
            self.log.warning('没有发现alert弹框。')
        return result

    def is_visibility(self, locator):
        """元素可见返回本身，不可见返回False"""
        result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.visibility_of_element_located(locator))
        return result

    def is_invisibility(self, locator):
        """元素可见返回本身，不可见返回True，没有找到元素也返回True"""
        result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.invisibility_of_element_located(locator))
        return result

    def is_clickAble(self, locator):
        """元素可以点击返回本身，不可点击返回False"""
        result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.element_to_be_clickable(locator))
        return result

    def is_locator(self, locator):
        """判断元素有没有被定位到（并不意味着可见），定位到返回element，没有定位到返回False"""
        result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.presence_of_element_located(locator))
        return result

    def move_to_element(self, locator):
        """鼠标悬停操作"""
        element = self.find_element(locator)
        ActionChains(self.driver).move_to_element(element).perform()

    def move(self, locator, locator1):
        """循环调用鼠标事件，死循环"""
        self.move_to_element(locator)
        time.sleep(2)
        element = self.find_element(locator1)
        self.move_to_element(locator1)
        try:
            if element.is_displayed:
                self.click(locator1)
            else:
                self.move(locator, locator1)
        except ElementNotVisibleException as e:
            self.log.error('鼠标点击事件失败：%s' % e)

    def drag_and_drop(self, element, element1):
        """拖拽"""
        ActionChains(self.driver).drag_and_drop(element, element1).perform()
        ActionChains(self.driver).click_and_hold(element).release(element1).perform()

    def switch_frame(self, frame):
        """切换ifarm"""
        try:
            self.driver.switch_to_frame(self.find_element(frame))
            self.log.info('切换iframe成功！')
        except:
            self.log.warning('没有发现iframe元素%s' % frame)

    def current_window_handle(self):
        """浏览器handle"""
        return self.driver.current_window_handle

    def switch_window_handle(self, n):
        """切换handle"""
        if not isinstance(n, int):
            self.driver.switch_to.window(n)
        else:
            all_handle = self.driver.window_handles
            self.driver.switch_to.window(all_handle[n])

    def back(self):
        """返回之前的网页"""
        self.driver.back()

    def forward(self):
        """前往下一个网页"""
        self.driver.forward()

    def close(self):
        """关闭当前网页"""
        self.driver.close()

    def quit(self):
        """关闭所有网页"""
        self.driver.quit()

    def get_title(self):
        """获取title"""
        return self.driver.title

    def get_texts(self, locator, n):
        """获取一组相同元素中的指定文本"""
        element = self.find_elements(locator)[n]
        if element:
            return element.text
        else:
            return None

    def get_text(self, locator):
        """获取文本"""
        element = self.find_element(locator)
        if element:
            return element.text
        else:
            return None

    def get_attribute(self, locator, name):
        """获取属性"""
        element = self.find_element(locator)
        if element:
            return element.get_attribute(name)

    def js_execute(self, js):
        """执行js"""
        return self.driver.execute_script(js)

    def js_focus_element(self, locator):
        """聚焦元素"""
        target = self.find_element(locator)
        self.driver.execute_script("arguments[0].scrollIntoView();", target)

    def js_scroll_top(self):
        """滚动到顶部"""
        js = "var q=document.documentElement.scrollTop=0"
        self.driver.execute_script(js)

    def js_scroll_bottom(self):
        """滚动到底部"""
        js = "var q=document.documentElement.scrollTop=10000"
        self.driver.execute_script(js)

    def select_by_index(self, locator, index):
        """通过索引，index是第几个，从0开始, 下拉框"""
        element = self.find_element(locator)
        Select(element).select_by_index(index)

    def select_by_value(self, locator, value):
        """通过value属性"""
        element = self.find_element(locator)
        Select(element).select_by_value(value)

    def select_by_text(self, locator, text):
        """通过text属性"""
        element = self.find_element(locator)
        Select(element).select_by_visible_text(text)

    def save_screenshot(self, img_path):
        """获取电脑屏幕截屏"""
        self.driver.save_screenshot(img_path)

    def save_report_html(self):
        """可以在html报告中使用的截图"""
        self.driver.get_screenshot_as_base64()

    def save_element_img(self, locator, img_path):
        """获取元素截图"""
        self.driver.save_screenshot(img_path)
        element = self.find_element(locator)
        left = element.location['x']
        top = element.location['y']
        right = element.location['x'] + element.size['width']
        bottom = element.location['y'] + element.size['height']
        im = Image.open(img_path)
        im = im.crop((left, top, right, bottom))
        im.save(img_path)

    def get_cookies(self):
        """获取cookies"""
        return self.driver.get_cookies()

    def swipeDown(self, t=500, n=1):
        '''向下滑动屏幕'''
        l = self.driver.get_window_size()
        x1 = l['width'] * 0.5  # x坐标
        y1 = l['height'] * 0.25  # 起始y坐标
        y2 = l['height'] * 0.75  # 终点y坐标
        for i in range(n):
            self.driver.swipe(x1, y1, x1, y2, t)

    def swipeUp(self, t=500, n=1):
        '''向上滑动屏幕'''
        l = self.driver.get_window_size()
        x1 = l['width'] * 0.5  # x坐标
        y1 = l['height'] * 0.65  # 起始y坐标
        y2 = l['height'] * 0.25  # 终点y坐标
        for i in range(n):
            self.driver.swipe(x1, y1, x1, y2, t)


if __name__ == '__main__':
    driver = open_browser()
    driver.get('file:///D:/UIAutomation/report/2019-01-15%2017-40-10report.html')