Selenium自动化之JS增删改查操作元素的属性

#JS增删改查操作元素的属性
#新增属性
driver.execute_script(“arguments[0].%s=arguments[1]” %attributeName,elementObj, value)
#修改属性
driver.execute_script(“arguments[0].setAttribute(arguments[1],arguments[2])”, elementObj, attributeName, value)
#获取属性
elementObj.get_attribute(attributeName)
#删除属性
driver.execute_script(“arguments[0].removeAttribute(arguments[1])”,elementObj, attributeName)

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 setAttri