Update main.py

This commit is contained in:
Nicolas 2024-05-22 10:45:43 -07:00
parent 3e63985e53
commit 3aa5f26627

View File

@ -33,8 +33,12 @@ async def root(body: UrlModel):
await page.goto(
body.url,
wait_until="load",
timeout=body.wait if body.wait else 15,
timeout=body.timeout if body.timeout else 15000,
)
# 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:
await page.wait_for_timeout(body.wait)
# await page.wait_for_selector("body", state="attached")
page_content = await page.content()
await context.close()
json_compatible_item_data = {"content": page_content}