A-Minos/nonebot-plugin-tetris-stats
1
0
Fork 0
mirror of https://github.com/A-Minos/nonebot-plugin-tetris-stats.git synced 2026-03-05 05:36:54 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🐛 优化等待逻辑，修复截图爆炸

Browse Source
This commit is contained in:
shoucandanghehe
2025-04-10 04:36:06 +08:00
parent 4983fa7c52
commit 55c4b0ad41
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
nonebot_plugin_tetris_stats/utils/screenshot.py
View File

@@ -16,6 +16,7 @@ async def screenshot(url: str) -> bytes:
context = await BrowserManager.get_context('screenshot', factory=context_factory) context = await BrowserManager.get_context('screenshot', factory=context_factory)
async with await context.new_page() as page: async with await context.new_page() as page:
await page.goto(url) await page.goto(url)
await page.wait_for_selector('#content')
size: ViewportSize = await page.evaluate(""" size: ViewportSize = await page.evaluate("""
() => { () => {
const element = document.querySelector('#content'); const element = document.querySelector('#content');
@@ -26,5 +27,4 @@ async def screenshot(url: str) -> bytes:
}; };
""") """)
await page.set_viewport_size(size) await page.set_viewport_size(size)
await page.wait_for_load_state('networkidle')
return await page.locator('id=content').screenshot(animations='disabled', timeout=5000, type='png') return await page.locator('id=content').screenshot(animations='disabled', timeout=5000, type='png')