selenium對元素定位和操作方法的封裝

前面介紹了selenium對元素的定位方法和操作方法，但是因為這些方法用起來重復的次數比較多，所以我們在下面對他們進行一個封裝

# -*- coding:UTF-8 -*-
from selenium import  webdriver
import time
class common(object):
    #新建對象就自動創建瀏覽器並且最大化窗口
    def __init__(self):
        self.driver=webdriver.Chrome()
        self.driver.maximize_window()
    #打開網頁功能
    def open(self,url):
        self.driver.get(url)
        self.driver.implicitly_wait(5)
    # 關閉網頁功能
    def close(self):
        time.sleep(3)
        self.driver.close()
    # 定位元素功能
    def locateElement(self,type,value):
        if type=="id":
            el=self.driver.find_element_by_id(value)
        elif type=="name":
            el=self.driver.find_element_by_name(value)
        elif type=="class_name":
            el=self.driver.find_element_by_class_name(value)
        elif type=="tag_name":
            el=self.driver.find_element_by_tag_name(value)
        elif type=="link_text":
            el=self.driver.find_element_by_link_text(value)
        elif type=="partial_link_text":
            el=self.driver.find_element_by_partial_link_text(value)
        elif type=="xpath":
            el=self.driver.find_element_by_xpath(value)
        elif type=="css_selector":
            el=self.driver.find_element_by_css_selector(value)

        return el

    #點擊元素功能
    def click(self,type,value):
        # 調用locateElement定位元素
        el=self.locateElement(type,value)
        #調用click()進行點擊操作
        el.click()

    # 對定位到元素進行輸入
    def input_data(self, type, value,data):
        # 調用locateElement定位元素
        el = self.locateElement(type, value)
        # 調用send_keys進行輸入
        el.send_keys(data)


    # 獲取定位到的元素中的文本內容<a>text</a>
    def getText(self, type, value):
        # 調用locateElement定位元素
        el = self.locateElement(type, value)
        # 返回文本內容
        return el.text

    # 獲取定位到的元素中的標簽值
    def getAttribute(self, type, value,name):
        # 調用locateElement定位元素
        el = self.locateElement(type, value)
        # 返回文本內容
        return el.get_attribute(name)

    # 刪除對象時自動執行的方法
    def __del__(self):
        self.driver.close()

#如果是引用該類則不執行該方法 if __name__=="__main__":
    test=common()
    url="https://www.baidu.com"
    test.open(url)
    test.input_data("id","kw","hiro")
    test.click("id","su")




# -*- author: Hiro -*-

還封裝了打開瀏覽器和全屏等方法