selenium对元素定位和操作方法的封装

Posted 2020-11-14 a565810497

前面介绍了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 -*-

还封装了打开浏览器和全屏等方法