给截图加个耗时统计

This commit is contained in:
2024-08-07 16:41:52 +08:00
parent e029d51494
commit ff71dba516
2 changed files with 23 additions and 0 deletions

View File

@@ -4,11 +4,13 @@ from playwright.async_api import TimeoutError, ViewportSize
from ..config.config import Config
from .browser import BrowserManager
from .retry import retry
from .time_it import time_it
config = get_plugin_config(Config)
@retry(exception_type=TimeoutError, reply='截图失败, 重试中')
@time_it
async def screenshot(url: str) -> bytes:
browser = await BrowserManager.get_browser()
async with (