Какой из двух элементов появился в webdriver selenium

Question

Я нажимаю кнопку в селене, и тогда у меня есть два возможных сценария ios: появляется элемент с class = "success" или элемент с class = "error". Как получить, какой элемент появился? Я пробовал это:

WebDriverWait(driver, 5).until(
lambda driver : driver.find_element_by_css_selector("[class='success'], [class='error']"))

, но он возвращает что-то вроде этого, что не дает мне много информации о появившемся элементе:

<selenium.webdriver.remote.webelement.WebElement (session="832d019b49d34662dfa548c823cc1d2e", element="e5a38fb7-88c1-4bd6-93b5-b48a1578d871")>

Answer 1

Если вы хотите узнать, что такое найденный элемент, вы должны позвонить element.get_attribute("class"), и вы получите информацию о классе элемента.

В основном вы могли бы

WebDriverWait(driver, 5).until(
lambda driver : driver.find_element_by_css_selector("[class='success'], [class='error']")).get_attribute('class')