from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.common.by import By
from selenium.webdriver.support.select import Select
 


class Pyse(object):
    def __init__(self, browser=‘chrome‘): # 初始化，生成driver
        if browser == "firefox" or browser == "ff":
            driver = webdriver.Firefox()
        elif browser == "chrome":
            option = webdriver.ChromeOptions()
            option.add_argument("--start-maximized")
            driver = webdriver.Chrome(chrome_options=option)
 

        elif browser == "internet explorer" or browser == "ie":
            driver = webdriver.Ie()
        elif browser == "opera":
            driver = webdriver.Opera()
        elif browser == "phantomjs":
            driver = webdriver.PhantomJS()
        elif browser == ‘edge‘:
            driver = webdriver.Edge()
 

        try:
            self.driver = driver
        except Exception:
            raise NameError(
                "Not found %s browser,You can enter ‘ie‘, ‘ff‘, ‘opera‘, ‘phantomjs‘, ‘edge‘ or ‘chrome‘." % browser)

    def element_wait(self, css, secs=5): # 元素等待
        ‘‘‘
        Waiting for an element to display.

        Usage:
        driver.element_wait("css=>#el",10)
        ‘‘‘
        if "=>" not in css:
            raise NameError("Positioning syntax errors, lack of ‘=>‘.")

        by = css.split("=>")[0]
        value = css.split("=>")[1]

        if by == "id":
            WebDriverWait(self.driver, secs, 1).until(EC.presence_of_element_located((By.ID, value)))
        elif by == "name":
            WebDriverWait(self.driver, secs, 1).until(EC.presence_of_element_located((By.NAME, value)))
        elif by == "class":
            WebDriverWait(self.driver, secs, 1).until(EC.presence_of_element_located((By.CLASS_NAME, value)))
        elif by == "link_text":
            WebDriverWait(self.driver, secs, 1).until(EC.presence_of_element_located((By.LINK_TEXT, value)))
        elif by == "xpath":
            WebDriverWait(self.driver, secs, 1).until(EC.presence_of_element_located((By.XPATH, value)))
        elif by == "css":
            WebDriverWait(self.driver, secs, 1).until(EC.presence_of_element_located((By.CSS_SELECTOR, value)))
        else:
            raise NameError(
                "Please enter the correct targeting elements,‘id‘,‘name‘,‘class‘,‘link_text‘,‘xpath‘,‘css‘.")

    def get_element(self, css): # 獲取元素
        ‘‘‘
        Judge element positioning way, and returns the element.
        ‘‘‘
        if "=>" not in css:
            raise NameError("Positioning syntax errors, lack of ‘=>‘.")

        by = css.split("=>")[0]
        value = css.split("=>")[1]
        self.element_wait(css)

        if by == "id":
            element = self.driver.find_element_by_id(value)
        elif by == "name":
            element = self.driver.find_element_by_name(value)
        elif by == "class":
            element = self.driver.find_element_by_class_name(value)
        elif by == "link_text":
            element = self.driver.find_element_by_link_text(value)
        elif by == "xpath":
            element = self.driver.find_element_by_xpath(value)
        elif by == "css":
            element = self.driver.find_element_by_css_selector(value)
        else:
            raise NameError(
                "Please enter the correct targeting elements,‘id‘,‘name‘,‘class‘,‘link_text‘,‘xpath‘,‘css‘.")
        return element

    def open(self, url): # 打開url
        ‘‘‘
        open url.

        Usage:
        driver.open("https://www.baidu.com")
        ‘‘‘
        self.driver.get(url)

    def max_window(self): # 瀏覽器窗口最大化
        ‘‘‘
        Set browser window maximized.

        Usage:
        driver.max_window()
        ‘‘‘
        self.driver.maximize_window()

    def set_window(self, wide, high): # 設置瀏覽器窗口的寬和高
        ‘‘‘
        Set browser window wide and high.

        Usage:
        driver.set_window(wide,high)
        ‘‘‘
        self.driver.set_window_size(wide, high)

    def type(self, css, text): # 向文本框發送字符串
        ‘‘‘
        Operation input box.

        Usage:
        driver.type("css=>#el","selenium")
        ‘‘‘
        el = self.get_element(css)
        el.send_keys(text)

    def clear(self, css): # 清除輸入框裏的內容
        ‘‘‘
        Clear the contents of the input box.

        Usage:
        driver.clear("css=>#el")
        ‘‘‘
        el = self.get_element(css)
        el.clear()

    def click(self, css): # 鼠標左鍵單擊
        ‘‘‘
        It can click any text / image can be clicked
        Connection, check box, radio buttons, and even drop-down box etc..

        Usage:
        driver.click("css=>#el")
        ‘‘‘
        el = self.get_element(css)
        el.click()

    def right_click(self, css): # 鼠標右鍵單擊
        ‘‘‘
        Right click element.

        Usage:
        driver.right_click("css=>#el")
        ‘‘‘
        el = self.get_element(css)
        ActionChains(self.driver).context_click(el).perform()

    def move_to_element(self, css): # 鼠標懸浮到目標元素上再點擊
        ‘‘‘
        Mouse over the element.

        Usage:
        driver.move_to_element("css=>#el")
        ‘‘‘
        el = self.get_element(css)
        ActionChains(self.driver).move_to_element(el).perform()

    def double_click(self, css): # 雙擊操作
        ‘‘‘
        Double click element.

        Usage:
        driver.double_click("css=>#el")
        ‘‘‘
        el = self.get_element(css)
        ActionChains(self.driver).double_click(el).perform()

    def drag_and_drop(self, el_css, ta_css): # 拖拽操作，拼拼圖
        ‘‘‘
        Drags an element a certain distance and then drops it.

        Usage:
        driver.drag_and_drop("css=>#el","css=>#ta")
        ‘‘‘
        element = self.get_element(el_css)
        target = self.get_element(ta_css)
        ActionChains(driver).drag_and_drop(element, target).perform()

    def click_text(self, text): # 包含link的點擊操作
        ‘‘‘
        Click the element by the link text

        Usage:
        driver.click_text("新聞")
        ‘‘‘
        self.driver.find_element_by_partial_link_text(text).click()

    def close(self): # 關閉當前瀏覽器
        ‘‘‘
        Simulates the user clicking the "close" button in the titlebar of a popup
        window or tab.

        Usage:
        driver.close()
        ‘‘‘
        self.driver.close()

    def quit(self): # 關閉所有打開的瀏覽器
        ‘‘‘
        Quit the driver and close all the windows.

        Usage:
        driver.quit()
        ‘‘‘
        self.driver.quit()

    def submit(self, css): # 提交操作
        ‘‘‘
        Submit the specified form.

        Usage:
        driver.submit("css=>#el")
        ‘‘‘
        el = self.get_element(css)
        el.submit()

    def F5(self): # 瀏覽器按鈕刷新操作
        ‘‘‘
        Refresh the current page.

        Usage:
        driver.F5()
        ‘‘‘
        self.driver.refresh()

    def js(self, script): # 執行js
        ‘‘‘
        Execute JavaScript scripts.

        Usage:
        driver.js("window.scrollTo(200,1000);")
        ‘‘‘
        self.driver.execute_script(script)

    def get_attribute(self, css, attribute): # 獲取屬性
        ‘‘‘
        Gets the value of an element attribute.

        Usage:
        driver.get_attribute("css=>#el","type")
        ‘‘‘
        el = self.get_element(css)
        return el.get_attribute(attribute)

    def get_text(self, css): # 獲取文本
        ‘‘‘
        Get element text information.

        Usage:
        driver.get_text("css=>#el")
        ‘‘‘
        el = self.get_element(css)
        return el.text

    def get_display(self, css): # 讓元素展示
        ‘‘‘
        Gets the element to display,The return result is true or false.

        Usage:
        driver.get_display("css=>#el")
        ‘‘‘
        el = self.get_element(css)
        return el.is_displayed()

    def get_title(self): # 獲取標題
        ‘‘‘
        Get window title.

        Usage:
        driver.get_title()
        ‘‘‘
        return self.driver.title

    def get_url(self): # 獲取url
        ‘‘‘
        Get the URL address of the current page.

        Usage:
        driver.get_url()
        ‘‘‘
        return self.driver.current_url

    def get_windows_img(self, file_path): # 截圖
        ‘‘‘
        Get the current window screenshot.

        Usage:
        driver.get_windows_img()
        ‘‘‘
        self.driver.get_screenshot_as_file(file_path)

    def wait(self, secs): # 隱式等待
        ‘‘‘
        Implicitly wait.All elements on the page.

        Usage:
        driver.wait(10)
        ‘‘‘
        self.driver.implicitly_wait(secs)

    def accept_alert(self): # 關閉提示框，確定
        ‘‘‘
        Accept warning box.

        Usage:
        driver.accept_alert()
        ‘‘‘
        self.driver.switch_to.alert.accept()

    def dismiss_alert(self): # 關閉提示框，取消
        ‘‘‘
        Dismisses the alert available.

        Usage:
        driver.dismiss_alert()
        ‘‘‘
        self.driver.switch_to.alert.dismiss()

    def switch_to_frame(self, css): # 向下跳一層
        ‘‘‘
        Switch to the specified frame.

        Usage:
        driver.switch_to_frame("css=>#el")
        ‘‘‘
        self.element_wait(css)
        iframe_el = self.get_element(css)
        self.driver.switch_to.frame(iframe_el)

    def switch_to_frame_out(self): # 跳到最外層
        ‘‘‘
        Returns the current form machine form at the next higher level.
        Corresponding relationship with switch_to_frame () method.

        Usage:
        driver.switch_to_frame_out()
        ‘‘‘
        self.driver.switch_to.default_content()

    def open_new_window(self, css): # 打開新窗口並切換到新打開的窗口
        ‘‘‘
        Open the new window and switch the handle to the newly opened window.

        Usage:
        driver.open_new_window()
        ‘‘‘
        original_windows = self.driver.current_window_handle
        el = self.get_element(css)
        el.click()
        all_handles = self.driver.window_handles
        for handle in all_handles:
            if handle != original_windows:
                self.driver.switch_to.window(handle)

    # def _save_png(self, name):
    #     self.get_windows_img(name)

    def wait_and_save_exception(self, css, name): # 等待並保存異常
        try:
            self.element_wait(css, secs=5)
            print(name)
            return True
        except Exception as e:
            from lib.core.path import PICTUREPATH
            name = PICTUREPATH + name + ‘.jpg‘
            print(name)
            self.get_windows_img(name)
            return False

    def wait_and_exception(self, css):
        try:
            self.element_wait(css, secs=10)
            return True
        except Exception as e:
            return False

    def select_by_value(self, css, value): # 通過value修改單選項
        el = self.get_element(css)
        Select(el).select_by_value(value)

if __name__ == ‘__main__‘:
    driver = Pyse("chrome")

官網pyse的修改版