python+selenium十：selenium的二次封装

Posted 2021-01-21 怪圣卡杰

 

from selenium import webdriver
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.select import Select
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.common.by import By
# BY的用法
# driver.find_element("id", "kw")
# driver.find_element(By.ID, "kw")

class Bace():
    ‘‘‘基于原生的selenium做二次封装‘‘‘

    def __init__(self, driver:webdriver.Firefox):  # driver:webdriver.Firefox：映射driver 为webdriver.Firefox
        self.driver = driver
        self.timeout = 10
        self.t = 0.5

    def find(self, locator, value=‘‘):
        ‘‘‘  定位到元素，返回元素对象，没定位到，Timeout异常 loctor 传元祖，如（"id", "kw"）  ‘‘‘
        if not isinstance(locator, tuple):
            print(‘locator参数类型错误，必须传元祖类型：loc = ("id", "value1")‘)
        else:
            print(f"正在定位元素信息：定位方式->{locator[0]}, 元素值->{locator[1]}，value值->{value}")
            if value != ‘‘:  # value值定位
                ele = WebDriverWait(self.driver, self.timeout, self.t).until(EC.text_to_be_present_in_element_value(locator, value))
                return ele
            else:  # 默认为此常规定位方法
                ele = WebDriverWait(self.driver, self.timeout, self.t).until(EC.presence_of_element_located(locator))
                if ele:
                    return ele
                else:
                    print(f"定位失败：定位方式->{locator[0]}, value值->{locator[1]}")
                    return False

    def finds(self, locator, value=‘‘):
        ‘‘‘ 定位到元素，返回元素对象，没定位到，Timeout异常  loctor 传元祖，如（"id", "kw"）  ‘‘‘
        if not isinstance(locator, tuple):
            print(‘locator参数类型错误，必须传元祖类型：loc = ("id", "value1")‘)
        else:
            print(f"正在定位元素信息：定位方式->{locator[0]}, 元素值->{locator[1]}，value值->{value}")
            if value != ‘‘:  # value值定位
                eles = WebDriverWait(self.driver, self.timeout, self.t).until(EC.text_to_be_present_in_element_value(locator, value))
                return eles
            else:  # 默认为此常规定位方法
                eles = WebDriverWait(self.driver, self.timeout, self.t).until(EC.presence_of_element_located(locator))
                if eles:
                    return eles
                else:
                    print(f"定位失败：定位方式->{locator[0]}, value值->{locator[1]}")
                    return []

    def sendKeys(self, locator, text):
        try:
            self.find(locator).send_keys(text)
        except:
            print(f"输入 {text} 失败")

    def click(self, locator):
        try:
            self.find(locator).click()
        except:
            print("点击失败")

    def clear(self, locator):
        try:
            self.find(locator).clear()
        except:
            print("清空内容失败")

    def isSelected(self, locator, Type=‘‘):
        ‘‘‘ 判断元素是否被选中，返回bool值 及点（选中/取消选中）‘‘‘
        ele = self.find(locator)
        try:
            if Type == ‘‘:  # 如果type参数为空，返回元素是否为选中状态，True/False (默认)
                r = ele.is_selected()
                return r
            elif Type == ‘click‘:  # 如果type参数为click，执行元素的点击操作
                ele.click()
            else:
                print(f"type参数 {Type} 错误，仅可为click或‘‘")
        except:
            return False

    def isElementExist(self, locator):
        ‘‘‘ 判断单个元素是否在DOM里面 （是否存在）‘‘‘
        try:
            self.find(locator)
            return True
        except:
            return False

    def isElementExists(self, locator):
        ‘‘‘ 判断一组元素是否在DOM里面 （是否存在），若不存在，返回一个空的list‘‘‘
        eles = self.finds(locator)
        n = len(eles)
        if n == 0:
            return False
        elif n == 1:
            return True
        else:
            print(f"定位到元素的个数：{n}")
            return True

    def title(self, title, Type=‘contains‘):
        ‘‘‘  根据传入的type类型判断title  ‘‘‘
        try:
            if Type == ‘is‘:  # 判断当前网页title名为title   返回bool值
                result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.title_is(title))
                return result
            elif Type == ‘contains‘:  # 判断当前网页title名含title   返回bool值 (默认)
                result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.title_contains(title))
                return result
            else:
                print(f"type参数 {Type} 错误，仅可为is、contains")
        except:
            return False

    def in_element(self, locator, value, Type=‘text‘):
        ‘‘‘  根据传入的type判断内容是否在指定元素里面  ‘‘‘
        if not isinstance(locator, tuple):
            print(‘locator参数类型错误，必须传元祖类型：loc = ("id", "value1")‘)
        try:
            if Type == ‘text‘:  # 判断当前获取到的text含value   返回bool值 （默认）
                result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.text_to_be_present_in_element(locator, value))
                return result
            elif Type == ‘value‘:  # 判断当前获取到的value含value 返回bool值, value为空字符串，返回False
                result = self.find(locator, value)
                return result
            else:
                print(f"type参数 {Type} 错误，仅可使用text或value属性定位")
                return False
        except:
            return False

    def alert(self, timeout=3, Type=‘‘):
        ‘‘‘ 根据传入的type判断alert弹窗及操作 ‘‘‘
        result = WebDriverWait(self.driver, timeout, self.t).until(EC.alert_is_present())
        try:
            if Type == ‘‘:  # 判断alert是否存在，如果有，就返回alert对象 （默认）
                if result:
                    return result
                else:
                    print("alert不存在")
                return False
            elif Type == ‘yes‘:  # 执行alert的确定按钮
                result.accept()
            elif Type == ‘no‘:  # 执行alert的取消按钮
                result.dismiss()
            else:
                print(f"type参数 {Type} 错误，仅可为yes、no、或‘‘")
        except:
            return False

    def get(self, locator, Type=‘text‘, name=‘‘):
        ‘‘‘ 根据传入的type判断获取指定的内容 （title、text、attribute）
        type==attribute:  获取元素属性  name:属性  className、name、text、value···  ‘‘‘
        try:
            if Type == ‘title‘:  # 获取当前网页 title
                return self.driver.title
            elif Type == ‘text‘:  # 获取元素文本值（默认）
                return self.find(locator).text
            elif Type == ‘attribute‘:  # 获取当前元素属性
                return self.find(locator).get_attribute(name)
            else:
                print(f"给的type参数 {Type} 错误，仅可用title、text、attribute")
        except:
            print(f"获取 {Type} 值失败")
            return ‘‘

    def select(self, locator, value, Type=‘index‘):
        ‘‘‘  下拉选项框 根据传入的type值判断（index、value、text）  ‘‘‘
        element = self.find(locator)  # 定位select这一栏
        try:
            if Type == ‘index‘:  # 用下标选择 （默认）
                Select(element).select_by_index(value)
            elif Type == ‘value‘:  # 根据value值选择
                Select(element).select_by_value(value)
            elif Type == ‘text‘:  # 根据选项的文本内容选择
                Select(element).select_by_visible_text(value)
            else:
                print(f"给的type参数 {Type} 错误，仅可为：int、text、value")
        except:
            print(f"根据 {value} 操作下拉框失败")

    def iframe(self, id_index_locator):
        ‘‘‘ 常规切换 iframe‘‘‘
        try:
            if isinstance(id_index_locator, int):  # 如果传入的是数字，则以该数字为下标取值
                self.driver.switch_to.frame(id_index_locator)
            elif isinstance(id_index_locator, str):  # 如果传入的是字符串，则用iframe名字取值
                self.driver.switch_to.frame(id_index_locator)
            elif isinstance(id_index_locator, tuple):  # 如果是元祖，则根据传入的locator取值
                ele = self.find(id_index_locator)
                self.driver.switch_to.frame(ele)
        except:
             print("iframe切换异常")

    def handle(self, value):
        ‘‘‘ 句柄切换，index、句柄名 ‘‘‘
        try:
            if isinstance(value, int):  # 切换到该下标对应的窗口
                handles = driver.window_handles
                self.driver.switch_to.window(handles[value])
            elif isinstance(value, str):  # 切换到该句柄名称对应的窗口
                self.driver.switch_to.window(value)
            else:
                print(f"传入的type参数 {value} 错误，仅可传int、str")
        except:
            print(f"根据 {value} 获取句柄失败")

    def move_to_element(self, locator):
        ‘‘‘ 鼠标悬停操作 ‘‘‘
        try:
            ele = self.find(locator)
            ActionChains(self.driver).move_to_element(ele).perform()
        except:
            print("鼠标悬停操作失败")
            return False
    ‘‘‘==============================js与jQuery相关=====================================‘‘‘

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

    def js_scroll_top(self):
        ‘‘‘ 滚动到顶部 ‘‘‘
        js = "window.scrollTo(0,0)"
        self.driver.execute_script(js)

    def js_scroll_end(self, x=0):
        ‘‘‘ 滚动到底部 ‘‘‘
        js = f"window.scrollTo({x},document.body.scrollHeight)"
        self.driver.execute_script(js)

    def js_find(self, action):
        ‘‘‘  js查找元素，并做相应操作（默认id属性） 输入值：value=‘XXX‘  点击：click() ‘‘‘
        js = f"document.getElementById(“id”).{action}"
        self.driver.execute_script(js)

    def js_finds(self, Type, element, index, action):
        ‘‘‘   js查找元素，并做相应操作   输入值：value=‘XXX‘     点击：click()
        js定位仅可为：id、Name、TagName、ClassName、Selector（CSS） ‘‘‘
        list = [‘Name‘, ‘TagName‘, ‘ClassName‘, ‘Selector‘]
        if type in list:
            print(f"正在执行js操作：定位方式->{Type}, 元素值->{element}， 下标值->{index}， 执行操作->{action}")
            if type == ‘Selector‘:
                js = f‘document.query{Type}All("{element}"){index}.{action}‘
            else:
                js = f‘document.getElementsBy{Type}({element})[{index}].{action};‘
            self.driver.execute_script(js)
        else:
            print(f"type参数 {Type} 错误，js定位仅可为：‘Name‘、‘TagName‘、‘ClassName‘、‘Selector‘（CSS）")

    def js_readonly(self, idElement, value):
        ‘‘‘ 去掉只读属性，并输入内容 一般为id ‘‘‘
        js = f‘document.getElementById({idElement}).removeAttribute("readonly");document.getElementById({idElement}).value="{value}"‘
        driver.execute_script(js)

    def js_iframe(self, Type, element, action, index=‘‘):
        ‘‘‘  Js处理iframe   无需先切换到iframe上，再切回来操作
        输入值：value=‘‘         点击：click()           type=id时，index=‘‘  ‘‘‘
        js = f‘document.getElementBy{Type}({element}){index}.contentWindow.document.body.{action}‘
        driver.execute_script(js)
    ‘‘‘
        jquery = ‘$(CSS).val("XXX");‘   # 根据css语法定位到元素，输入内容
        jquery = ‘$(CSS).val(‘‘);‘      # 清空
        jquery = ‘$(CSS).click();‘      # 点击
        driver.execute_script(jquery)
    ‘‘‘

    # def switch_alert(self):
    #     ‘‘‘ 获取alert弹窗 ‘‘‘
    #     r = self.is_alert()
    #     if not r:
    #         print("alert不存在")
    #     else:
    #         return r

    # def is_title(self, title):
    #     ‘‘‘判断当前title名为title   返回bool值‘‘‘
    #     try:
    #         result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.title_is(title))
    #         return result
    #     except:
    #         return False
    # def is_title_contains(self, title):
    #     ‘‘‘判断当前title名含title   返回bool值‘‘‘
    #     try:
    #         result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.title_contains(title))
    #         return result
    #     except:
    #         return False

    # def is_text_in_element(self, locator, _text=‘‘):
    #     ‘‘‘判断当前获取到的text含_text=‘‘   返回bool值‘‘‘
    #     if not isinstance(locator, tuple):
    #         print(‘locator参数类型错误，必须传元祖类型：loc = ("id", "value1")‘)
    #     try:
    #         result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.text_to_be_present_in_element(locator, _text))
    #         return result
    #     except:
    #         return False
    # def is_value_in_element(self, locator, _value=‘‘):
    #     ‘‘‘返回bool值, value为空字符串，返回False‘‘‘
    #     if not isinstance(locator, tuple):
    #         print(‘locator参数类型错误，必须传元祖类型：loc = ("id", "value1")‘)
    #     try:
    #         result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.text_to_be_present_in_element_value(locator, _value))
    #         return result
    #     except:
    #         return False

    # def get_title(self):
    #     ‘‘‘获取title‘‘‘
    #     return self.driver.title
    # def get_text(self, locator):
    #     ‘‘‘获取文本‘‘‘
    #     try:
    #         t = self.find(locator).text
    #         return t
    #     except:
    #         print("获取text失败，返回‘‘ ")
    #         return ""
    # def get_attribute(self, locator, name):
    #     ‘‘‘获取属性‘‘‘
    #     try:
    #         element = self.find(locator)
    #         return element.get_attribute(name)
    #     except:
    #         print("获取%s属性失败，返回‘‘ "%name)
    #         return ""

    # def select_by_index(self, locator, index=0):
    #     ‘‘‘通过索引,index是索引第几个，从0开始，默认选第一个‘‘‘
    #         element = self.find(locator)  # 定位select这一栏
    #         Select(element).select_by_index(index)

    # def select_by_value(self, locator, value):
    #     ‘‘‘通过value属性‘‘‘
    #     element = self.find(locator)
    #     Select(element).select_by_value(value)

    # def select_by_text(self, locator, text):
    #     ‘‘‘通过文本值定位‘‘‘
    #     element = self.find(locator)
    #     Select(element).select_by_visible_text(text)

    # def switch_handle_window_name(self, window_name):
    #     ‘‘‘ 根据句柄名字切换句柄 ‘‘‘
    #     self.driver.switch_to.window(window_name)
    # def switch_handle_index(self, index):
    #     ‘‘‘  根据句柄下标切换句柄  ‘‘‘
    #     handles = driver.window_handles
    #     self.driver.switch_to.window(handles[index])

    # def js_find(self, action):
    #     ‘‘‘
    #     输入值：value=‘XXX‘     点击：click()
    #     ‘‘‘
    #     print("正在执行js操作，操作行为：%s"%action)
    #     js = "document.getElementById(“id”).%s"%action
    #     self.driver.execute_script(js)

if __name__ == "__main__":
    driver = webdriver.Firefox()
    driver.get("")
    zentao = Base(driver)
    # loc1 = (By.ID, "account")
    # loc2 = (By.CSS_SELECTOR, "[name=‘password‘]")
    # loc3 = (By.XPATH, "//*[@id=‘submit‘]")

    loc1 = ("id", "account")
    loc2 = ("css selector", "[name=‘password‘]")
    loc3 = ("xpath", "//*[@id=‘submit‘]")
    zentao.sendKeys(loc2, 123)

    zentao.move_to_element(loc3)