update 0.0.6.2

plugins/genshin/almanac/data_source.py

@@ -1,43 +1,26 @@
-from utils.browser import get_browser
-from utils.message_builder import image
-from datetime import datetime
-from services.log import logger
-from pathlib import Path
-import os
-
-url = "https://genshin.pub"
-
-
-async def get_alc_image(path: Path):
-    date = datetime.now().date()
-    for file in os.listdir(path):
-        if f'{date}.png' != file:
-            file = path / file
-            file.unlink()
-    if f'{date}.png' in os.listdir(path):
-        return image(f'{date}.png', 'genshin/alc')
-    page = None
-    try:
-        browser = await get_browser()
-        page = await browser.new_page()
-        await page.goto(url, wait_until="networkidle", timeout=10000)
-        await page.set_viewport_size({"width": 2560, "height": 1080})
-        card = await page.query_selector('.GSAlmanacs_gs_almanacs__3qT_A')
-        await card.screenshot(path=path / f'{date}.png', timeout=100000)
-    except Exception as e:
-        logger.error(f'获取原神黄历发生错误..{type(e)}: {e}')
-    finally:
-        if page:
-            await page.close()
-    return image(f'{date}.png', 'genshin/alc')
-
-
-
-
-
-
-
-
-
-
-
+from utils.message_builder import image
+from datetime import datetime
+from pathlib import Path
+from utils.http_utils import AsyncPlaywright
+from nonebot.adapters.cqhttp import MessageSegment
+from typing import Optional
+import os
+
+url = "https://genshin.pub"
+
+
+async def get_alc_image(path: Path) -> Optional[MessageSegment]:
+    """
+    截取黄历
+    :param path: 存储路径
+    """
+    date = datetime.now().date()
+    for file in os.listdir(path):
+        if f"{date}.png" != file:
+            file = path / file
+            file.unlink()
+    if f"{date}.png" in os.listdir(path):
+        return image(f"{date}.png", "genshin/alc")
+    return await AsyncPlaywright.screenshot(
+        url, path / f"{date}.png", ".GSAlmanacs_gs_almanacs__3qT_A"
+    )