как изменить атрибут элемента

Здравствуйте

Подскажите, пожалуйста, кто знает, как изменить атрибут элемента на странице напрямую (например, в div поменять атрибут class)

я использую selenium webdriver, пишу на python

Спасибо

WebDriver driver; // Assigned elsewhere
JavascriptExecutor js = (JavascriptExecutor) driver;
js.executeScript("document.getElementById('idElement').setAttribute('class','className'));