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”)