From 967f1a0e5b28ae2aeb6abe9c4c7e3c4ec06679be Mon Sep 17 00:00:00 2001
From: snowykami <snowykami@outlook.com>
Date: Sun, 1 Sep 2024 10:59:52 +0800
Subject: [PATCH] :bug: [utils]: htmlrender load error

---
 src/utils/message/html_tool.py | 70 +++++++---------------------------
 1 file changed, 14 insertions(+), 56 deletions(-)

diff --git a/src/utils/message/html_tool.py b/src/utils/message/html_tool.py
index 341d0352..68de92e8 100644
--- a/src/utils/message/html_tool.py
+++ b/src/utils/message/html_tool.py
@@ -1,20 +1,19 @@
 import os.path
-# import time
 from os import getcwd
-# from typing import Literal
 
-import aiofiles
+import aiofiles  # type: ignore
 import nonebot
-"""
-from src.utils.htmlrender import (
+from nonebot import require
+
+require("nonebot_plugin_htmlrender")
+
+from nonebot_plugin_htmlrender import (
     template_to_html,
     template_to_pic,
-    # get_new_page,
 )
-"""
-from nonebot_plugin_htmlrender import *
 from .tools import random_hex_string
 
+
 async def html2image(
         html: str,
         wait: int = 0,
@@ -39,14 +38,12 @@ async def template2html(
 
 
 async def template2image(
-    template: str,
-    templates: dict,
-    ###
-    pages=None,
-    ###
-    wait: int = 0,
-    scale_factor: float = 1,
-    debug: bool = False,
+        template: str,
+        templates: dict,
+        pages=None,
+        wait: int = 0,
+        scale_factor: float = 1,
+        debug: bool = False,
 ) -> bytes:
     """
     template -> html -> image
@@ -66,11 +63,9 @@ async def template2image(
         pages = {
                 "viewport": {
                         "width" : 1080,
-                        "height": 10
                 },
                 "base_url": f"file://{getcwd()}",
         }
-    ###
 
     template_path = os.path.dirname(template)
     template_name = os.path.basename(template)
@@ -84,18 +79,10 @@ async def template2image(
         )
         random_file_name = f"debug-{random_hex_string(6)}.html"
         async with aiofiles.open(
-            os.path.join(template_path, random_file_name), "w", encoding="utf-8"
+                os.path.join(template_path, random_file_name), "w", encoding="utf-8"
         ) as f:
             await f.write(raw_html)
         nonebot.logger.info("Debug HTML: %s" % f"{random_file_name}")
-
-    """
-    viewport={
-        "width": 1080,
-        "height": 10,
-        "deviceScaleFactor": scale_factor,
-    },"""
-
     return await template_to_pic(
         template_name=template_name,
         template_path=template_path,
@@ -107,32 +94,3 @@ async def template2image(
         device_scale_factor=scale_factor
         ###
     )
-    
-async def url2image(
-        url: str,
-        wait: int = 0,
-        scale_factor: float = 1,
-        type: str = "png",
-        quality: int = 100,
-        **kwargs
-) -> bytes:
-    """
-    Args:
-        quality:
-        type:
-        url: str: URL
-        wait: int: 等待时间
-        scale_factor: float: 缩放因子
-        **kwargs: page 参数
-    Returns:
-        图片二进制数据
-    """
-    async with get_new_page(scale_factor) as page:
-        await page.goto(url)
-        await page.wait_for_timeout(wait)
-        return await page.screenshot(
-            full_page=True,
-            type=type,
-            quality=quality
-        )
-