二次封装selenium，简易框架Jzps

港控/mmm° 2022-07-17 00:17 243阅读 0赞

大家可以看一下最后的例子，希望大家使用起来可以更方便

![这里写图片描述][20160823153401164]

#!/usr/bin/env python
    # coding:utf-8
    
    """ FuncName: Jzps.py Desc: Jzps framework are committed to a simpler automated testing,based on the original selenium. Date: 2016-06-20 10:00 Home: http://blog.csdn.net/z_johnny Author: johnny """
    
    from selenium import webdriver
    from selenium.webdriver.common.action_chains import ActionChains
    from selenium.common.exceptions import StaleElementReferenceException
    from selenium.webdriver.support import expected_conditions as EC
    from selenium.webdriver.support.ui import WebDriverWait
    from selenium.webdriver.common.by import By
    import os,sys,time
    reload(sys)
    sys.setdefaultencoding('utf-8')
    
    class Jzps(object):
        """ Jzps framework are committed to a simpler automated testing, based on the original selenium. """
    
        def __init__(self, browser='firefox'):
            """ Run class initialization method, the default is proper to drive the Firefox browser,. Of course, you can also pass parameter for other browser, such as Chrome browser for the "Chrome", the Internet Explorer browser for "internet explorer" or "ie". """
            if browser == "firefox" :
                driver = webdriver.Firefox()
            elif browser == "chrome":
                driver = webdriver.Chrome()
            elif browser == "ie" :
                driver = webdriver.Ie()
            elif browser == "phantomjs":
                driver = webdriver.PhantomJS()
            try:
                self.driver = driver
            except Exception:
                raise NameError("Not found this browser,You can enter 'firefox', 'chrome', 'ie' or 'phantomjs'.")
    
        def get(self, url):
            """ Open url,same as get. Usage: driver.get("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_size(self, wide, high):
            """ Set browser window wide and high. Usage: driver.set_window_size(wide,high) """
            self.driver.set_window_size(wide, high)
    
        def wait(self, secsonds):
            """ Implicitly wait.All elements on the page. Usage: driver.wait(10) """
            self.driver.implicitly_wait(secsonds)
    
        def find_element(self,element):
            """ Judge element positioning way, and returns the element. Usage: driver.find_element("id=kw") """
            if "=" not in element:
                raise NameError("SyntaxError: invalid syntax, lack of '='.")
    
            by = element.split("=")[0]
            value = element.split("=")[1]
    
            if by == "id":
                return self.driver.find_element_by_id(value)
            elif by == "name":
                return self.driver.find_element_by_name(value)
            elif by == "class":
                return self.driver.find_element_by_class_name(value)
            elif by == "text":
                return self.driver.find_element_by_link_text(value)
            elif by == "text_part":
                return self.driver.find_element_by_partial_link_text(value)
            elif by == "xpath":
                return self.driver.find_element_by_xpath(value)
            elif by == "css":
                return self.driver.find_element_by_css_selector(value)
            else:
                raise NameError("Please enter the correct targeting elements,'id','name','class','text','xpath','css'.")
    
        def wait_element(self, element, seconds=5):
            """ Waiting for an element to display. Usage: driver.wait_element("id=kw",10) """
            if "=" not in element:
                raise NameError("SyntaxError: invalid syntax, lack of '='.")
    
            by = element.split("=")[0]
            value = element.split("=")[1]
    
            if by == "id":
                WebDriverWait(self.driver,seconds,1).until(EC.presence_of_element_located((By.ID, value)))
            elif by == "name":
                WebDriverWait(self.driver,seconds,1).until(EC.presence_of_element_located((By.NAME, value)))
            elif by == "class":
                WebDriverWait(self.driver,seconds,1).until(EC.presence_of_element_located((By.CLASS_NAME, value)))
            elif by == "text":
                WebDriverWait(self.driver,seconds,1).until(EC.presence_of_element_located((By.LINK_TEXT, value)))
            elif by == "xpath":
                WebDriverWait(self.driver,seconds,1).until(EC.presence_of_element_located((By.XPATH, value)))
            elif by == "css":
                WebDriverWait(self.driver,seconds,1).until(EC.presence_of_element_located((By.CSS_SELECTOR, value)))
            else:
                raise NameError("Please enter the correct targeting elements,'id','name','class','text','xpaht','css'.")
    
        def send_keys(self, element, text):
            """ Operation input content after clear. Usage: driver.send_keys("id=kw","selenium") """
            self.wait_element(element)
            self.find_element(element).clear()
            self.find_element(element).send_keys(text)
    
        def click(self, element):
            """ It can click any text / image can be clicked Connection, check box, radio buttons, and even drop-down box etc.. Usage: driver.click("id=kw") """
            self.wait_element(element)
            self.find_element(element).click()
    
        def right_click(self, element):
            """ Right click element. Usage: driver.right_click("class=right") """
            self.wait_element(element)
            ActionChains(self.driver).context_click(self.find_element(element)).perform()
    
        def move_to_element(self, element):
            ''' Mouse over the element. Usage: driver.move_to_element("css=choose") '''
            self.wait_element(element)
            ActionChains(self.driver).move_to_element(self.find_element(element)).perform()
    
        def double_click(self, element):
            """ Double click element. Usage: driver.double_click("name=baidu") """
            self.wait_element(element)
            ActionChains(self.driver).double_click(self.find_element(element)).perform()
    
        def drag_and_drop(self, source_element, target_element):
            """ Drags an element a certain distance and then drops it. Usage: driver.drag_and_drop("id=s","id=t") """
            self.wait_element(source_element)
            self.wait_element(target_element)
            ActionChains(self.driver).drag_and_drop(self.find_element(source_element), self.find_element(target_element)).perform()
    
        def back(self):
            """ Back to old window. Usage: driver.back() """
            self.driver.back()
    
        def forward(self):
            """ Forward to old window. Usage: driver.forward() """
            self.driver.forward()
    
        def get_attribute(self, element, attribute):
            """ Gets the value of an element attribute. Usage: driver.get_attribute("id=kw","attribute") """
            self.wait_element(element)
            return self.find_element(element).get_attribute(attribute)
    
        def get_text(self, element):
            """ Get element text information. Usage: driver.get_text("name=johnny") """
            self.wait_element(element)
            return self.find_element(element).text
    
        def get_display(self, element):
            """ Gets the element to display,The return result is true or false. Usage: driver.get_display("id=ppp") """
            self.wait_element(element)
            return self.find_element(element).is_displayed()
    
        def get_title(self):
            """ Get window title. Usage: driver.get_title() """
            return self.driver.title
    
        def get_url(self):
            """ Get the URL address of the current page. Usage: driver.get_url() """
            return self.driver.current_url
    
        def get_screenshot(self, file_path):
            """ Get the current window screenshot. Usage: driver.get_screenshot("./pic.png") """
            self.driver.get_screenshot_as_file(file_path)
    
        def submit(self, element):
            """ Submit the specified form. Usage: driver.submit("id=mainFrame") """
            self.wait_element(element)
            self.find_element(element).submit()
    
        def switch_to_frame(self, element):
            """ Switch to the specified frame. Usage: driver.switch_to_frame("id=mainFrame") """
            self.wait_element(element)
            self.driver._switch_to_frame(self.find_element(element))
    
        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, element):
            """ Open the new window and switch the handle to the newly opened window. Usage: driver.open_new_window(id=johnny) """
            current_windows = self.driver.current_window_handle
            self.find_element(element).click()
            all_handles = self.driver.window_handles
            for handle in all_handles:
                if handle != current_windows:
                    self.driver.switch_to.window(handle)
        def F5(self):
            ''' Refresh the current page. Usage: driver.F5() '''
            self.driver.refresh()
    
        def js(self, script):
            """ Execute JavaScript scripts. Usage: driver.js("window.scrollTo(200,1000);") """
            self.driver.execute_script(script)
    
        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 close(self):
            """ Close the windows. Usage: driver.close() """
            self.driver.close()
    
        def quit(self):
            """ Quit the driver and close all the windows. Usage: driver.quit() """
            self.driver.quit()
    
    
    if __name__ == '__main__':
        # from Jzps import Jzps
        driver = Jzps("chrome")               # 调用浏览器，支持 'firefox', 'chrome', 'ie' or 'phantomjs'
        driver.get("http://www.baidu.com")
        driver.find_element("id=kw")          # 通过id定位，支持'id','name','class','text','xpaht','css'
        driver.set_window_size(1366,768)       # 设置窗口尺寸
        time.sleep(2)
        driver.max_window()                    # 浏览器最大化
        driver.send_keys("id=kw",u"Johnny'lab")  # 该元素位置输入内容
        time.sleep(2)
        driver.click("id=su")
        time.sleep(2)
        driver.click("id=result_logo")      # 点击元素
        time.sleep(2)
        driver.F5()                            # 刷新页面
        driver.get_screenshot("./Jzps.png")  # 截图
        time.sleep(2)
        driver.right_click("text=登录")      # 鼠标右击
        print driver.get_url()
        print driver.get_text("name=tj_trnuomi")
        driver.move_to_element("id=su")
        print driver.get_title()
        driver.open_new_window("name=tj_trnuomi")
        driver.back()                         # 后退
        time.sleep(2)
        driver.forward()                      # 前进
        driver.close()

[20160823153401164]: /images/20220717/451684f8b398452c819775b2fc396210.png