mirror of
https://github.com/A-Minos/nonebot-plugin-tetris-stats.git
synced 2026-03-05 05:36:54 +08:00
a116f9901c
* ⬆️ auto update by pre-commit hooks updates: - [github.com/astral-sh/ruff-pre-commit: v0.7.3 → v0.8.1](https://github.com/astral-sh/ruff-pre-commit/compare/v0.7.3...v0.8.1) * 🚨 auto fix by pre-commit hooks * 🚨 fix ruff --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: shoucandanghehe <wallfjjd@gmail.com>
31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
from playwright.async_api import BrowserContext, TimeoutError, ViewportSize # noqa: A004
|
|
|
|
from ..config.config import config
|
|
from .browser import BrowserManager
|
|
from .retry import retry
|
|
from .time_it import time_it
|
|
|
|
|
|
async def context_factory() -> BrowserContext:
|
|
return await (await BrowserManager.get_browser()).new_context(device_scale_factor=config.tetris.screenshot_quality)
|
|
|
|
|
|
@retry(exception_type=TimeoutError, reply='截图失败, 重试中')
|
|
@time_it
|
|
async def screenshot(url: str) -> bytes:
|
|
context = await BrowserManager.get_context('screenshot', factory=context_factory)
|
|
async with await context.new_page() as page:
|
|
await page.goto(url)
|
|
size: ViewportSize = await page.evaluate("""
|
|
() => {
|
|
const element = document.querySelector('#content');
|
|
return {
|
|
width: element.offsetWidth,
|
|
height: element.offsetHeight,
|
|
};
|
|
};
|
|
""")
|
|
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')
|