Comment obtenir le nom de la classe WebElement en Python Sélénium?

J'utilise Selenium webdriver à la ferraille un tableau extrait de la page web, écrit en JavaScript.
Je suis une itération sur une liste de lignes de la table. Chaque ligne peut être de classe différente. Je veux obtenir le nom de cette classe, afin que je puisse choisir l'action appropriée pour chaque ligne.

table_body=table.find_element_by_tag_name('tbody')
rows=table_body.find_elements_by_tag_name('tr')
for row in rows:
    if(row.GetClassName()=="date"):
        Action1()
    else:
        Action2()

Est-ce possible avec du Sélénium? Ou suggérer une autre approche.