From 2a39b5382b3f7985aa06b5205fa783191841506c Mon Sep 17 00:00:00 2001
From: Matt Joyce <matt.joyce@gmail.com>
Date: Sat, 1 Jun 2024 13:10:14 +1000
Subject: [PATCH] Add timeout to class and provide default.

---
 apps/playwright-service/main.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/apps/playwright-service/main.py b/apps/playwright-service/main.py
index 9cb0c4e..805c53f 100644
--- a/apps/playwright-service/main.py
+++ b/apps/playwright-service/main.py
@@ -15,6 +15,7 @@ app = FastAPI()
 class UrlModel(BaseModel):
     url: str
     wait: int = None
+    timeout: int = 15000
 
 
 browser: Browser = None
@@ -50,7 +51,7 @@ async def root(body: UrlModel):
     await page.goto(
         body.url,
         wait_until="load",
-        timeout=body.timeout if body.timeout else 15000,
+        timeout=body.timeout,
     )
     # Wait != timeout. Wait is the time to wait after the page is loaded - useful in some cases were "load" / "networkidle" is not enough
     if body.wait: