если вы используете POM, вам нужно создать отдельный класс со всеми связанными элементами на странице. Кроме того, вы можете использовать ниже xpath, чтобы найти количество строк, связанных в таблице. Ниже приведен ссылочный код, который может вам понадобиться
row_count =len(WebDriverWait(driver, 30).until(EC.presence_of_all_elements_located((By.XPATH, "//table[@id='example']/tbody/tr"))))
print row_count
Также вам необходимо использовать приведенный ниже код для проверки наличия следующей кнопки или ее отсутствия. Если затем нажмите на него, пока не перейдете на последнюю страницу.
while flag:
try:
element = wait.until(EC.element_to_be_clickable((By.XPATH, "//a[contains(text(),'Next')]")))
if (element != 0):
element.click()
except TimeoutException as ex:
print "It is all good, no element there"
Вам необходимо добавить ниже import
from selenium import webdriver
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait as Wait
from selenium.webdriver.support.ui import WebDriverWait
from selenium.common.exceptions import TimeoutException