webdriver高级应用- 改变一个页面对象的属性值

适用于一些无法操作的元素，可以直接改他的属性从而操作，代码如下：

#encoding=utf-8
from selenium import webdriver
import unittest
import time
 
def addAttribute(driver, elementObj, attributeName, value):
    # 封装向页面标签中添加新属性方法
    # 调用JavaScript代码给页面标签添新属性，arguments[0]－［2］分别会用后面的
    # element、attributeName和value参数值进行替换，并执行该JavaScript代码
    # 添加新属性的JavaScript代码语法为：element.attributeName = value
    # 比如input.name="test"
    driver.execute_script("arguments[0].%s=arguments[1]" %attributeName,\
                          elementObj, value)
 
def setAttribute(driver, elementObj, attributeName, value):
    # 封装设置页面对象的属性值的方法
    # 调用JavaScript代码修改页面元素的属性值，arguments[0]－［2］分别会用后面的
    # element、attributeName和value参数值进行替换，并执行该JavaScript代码
    driver.execute_script("arguments[0].setAttribute\
    (arguments[1],arguments[2])", elementObj, attributeName, value)
 
def getAttribute(elementObj, attributeName):
    # 封装获取页面对象的属性值的方法
    return elementObj.get_attribute(attributeName)
 
def removeAttribute(driver, elementObj, attributeName):
    # 封装删除页面元素属性的方法
    # 调用JavaScript代码删除页面元素的指定的属性，arguments[0]－［1］分别会用后面的
    # element、attributeName参数值进行替换，并执行该JavaScript代码
    driver.execute_script("arguments[0].removeAttribute(arguments[1])",\
                          elementObj, attributeName)
 
class TestDemo(unittest.TestCase):
 
    def setUp(self):
        # 启动Chrome浏览器
        #self.driver = webdriver.Ie(executable_path = "e:\\IEDriverServer")
        self.driver = webdriver.Chrome(executable_path = "e:\\chromedriver")
        #self.driver = webdriver.Firefox(executable_path = "e:\\geckodriver")
    def test_dataPicker(self):
        url = "http://127.0.0.1/test_change_attr.html"
        # 访问自定义网页
        self.driver.get(url)
        # 找到页面上标签名为input的页面元素
        element = self.driver.find_element_by_xpath("//input")
 
        # 向页面文本框input标签中添加新属性name="search"
        addAttribute(self.driver, element, 'name', "search")
        # 添加新属性后，查看一下新添加的属性
        print u'添加的新属性值%s="%s"' %("name", getAttribute(element, "name"))
 
        # 查看修改前文本框input标签的value属性值
        print u"更改文本框中的内容前的内容：", getAttribute(element, "value")
        # 更改input页面元素的value属性值为“这是更改后的文字内容”
        setAttribute(self.driver, element, "value", u"xxxxxxx")
        # 更改input页面元素的value属性值后，再次查看其value属性值
        print u"更改文本框中内容后的内容：", getAttribute(element, "value")
        time.sleep(3)
 
        # 查看修改前文本框input页面元素中的size属性值
        print u"更改前文本框标签中的size属性值：", getAttribute(element, "size")
        # 更改input页面元素的size属性值为“20”
        setAttribute(self.driver, element, "size", 20)
        # 更改input页面元素的size属性值后，再次查看其size属性值
        print u"更改后文本框标签中的size属性值：", getAttribute(element, "size")
        time.sleep(3)
 
        # 查看删除input页面元素value属性前value属性值
        print u"文本框value属性值：", getAttribute(element, "value")
        # 删除文本框的value属性
        removeAttribute(self.driver, element, "value")
        # 删除文本框的value属性后，再次查看value属性值
        print u"删除value属性值后value属性值：", getAttribute(element, "value")
        time.sleep(3)
 
    def tearDown(self):
        # 退出IE浏览器
        self.driver.quit()
 
if __name__ == '__main__':
    unittest.main()