feat: enhance setupStats function with retry logic and improve error handling; update entrypoint to run main script as PID 1

main.py

@@ -181,19 +181,22 @@ def convert_to_bytes(down, downUnit):
     return float(down) * (1024 ** {'B': 0, 'KB': 1, 'MB': 2, 'GB': 3}[downUnit])
 
 @yaspin()
-def setupStats(driver: webdriver.Remote, url: str):
+def setupStats(driver: webdriver.Remote, url: str, retries: int = 5) -> webdriver.Remote:
     logger.log(logging.INFO, 'Setting up stats.')
     actions = ActionChains(driver)
     wait = WebDriverWait(driver, 30, poll_frequency=0.2)
 
-    driver.get(url)
-
-    try:
-        wait.until(ec.presence_of_element_located((By.CLASS_NAME, 'vjs-big-play-button')))
-    except Exception:
-        logger.error('Timeout while waiting for the big play button to be present.')
-        driver.quit()
-        raise SystemExit(1)
+    for attempt in range(retries):
+        driver.get(url)
+        try:
+            wait.until(ec.presence_of_element_located((By.CLASS_NAME, 'vjs-big-play-button')))
+            break
+        except Exception:
+            logger.error(f'Timeout while waiting for the big play button to be present. Attempt {attempt + 1} of {retries}')
+            if attempt == retries - 1:
+                logger.error('Timeout limit reached. Exiting.')
+                driver.quit()
+                raise SystemExit(1)
 
     actions.click(driver.find_element(By.CLASS_NAME ,'video-js')).perform()
     wait.until(ec.visibility_of_element_located((By.CLASS_NAME, 'vjs-control-bar')))