feat: 配置项目的热修改

liteyuki/liteyuki_main/core.py

@@ -1,24 +1,23 @@
-import json
 from typing import Any
 
-import aiofiles
-import yaml
-from nonebot import require
-from nonebot.permission import SUPERUSER
+import nonebot
 from git import Repo
+from nonebot import require, get_driver
+from nonebot.permission import SUPERUSER
 
 from liteyuki.utils.config import config, load_from_yaml
-from liteyuki.utils.ly_typing import T_Bot, T_MessageEvent
-
-from liteyuki.utils.language import get_user_lang
-from liteyuki.utils.message import Markdown as md, send_markdown
-
-from .reloader import Reloader
 from liteyuki.utils.data_manager import StoredConfig, common_db
+from liteyuki.utils.language import get_user_lang
+from liteyuki.utils.ly_typing import T_Bot, T_MessageEvent
+from liteyuki.utils.message import Markdown as md, send_markdown
+from .reloader import Reloader
+from liteyuki.utils.htmlrender import launch_browser, stop_browser
 
 require("nonebot_plugin_alconna")
 from nonebot_plugin_alconna import on_alconna, Alconna, Args, Subcommand, Arparma
 
+driver = get_driver()
+
 cmd_liteyuki = on_alconna(
     Alconna(
         "liteyuki"
@@ -121,3 +120,15 @@ async def _(result: Arparma, event: T_MessageEvent, bot: T_Bot):
                     reply += f"\n{k}={v}"
                 reply += "\n```"
         await send_markdown(reply, bot, event=event)
+
+
+@driver.on_startup
+async def on_startup():
+    await launch_browser()
+    nonebot.logger.info("Browser Started.")
+
+
+@driver.on_shutdown
+async def on_shutdown():
+    await stop_browser()
+    nonebot.logger.info("Browser Stopped.")

liteyuki/liteyuki_main/webdash.py

@@ -1,91 +1,15 @@
-import nonebot
-import psutil
-from dash import Dash, Input, Output, dcc, html
-from starlette.middleware.wsgi import WSGIMiddleware
+from nonebot.adapters.onebot.v11 import MessageSegment
+from nonebot.permission import SUPERUSER
+from liteyuki.utils.htmlrender import render_html
 
-from liteyuki.utils.language import Language
-from liteyuki.utils.tools import convert_size
+from liteyuki.utils.resource import get
+from nonebot import on_command
 
-app = nonebot.get_app()
+stats = on_command("stats", priority=5, permission=SUPERUSER)
 
 
-def get_system_info():
-    cpu_percent = psutil.cpu_percent(interval=0.1)
-    memory_info = psutil.virtual_memory()
-    memory_percent = memory_info.percent
-    return {
-            "cpu_percent"   : cpu_percent,
-            "memory_percent": memory_percent
-    }
-
-
-@app.get("/system_info")
-async def system_info():
-    return get_system_info()
-
-
-lang = Language()
-dash_app = Dash(__name__)
-dash_app.layout = dash_app.layout = html.Div(children=[
-        html.H1(children=lang.get("main.monitor.title"), style={
-                "textAlign": "center"
-        }),
-
-        dcc.Graph(id="live-update-graph"),
-        dcc.Interval(
-            id="interval-component",
-            interval=1 * 1000,  # in milliseconds
-            n_intervals=0
-        )
-])
-
-
-@dash_app.callback(Output("live-update-graph", "figure"),
-                   [Input("interval-component", "n_intervals")])
-def update_graph_live(n):
-    lang = Language()
-    system_inf = get_system_info()
-    dash_app.layout = html.Div(children=[
-            html.H1(children=lang.get("main.monitor.title"), style={
-                    "textAlign": "center"
-            }),
-
-            dcc.Graph(id="live-update-graph"),
-            dcc.Interval(
-                id="interval-component",
-                interval=2 * 1000,  # in milliseconds
-                n_intervals=0
-            )
-    ])
-    mem = psutil.virtual_memory()
-    cpu_f = psutil.cpu_freq()
-    figure = {
-            "data"  : [
-                    {
-                            "x"   : [f"{cpu_f.current / 1000:.2f}GHz {psutil.cpu_count(logical=False)}c{psutil.cpu_count()}t"],
-                            "y"   : [system_inf["cpu_percent"]],
-                            "type": "bar",
-                            "name": f"{lang.get('main.monitor.cpu')} {lang.get('main.monitor.usage')}"
-
-                    },
-                    {
-                            "x"   : [f"{convert_size(mem.used, add_unit=False)}/{convert_size(mem.total)}({mem.used / mem.total * 100:.2f}%)"],
-                            "y"   : [system_inf["memory_percent"]],
-                            "type": "bar",
-                            "name": f"{lang.get('main.monitor.memory')} {lang.get('main.monitor.usage')}"
-                    },
-            ],
-            "layout": {
-                    "title": lang.get("main.monitor.description"),
-                    # "xaxis": {
-                    #         "range": [0, 10]
-                    #         },  # 设置x轴的范围
-                    "yaxis": {
-                            "range": [0, 100]
-                    },  # 设置y轴的范围
-            }
-    }
-    return figure
-
-
-app.mount("/", WSGIMiddleware(dash_app.server))
+@stats.handle()
+async def _():
+    html = get("templates/stats.html")
+    html_bytes = await render_html(open(html, "r", encoding="utf-8").read())
+    await stats.finish(MessageSegment.image(html_bytes))