使元素高亮js代码封装

#使元素高亮
def highlight_element(driver, element):
    driver.execute_script("arguments[0].setAttribute('style', arguments[1]);", element,"background:green; border:2px solid red;")