修改截图方式

This commit is contained in:
2024-06-08 11:57:21 +08:00
parent 0ab0d11a98
commit fa05b80e61

View File

@@ -8,8 +8,8 @@ from .retry import retry
async def screenshot(url: str) -> bytes: async def screenshot(url: str) -> bytes:
browser = await BrowserManager.get_browser() browser = await BrowserManager.get_browser()
async with ( async with (
await browser.new_page(no_viewport=True, viewport={'width': 0, 'height': 0}) as page, await browser.new_page(viewport={'width': 3000, 'height': 3000}) as page,
): ):
await page.goto(url) await page.goto(url)
await page.wait_for_load_state('networkidle', timeout=5000) await page.wait_for_load_state('networkidle', timeout=5000)
return await page.screenshot(timeout=5000, full_page=True, type='png') return await page.locator('id=content').screenshot(timeout=5000, type='png')