selenium通过js更改属性值

from selenium import webdriver
import unittest

def addAttribute(driver,elementObj,attributeName,value):
    driver.execute_script("arguments[0].%s=arguments[1]" % attributeName,elementObj,value)

def setAttribute(driver,elementObj,attributeName,value):
    driver.execute_script('arguments[0].setAttribute(auguments[1],arguments[2])',elementObj,attributeName,value)

def getAttribute(elementObj,attributeName):
    return elementObj.get_attribute(attributeName)

def removeAttribute(driver,elementObj,attributeName):
    driver.execute_script("arguments[0].removeAttribute(arguments[1])",elementObj,attributeName)


class TestDemo(unittest.TestCase):

    def setUp(self):
        self.driver = webdriver.Chrome(executable_path='')

    def test_DataPicker(self):
        url = ''
        self.driver.get(url)
        element = self.driver.find_element_by_xpath("//input")
        addAttribute(self.driver,element,'name','seach')
        print("添加的新属性值'%s'为'%s'" % ("name",getAttribute(element,'name')))

        print("更改之前的内容为:",getAttribute(element,'value'))
        setAttribute(self.driver,element,'value',"更改之后的内容")
        print("更改之后的内容：",getAttribute(element,"value"))

        print("更改文本框中的size属性值",getAttribute(element,"size"))
        setAttribute(self.driver,element,"size",20)
        print("更改之后的size属性值",getAttribute(element,'size'))

        print("文本框value属性值：",getAttribute(element,'value'))
        removeAttribute(self.driver,element,"value")
        print("删除属性值之后的value：",getAttribute(element,'value'))

    def tearDown(self):
        self.driver.quit()


if __name__ == '__main__':
    unittest.main()