find elements moved to try except

This commit is contained in:
Dmitry Afanasyev 2022-08-27 18:51:43 +03:00
parent 4c49b58546
commit 45ffaf4b4b

View File

@ -57,15 +57,15 @@ def parse_site(url: str, message: str, driver: RemoteWebDriver | None = None) ->
return 'Что-то пошло не так. :( Драйвер Firefox не сконфигурирован.' return 'Что-то пошло не так. :( Драйвер Firefox не сконфигурирован.'
driver.get(url) driver.get(url)
time.sleep(1) time.sleep(1)
elements = driver.find_elements(
by='class name', value='masstransit-vehicle-snippet-view'
)
bus_300, bus_t19 = None, None bus_300, bus_t19 = None, None
bus_300_arrival, bus_t19_arrival = None, None bus_300_arrival, bus_t19_arrival = None, None
for element in elements: try:
try: elements = driver.find_elements(
by='class name', value='masstransit-vehicle-snippet-view'
)
for element in elements:
bus_300 = element.find_element( bus_300 = element.find_element(
by='css selector', value='[aria-label="300"]' by='css selector', value='[aria-label="300"]'
) )
@ -78,10 +78,10 @@ def parse_site(url: str, message: str, driver: RemoteWebDriver | None = None) ->
bus_t19_arrival = element.find_element( bus_t19_arrival = element.find_element(
by='class name', value='masstransit-prognoses-view__title-text' by='class name', value='masstransit-prognoses-view__title-text'
) )
except NoSuchElementException: except NoSuchElementException:
pass pass
except StaleElementReferenceException: except StaleElementReferenceException:
pass pass
answer = f'{message}\n\n' answer = f'{message}\n\n'
if not all([bus_300, bus_t19]) or not all([bus_300_arrival, bus_t19_arrival]): if not all([bus_300, bus_t19]) or not all([bus_300_arrival, bus_t19_arrival]):
return 'Автобусов 300 или Т19 не найдено. \n\nСмотри на карте :)' return 'Автобусов 300 или Т19 не найдено. \n\nСмотри на карте :)'