Трудно понять, что именно вы хотите из-за отсутствия описания.
Вы можете попробовать следующий подход (изменить / улучшить его), чтобы выделить, например, нажатую кнопку:
java_script = """var el = document.querySelectorAll('button');
for(var i=0;i<el.length;i++)
{
el[i].onclick = function()
{
this.setAttribute('style', "background: yellow; border: 2px solid red;")
};
}"""
driver.execute_script(java_script)
Это позволит вам выделить button
(изменить на любой тип элемента, который вы хотите), по которому щелкнули.
Вы также можете заменить this.setAttribute('style', "background: yellow; border: 2px solid red;")
на alert(this.textContent)
, чтобы отобразить предупреждение с текстом выбранного элемента