Save timeout errors to log

waternsw_grabber/waternsw_grabber.py

@@ -51,8 +51,12 @@ def wait_for_element(driver, by, x, timeout=180):
     Raises
         TimeoutException if element does not load within timeout period
     """
+    try:
         element_present = EC.presence_of_element_located((by, x))
         WebDriverWait(driver, timeout).until(element_present)
+    except TimeoutError:
+        raise TimeoutError(
+            'Request timed out on {}. Try again later?'.format(bore_id))
 
 
 def get_telemetered_bore(driver, bore_id, start_date, end_date):
@@ -214,7 +218,7 @@ def telemetered_bore_downloader(bore_ids, start_date, end_date, download_dir):
         pbar.set_description(bore_id)
         try:
             get_telemetered_bore(driver, bore_id, start_date, end_date)
-        except ValueError as e:
+        except (ValueError, TimeoutError) as e:
             logging.error(e)
 
     # Tidy up console after tqdm