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)