From 45ffaf4b4bffaa4d00d73e55a034f43da5fee42e Mon Sep 17 00:00:00 2001 From: Dmitry Afanasyev Date: Sat, 27 Aug 2022 18:51:43 +0300 Subject: [PATCH] find elements moved to try except --- app/core/parse_web.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/app/core/parse_web.py b/app/core/parse_web.py index 4969582..ab8f6d1 100644 --- a/app/core/parse_web.py +++ b/app/core/parse_web.py @@ -57,15 +57,15 @@ def parse_site(url: str, message: str, driver: RemoteWebDriver | None = None) -> return 'Что-то пошло не так. :( Драйвер Firefox не сконфигурирован.' driver.get(url) time.sleep(1) - elements = driver.find_elements( - by='class name', value='masstransit-vehicle-snippet-view' - ) bus_300, bus_t19 = 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( 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( by='class name', value='masstransit-prognoses-view__title-text' ) - except NoSuchElementException: - pass - except StaleElementReferenceException: - pass + except NoSuchElementException: + pass + except StaleElementReferenceException: + pass answer = f'{message}\n\n' if not all([bus_300, bus_t19]) or not all([bus_300_arrival, bus_t19_arrival]): return 'Автобусов 300 или Т19 не найдено. \n\nСмотри на карте :)'