✨ 新增更新本插件的命令

Author: forchannot

__init__.py

@@ -1,31 +1,29 @@
 from typing import Union
 
-from nonebot.internal.matcher import Matcher
-from nonebot.params import CommandArg
-
-from .captcha.captcha import gain_num
-from .handle.ssbq_handler import handle_ssbq, sub_list, get_subs
-from .handle.coin_handle import mhy_bbs_coin, bbs_auto_coin
-from .handle.sign_handle import mhy_bbs_sign, bbs_auto_sign
-from .handle.sr_sign_handle import sr_bbs_auto_sign
-from .config.config import config
-from .utils.logger import Logger
-from .web import web_api, web_page  # noqa
-
-from LittlePaimon.database import MihoyoBBSSub, PrivateCookie, DailyNoteSub
-from LittlePaimon.utils.message import CommandPlayer, CommandUID, CommandSwitch
-
-from nonebot import on_command, Bot
-from nonebot.typing import T_State
+from LittlePaimon.database import DailyNoteSub, MihoyoBBSSub, PrivateCookie
+from LittlePaimon.utils.message import CommandPlayer, CommandSwitch, CommandUID
+from nonebot import Bot, on_command
 from nonebot.adapters.onebot.v11 import (
-    Message,
     GroupMessageEvent,
+    Message,
     PrivateMessageEvent,
 )
+from nonebot.internal.matcher import Matcher
+from nonebot.params import CommandArg
 from nonebot.permission import SUPERUSER
 from nonebot.plugin import PluginMetadata
 from nonebot.rule import to_me
+from nonebot.typing import T_State
 
+from .captcha.captcha import gain_num
+from .config.config import config
+from .handle.coin_handle import bbs_auto_coin, mhy_bbs_coin
+from .handle.sign_handle import bbs_auto_sign, mhy_bbs_sign
+from .handle.sr_sign_handle import sr_bbs_auto_sign
+from .handle.ssbq_handler import get_subs, handle_ssbq, sub_list
+from .utils.logger import Logger
+from .utils.update import do_update
+from .web import web_api, web_page  # noqa
 
 __plugin_meta__ = PluginMetadata(
     name="加强小派蒙验证",
@@ -144,6 +142,20 @@
         "pm_priority": 3,
     },
 )
+update_self = on_command(
+    "验证签到插件更新",
+    aliases={"签到插件更新"},
+    priority=10,
+    block=True,
+    permission=SUPERUSER,
+    rule=to_me(),
+    state={
+        "pm_name": "验证签到插件更新",
+        "pm_description": "更新验证签到插件",
+        "pm_usage": "@Bot 验证签到插件更新",
+        "pm_priority": 10,
+    },
+)
 
 signing_list = []
 coin_getting_list = []
@@ -520,3 +532,10 @@ async def _(
                     f"UID{sr_uid}尚未开启星铁自动签到，无需关闭！", at_sender=True
                 )
     await sr_sign.finish(config.hfu, at_sender=True)
+
+
+@update_self.handle()
+async def _():
+    await update_self.send("正在更新验证签到插件，请稍后", at_sender=True)
+    result = await do_update()
+    await update_self.finish(result, at_sender=True)

utils/update.py

@@ -0,0 +1,35 @@
+from pathlib import Path
+
+import git
+from git.exc import GitCommandError, InvalidGitRepositoryError
+from LittlePaimon.utils import NICKNAME
+from nonebot.utils import run_sync
+
+from .logger import Logger
+
+
+@run_sync
+def do_update():
+    try:
+        repo = git.Repo(Path(__file__).parent.parent.absolute())
+    except InvalidGitRepositoryError:
+        return "没有发现git仓库，无法通过git更新，请手动下载最新版本的文件进行替换。"
+    Logger.info("验证签到插件更新", "开始执行<m>git pull</m>更新操作")
+    origin = repo.remotes.origin
+    try:
+        origin.pull()
+        repo_msg = (
+            repo.head.commit.message.replace(":bug:", "🐛")
+            .replace(":sparkles:", "✨")
+            .replace(":memo:", "📝")
+            .replace(":art:", "🎨")
+        )
+        msg = f"签到插件更新完成\n最新更新日志为：\n{repo_msg}\n可使用命令[@bot 重启]重启{NICKNAME}"
+    except GitCommandError as e:
+        if "timeout" in e.stderr or "unable to access" in e.stderr:
+            msg = "签到插件更新失败，连接git仓库超时，请重试或修改源为代理源后再重试。"
+        elif "Your local changes" in e.stderr:
+            msg = f"签到插件更新失败，本地修改过文件导致冲突，请解决冲突后再更新。\n{e.stderr}"
+        else:
+            msg = f"签到插件更新失败，错误信息：{e.stderr}，请尝试手动进行更新"
+    return msg