python 页面基类 Page封装 →driver层的封装（最底层的封装Page）

# coding=utf-8
import time
from selenium.common.exceptions import NoSuchElementException
from selenium.webdriver import ActionChains
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.select import Select
from selenium.webdriver.support.wait import WebDriverWait

from Common.LoggerBase import Logger                    # 自己封装的logger类
from pathlib import *
'''--create a logger instance--'''
Logger = Logger()

class BasePage(object):
    """
        定义一个页面基类，让所有页面都继承这个类，封装一些常用的页面操作方法到这个类
    """

    '''--初始化--'''
    def __init__(self, driver):
        self.driver = driver
        self.timeout = 10
        self.poll_frequency = 0.5

    '''--定义open方法--'''
    def open(self, url):
        self.driver.maximize_window()
        self.driver.get(url)

    '''--quit browser and end testing--'''
    def quit_browser(self):
        self.driver.quit()

    '''--浏览器前进操作--'''
    def forward(self):
        self.driver.forward()
        Logger.info("Click forward on current page.")

    '''--浏览器后退操作--'''
    def back(self):
        self.driver.back()
        Logger.info("Click back on current page.")

    '''--隐式等待--'''
    def wait(self, seconds):
        self.driver.implicitly_wait(seconds)
        Logger.info("wait for %d seconds." % seconds)

    '''--点击关闭当前窗口--'''
    def close(self):
        try:
            self.driver.close()
            Logger.info("Closing and quit the browser.")
        except NameError as e:
            Logger.error("Failed to quit the browser with %s" % e)

    '''--保存图片--'''
    def get_windows_img(self):
        """
            在这里我们把file_path这个参数写死，直接保存到我们项目根目录的一个文件夹.Screenshots下
        """
        file_path = Path(Path.cwd().parent) / 'Screenshots'
        rq = time.strftime('%Y%m%d%H%M', time.localtime(time.time()))
        screen_name = Path.joinpath(file_path, rq + '.png')
        try:
            self.driver.get_screenshot_as_file(str(screen_name))
            Logger.info("Had take screenshot and save to folder : /Screenshots")
        except NameError as e:
            Logger.error("Failed to take screenshot! %s" % e)
            self.get_windows_img()

    '''--重写定位元素方法--'''
    def find_element(self, selector):
        """
             这个地方为什么是根据=>来切割字符串，请看页面里定位元素的方法
             submit_btn = "id=>su"
             login_lnk = "xpath => //*[@id='u1']/a[7]"  # 百度首页登录链接定位
             如果采用等号，结果很多xpath表达式中包含一个=，这样会造成切割不准确，影响元素定位
            :param selector:
            :return: element
        """
        element = ''
        if '=>' not in selector:
            Logger.info("Not found '=>', default is ID")
            return self.driver.find_element_by_id(selector)
        selector_by = selector.split('=>')[0]
        selector_value = selector.split('=>')[1]

        if selector_by == "i" or selector_by == 'id':
            try:
                element = self.driver.find_element_by_id(selector_value)
                Logger.info("Had find the element " %s " successful "
                            "by %s via value: %s " % (element.text, selector_by, selector_value))
            except NoSuchElementException as e:
                Logger.error("NoSuchElementException: %s" % e)
                self.get_windows_img()  # take screenshot

        elif selector_by == "n" or selector_by == 'name':
            try:
                element = self.driver.find_element_by_name(selector_value)
                Logger.info("Had find the element " %s " successful "
                            "by %s via value: %s " % (element.text, selector_by, selector_value))
            except NoSuchElementException as e:
                Logger.error("NoSuchElementException: %s" % e)
                self.get_windows_img()

        elif selector_by == "c" or selector_by == 'class_name':
            try:
                element = self.driver.find_element_by_class_name(selector_value)
                Logger.info("Had find the element " %s " successful "
                            "by %s via value: %s " % (element.text, selector_by, selector_value))
            except NoSuchElementException as e:
                Logger.error("NoSuchElementException: %s" % e)

        elif selector_by == "l" or selector_by == 'link_text':
            try:
                element = self.driver.find_element_by_link_text(selector_value)
                Logger.info("Had find the element " %s " successful "
                            "by %s via value: %s " % (element.text, selector_by, selector_value))
            except NoSuchElementException as e:
                Logger.error("NoSuchElementException: %s" % e)

        elif selector_by == "p" or selector_by == 'partial_link_text':
            try:
                element = self.driver.find_element_by_partial_link_text(selector_value)
                Logger.info("Had find the element " %s " successful "
                            "by %s via value: %s " % (element.text, selector_by, selector_value))
            except NoSuchElementException as e:
                Logger.error("NoSuchElementException: %s" % e)

        elif selector_by == "t" or selector_by == 'tag_name':
            try:
                element = self.driver.find_element_by_tag_name(selector_value)
                Logger.info("Had find the element " %s " successful "
                            "by %s via value: %s " % (element.text, selector_by, selector_value))
            except NoSuchElementException as e:
                Logger.error("NoSuchElementException: %s" % e)

        elif selector_by == "x" or selector_by == 'xpath':
            try:
                element = self.driver.find_element_by_xpath(selector_value)
                Logger.info("Had find the element " %s " successful "
                            "by %s via value: %s " % (element.text, selector_by, selector_value))
            except NoSuchElementException as e:
                Logger.error("NoSuchElementException: %s" % e)
                self.get_windows_img()

        elif selector_by == "s" or selector_by == 'selector_selector':
            try:
                element = self.driver.find_element_by_css_selector(selector_value)
                Logger.info("Had find the element " %s " successful "
                            "by %s via value: %s " % (element.text, selector_by, selector_value))
            except NoSuchElementException as e:
                Logger.error("NoSuchElementException: %s" % e)

        else:
            raise NameError("Please enter a valid type of targeting elements.")

        return element

    '''--输入--'''
    def input_text(self, selector, text):
        el = self.find_element(selector)
        try:
            el.send_keys(text)
            Logger.info("Had type " %s " in inputBox" % text)
        except NameError as e:
            Logger.error("Failed to type in input box with %s" % e)
            self.get_windows_img()

    '''--清除文本框--'''
    def clear(self, selector):
        el = self.find_element(selector)
        try:
            el.clear()
            Logger.info('Clear text in input box before typing')
        except NameError as e:
            Logger.error("Failed to clear in input box with %s" % e)
            self.get_windows_img()

    '''--点击元素--'''
    def click(self, selector):
        el = self.find_element(selector)
        try:
            el.click()
            Logger.info("The element " %s " was clicked." % el)
        except NameError as e:
            Logger.error("Failed to click the element with %s" % e)

    '''--获取网页标题--'''
    def get_page_title(self):
        Logger.info("Current page title is %s" % self.driver.title)
        return self.driver.title

    def send_key(self, locator, value):
        """

        @param locator: 定位器
        @param value: value
        """
        elem = self.find_element(locator)
        try:
            elem.send_keys(value)
            Logger.info("元素对象输入值成功，值为：{}".format(value))
        except Exception as e:
            Logger.error("元素对象输入值失败，错误信息为：{}".format(e))

    def get_text(self, locator):
        """
        @param locator:定位器
        @return:元素文本值
        """
        elem_text = None
        elem = self.find_element(locator)
        try:
            elem_text = elem.text
        except Exception as e:
            Logger.error("元素text获取失败，错误信息为：{}".format(e))
            Logger.info("元素text值：{}".format(elem_text))
        return elem_text

    '''--重写跳转switch_to_frame方法'''
    def switch_to_frame(self, selector):
        """
        :param selector: 传入定位器参数
        :return:
        """
        elem = self.find_element(selector)
        try:
            self.driver.switch_to.frame()
            Logger.info("frame切换成功")
        except Exception as e:
            Logger.error("frame切换失败，错误信息为：{}".format(e))

    '''--切回父级frame--'''
    def turn_back_frame(self):
        self.driver.switch_to.default_content()

    def switch_to_handle(self, index):
        """
            切换窗口句柄
        """
        # 获取当前所有窗口句柄
        global handles
        try:
            handles = self.driver.window_handles
            Logger.info("获取当前所有窗口句柄成功，句柄对象列表为：{}".format(handles))
        except Exception as e:
            Logger.error("获取当前所有窗口句柄失败，错误信息为：{}".format(e))
        # 切换到新窗口句柄
        try:
            self.driver.switch_to.window(handles[index])
            Logger.info("切换新窗口句柄成功，切换窗口的索引index为：{}".format(index))
        except Exception as e:
            Logger.error("切换新窗口句柄失败，错误信息为：{}".format(e))

    """--多选、单选元素获取方法--"""
    def select_by_text(self, selectByText):
        if selectByText is not "":
            select_by_text = self.find_element("xpath=>//*[text()="%s"]" % selectByText)
            select_by_text.click()
        else:
            print("Not find the selectByText, get the default value!")
            pass

    def move_mouse_to_element(self, locator):
        """
        移动鼠标到某个元素上面
        @param locator:
        @return:
        """
        elem = self.find_element(locator)
        action = ActionChains(self.driver)
        action.move_to_element(elem).perform()

    def clear_input_value(self, locator):
        """
        清除输入框中的内容
        @param locator:
        @return:
        """
        elem = self.find_element(locator)
        elem.send_keys(Keys.CONTROL, "a")
        elem.send_keys(Keys.DELETE)

    def get_value(self, locator):
        """
        获取输入框的value
        @param locator:
        @return:
        """
        elem = self.find_element(locator)
        return elem.get_attribute("value")

    def double_click_elem(self, locator):
        """
        双击元素
        @param locator:
        @return:
        """
        elem = self.find_element(locator)
        ActionChains(self.driver).double_click(elem).perform()

    def elem_is_display(self, locator):
        """
        判断元素在页面是否显示，显示返回True,不显示返回false
        @param locator:
        @return:
        """
        elem = self.find_element(locator)
        return elem.is_displayed()

    def elem_is_selected(self, locator):
        """
        判断元素是否被选中，用于多选框，如果多选框被选中状态，返回True，否则返回False
        @param locator:
        @return:
        """
        elem = self.find_element(locator)
        return elem.is_selected()

    def elem_is_enable(self, locator):
        """
        判断页面元素是否可用
        @param locator:
        @return:
        """
        elem = self.find_element(locator)
        Logger.info("按钮的点击状态，是否可点击：{}".format(elem.is_enabled()))
        return elem.is_enabled()

    def choose_select_by_value(self, locator, value):
        """
        根据内置属性value值,选择下拉输入框
        @param locator:
        @param value:
        @return:
        """
        elem = self.find_element(locator)
        Select(elem).select_by_value(value)

    def choose_select_by_index(self, locator, index):
        """
        根据索引选择下拉框
        @param locator:
        @param index:
        @return:
        """
        elem = self.find_element(locator)
        Select(elem).select_by_index(index)

    def choose_select_by_visible_value(self, locator, value):
        """
        根据下拉选项的文本值选择下拉框
        @param locator:
        @param value:
        @return:
        """
        elem = self.find_element(locator)
        Select(elem).select_by_visible_text(value)

    def elem_object_click(self, elem):
        """
        元素点击，传入参数为元素对象
        @param elem:
        @return:
        """
        elem.click()

    def find_elements(self, selector):
        """
        :param selector: 传入定位器参数locator => 标识切割
        :return: 返回元素对象列表
        """
        Logger.info("输出定位器信息：{}".format(selector))

        element = ''
        if '=>' not in selector:
            Logger.info("Not found '=>', default is ID")
            return self.driver.find_element_by_id(selector)
        selector_by = selector.split('=>')[0]
        selector_value = selector.split('=>')[1]

        if selector_by is None and selector_value is None:
            Logger.error('find_elements：selector_by and selector_value is empty:，Please pass the correct value)')
        else:
            Logger.info("find_elements：正在定位元素信息：定位方式->%s,value值->%s" % (selector_by, selector_value))
            try:
                time.sleep(1)
                elems = WebDriverWait(self.driver, self.timeout, self.poll_frequency).until(
                    lambda x: x.find_elements(*selector))
                Logger.info("元素对象为：{}".format(elems))
                return elems
            except Exception as e:
                Logger.error("定位不到元素，错误信息为:{}".format(e))
                return False

    def choose_elem_by_visible_value(self, locator, value):
        """
        根据一组元素对象中某一个元素对象的文本值确定，哪一个元素对象
        @param locator:
        @param value:
        @return: 单个元素对象
        """
        elems = self.find_elements(locator)
        for item in elems:
            if self.elem_object_get_text(item) == value:
                return item

    def elem_object_get_text(self, elem):
        """
        元素对象获取text值，传入元素对象
        @param elem:
        @return:
        """
        elem_text = elem.text
        return elem_text

    def get_text_by_elements(self, locator, index):
        """
        @param locator: 定位器
        @return: 返回定位对象组的第一个元素的值
        """
        global elem_text
        elem = self.find_elements(locator)
        try:
            elem_text = elem[index].text
            Logger.info("获取元素组对象，索引位置{}的值成功，值为：{}".format(index, elem_text))
        except Exception as e:
            Logger.error("获取元素组对象，索引位置{}的值失败，失败信息为: {}".format(e, e))
        return elem_text

    '''--静态方法--'''
    @staticmethod
    def sleep(seconds):
        time.sleep(seconds)
        Logger.info("Sleep for %d seconds" % seconds)