空调调温优化与排行榜，浏览器添加本地HTML支持

konabot/plugins/air_conditioner/__init__.py

@@ -1,13 +1,16 @@
 from io import BytesIO
 from typing import Optional, Union
+import cv2
 from nonebot.adapters import Event as BaseEvent
 from nonebot.adapters.console.event import MessageEvent as ConsoleMessageEvent
 from nonebot.adapters.discord.event import MessageEvent as DiscordMessageEvent
 from nonebot_plugin_alconna import Alconna, AlconnaMatcher, Args, UniMessage, on_alconna
 from PIL import Image
+import numpy as np
 from konabot.common.longtask import DepLongTaskTarget
 from konabot.common.path import ASSETS_PATH
-from konabot.plugins.air_conditioner.ac import AirConditioner, generate_ac_image
+from konabot.common.web_render import WebRenderer
+from konabot.plugins.air_conditioner.ac import AirConditioner, CrashType, generate_ac_image, wiggle_transform
 
 import random
 import math
@@ -73,17 +76,20 @@ async def _(event: BaseEvent, target: DepLongTaskTarget):
     await send_ac_image(evt, ac)
 
 evt = on_alconna(Alconna(
-    "空调升温"
+    "空调升温",
+    Args["temp?", Optional[Union[int, float]]]  # 可选参数，升温的度数，默认为1
 ), use_cmd_start=True, use_cmd_sep=False, skip_for_unmatch=True)
 
 @evt.handle()
-async def _(event: BaseEvent, target: DepLongTaskTarget):
+async def _(event: BaseEvent, target: DepLongTaskTarget, temp: Optional[Union[int, float]] = 1):
+    if temp <= 0:
+        return
     id = target.channel_id
     ac = get_ac(id)
     if not ac.on or ac.burnt == True or ac.frozen == True:
         await send_ac_image(evt, ac)
         return
-    ac.temperature += 1
+    ac.temperature += temp
     if ac.temperature > 40:
         # 根据温度随机出是否爆炸，40度开始，呈指数增长
         possibility = -math.e ** ((40-ac.temperature) / 50) + 1
@@ -91,30 +97,37 @@ async def _(event: BaseEvent, target: DepLongTaskTarget):
             # 打开爆炸图片
             with open(ASSETS_PATH / "img" / "other" / "boom.jpg", "rb") as f:
                 output = BytesIO()
-                Image.open(f).save(output, format="GIF")
+                # 爆炸抖动
+                frames = wiggle_transform(np.array(Image.open(f)), intensity=5)
+                pil_frames = [Image.fromarray(frame) for frame in frames]
+                pil_frames[0].save(output, format="GIF", save_all=True, append_images=pil_frames[1:], loop=0, duration=35, disposal=2)
+                output.seek(0)
                 await evt.send(await UniMessage().image(raw=output).export())
-            ac.burnt = True
+            ac.broke_ac(CrashType.BURNT)
             await evt.send("太热啦，空调炸了！")
             return
     await send_ac_image(evt, ac)
 
 evt = on_alconna(Alconna(
-    "空调降温"
+    "空调降温",
+    Args["temp?", Optional[Union[int, float]]]  # 可选参数，降温的度数，默认为1
 ), use_cmd_start=True, use_cmd_sep=False, skip_for_unmatch=True)
 
 @evt.handle()
-async def _(event: BaseEvent, target: DepLongTaskTarget):
+async def _(event: BaseEvent, target: DepLongTaskTarget, temp: Optional[Union[int, float]] = 1):
+    if temp <= 0:
+        return
     id = target.channel_id
     ac = get_ac(id)
     if not ac.on or ac.burnt == True or ac.frozen == True:
         await send_ac_image(evt, ac)
         return
-    ac.temperature -= 1
+    ac.temperature -= temp
     if ac.temperature < 0:
         # 根据温度随机出是否冻结，0度开始，呈指数增长
         possibility = -math.e ** (ac.temperature / 50) + 1
         if random.random() < possibility:
-            ac.frozen = True
+            ac.broke_ac(CrashType.FROZEN)
     await send_ac_image(evt, ac)
 
 evt = on_alconna(Alconna(
@@ -126,4 +139,24 @@ async def _(event: BaseEvent, target: DepLongTaskTarget):
     id = target.channel_id
     ac = get_ac(id)
     ac.change_ac()
-    await send_ac_image(evt, ac)
+    await send_ac_image(evt, ac)
+
+evt = on_alconna(Alconna(
+    "空调炸炸排行榜",
+), use_cmd_start=True, use_cmd_sep=False, skip_for_unmatch=True)
+
+@evt.handle()
+async def _(event: BaseEvent, target: DepLongTaskTarget):
+    id = target.channel_id
+    ac = get_ac(id)
+    number, ranking = ac.get_crashes_and_ranking()
+    params = {
+        "number": number,
+        "ranking": ranking
+    }
+    image = await WebRenderer.render_file(
+        file_path=ASSETS_PATH / "webpage" / "ac" / "index.html",
+        target=".box",
+        params=params
+    )
+    await evt.send(await UniMessage().image(raw=image).export())