chore: Add config options for Playwright wait_until and default timeout

superset/config.py

@@ -682,6 +682,11 @@ class D3Format(TypedDict, total=False):
 SCREENSHOT_WAIT_FOR_ERROR_MODAL_VISIBLE = 5
 # Max time to wait for error message modal to close, in seconds
 SCREENSHOT_WAIT_FOR_ERROR_MODAL_INVISIBLE = 5
+# event that Playwright waits for when loading a new page
+# possible values: "load", "commit", "domcontentloaded", "networkidle"
+SCREENSHOT_PLAYWRIGHT_WAIT_UNTIL = "load"
+# default timeout for Playwright browser context for all operations
+SCREENSHOT_PLAYWRIGHT_DEFAULT_TIMEOUT = 10000
 
 # ---------------------------------------------------
 # Image and file configuration

superset/utils/webdriver.py

@@ -49,7 +49,7 @@
     from playwright.sync_api import (
         BrowserContext,
         ElementHandle,
-        Error,
+        Error as PlaywrightError,
         Page,
         sync_playwright,
         TimeoutError as PlaywrightTimeout,
@@ -140,9 +140,9 @@ def find_unexpected_errors(page: Page) -> list[str]:
                         "(node, error_html) => node.innerHtml = error_html",
                         [error_as_html],
                     )
-                except Error:
+                except PlaywrightError:
                     logger.exception("Failed to update error messages using alert_div")
-        except Error:
+        except PlaywrightError:
             logger.exception("Failed to capture unexpected errors")
 
         return error_messages
@@ -161,9 +161,14 @@ def get_screenshot(self, url: str, element_name: str, user: User) -> bytes | Non
                 },
                 device_scale_factor=pixel_density,
             )
+            context.set_default_timeout(
+                current_app.config["SCREENSHOT_PLAYWRIGHT_DEFAULT_TIMEOUT"]
+            )
             self.auth(user, context)
             page = context.new_page()
-            page.goto(url)
+            page.goto(
+                url, wait_until=current_app.config["SCREENSHOT_PLAYWRIGHT_WAIT_UNTIL"]
+            )
             img: bytes | None = None
             selenium_headstart = current_app.config["SCREENSHOT_SELENIUM_HEADSTART"]
             logger.debug("Sleeping for %i seconds", selenium_headstart)
@@ -236,14 +241,9 @@ def get_screenshot(self, url: str, element_name: str, user: User) -> bytes | Non
             except PlaywrightTimeout:
                 # raise again for the finally block, but handled above
                 pass
-            except StaleElementReferenceException:
-                logger.exception(
-                    "Selenium got a stale element while requesting url %s",
-                    url,
-                )
-            except WebDriverException:
+            except PlaywrightError:
                 logger.exception(
-                    "Encountered an unexpected error when requeating url %s", url
+                    "Encountered an unexpected error when requesting url %s", url
                 )
             return img