Add timeout to class and provide default.

This commit is contained in:
Matt Joyce 2024-06-01 13:10:14 +10:00
parent 03ba4a9fa7
commit 2a39b5382b
1 changed files with 2 additions and 1 deletions

View File

@ -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: