update check_update

plugins/check_zhenxun_update/__init__.py

@@ -1,124 +1,124 @@
-# from nonebot.adapters.onebot.v11 import Bot
-# from nonebot.permission import SUPERUSER
-# from nonebot import on_command
-# from .data_source import check_update, get_latest_version_data
-# from services.log import logger
-# from utils.utils import scheduler, get_bot
-# from pathlib import Path
-# from configs.config import Config
-# from nonebot.rule import to_me
-# from nonebot.params import ArgStr
-# import platform
-# import os
-#
-#
-# __zx_plugin_name__ = "自动更新 [Superuser]"
-# __plugin_usage__ = """
-# usage：
-#     检查更新真寻最新版本，包括了自动更新
-#     指令：
-#         检查更新真寻
-#         重启
-# """.strip()
-# __plugin_des__ = "就算是真寻也会成长的"
-# __plugin_cmd__ = ["检查更新真寻", "重启"]
-# __plugin_version__ = 0.1
-# __plugin_author__ = "HibiKier"
-# __plugin_configs__ = {
-#     "AUTO_UPDATE_ZHENXUN": {
-#         "value": False,
-#         "help": "真寻是否自动检查更新",
-#         "default": False,
-#     }
-# }
-#
-# update_zhenxun = on_command("检查更新真寻", permission=SUPERUSER, priority=1, block=True)
-#
-# restart = on_command(
-#     "重启",
-#     aliases={"restart"},
-#     permission=SUPERUSER,
-#     rule=to_me(),
-#     priority=1,
-#     block=True,
-# )
-#
-#
-# @update_zhenxun.handle()
-# async def _(bot: Bot):
-#     try:
-#         code, error = await check_update(bot)
-#         if error:
-#             logger.error(f"更新真寻未知错误 {error}")
-#             await bot.send_private_msg(
-#                 user_id=int(list(bot.config.superusers)[0]), message=f"更新真寻未知错误 {error}"
-#             )
-#     except Exception as e:
-#         logger.error(f"更新真寻未知错误 {type(e)}：{e}")
-#         await bot.send_private_msg(
-#             user_id=int(list(bot.config.superusers)[0]),
-#             message=f"更新真寻未知错误 {type(e)}：{e}",
-#         )
-#     else:
-#         if code == 200:
-#             await bot.send_private_msg(
-#                 user_id=int(list(bot.config.superusers)[0]), message=f"更新完毕，请重启真寻...."
-#             )
-#
-#
-# @restart.handle()
-# async def _():
-#     if str(platform.system()).lower() == "windows":
-#         await restart.finish("暂无windows重启脚本...")
-#
-#
-# @restart.got("flag", prompt="确定是否重启真寻？（重启失败咱们将失去联系，请谨慎！）")
-# async def _(flag: str = ArgStr("flag")):
-#     if flag.lower() in ["true", "是", "好", "确定", "确定是"]:
-#         await restart.send("开始重启真寻..请稍等...")
-#         open("is_restart", "w")
-#         os.system("./restart.sh")
-#     else:
-#         await restart.send("已取消操作...")
-#
-#
-# @scheduler.scheduled_job(
-#     "cron",
-#     hour=12,
-#     minute=0,
-# )
-# async def _():
-#     if Config.get_config("check_zhenxun_update", "AUTO_UPDATE_ZHENXUN"):
-#         _version = "v0.0.0"
-#         _version_file = Path() / "__version__"
-#         if _version_file.exists():
-#             _version = (
-#                 open(_version_file, "r", encoding="utf8")
-#                 .readline()
-#                 .split(":")[-1]
-#                 .strip()
-#             )
-#         data = await get_latest_version_data()
-#         if data:
-#             latest_version = data["name"]
-#             if _version != latest_version:
-#                 bot = get_bot()
-#                 await bot.send_private_msg(
-#                     user_id=int(list(bot.config.superusers)[0]),
-#                     message=f"检测到真寻版本更新\n"
-#                     f"当前版本：{_version}，最新版本：{latest_version}",
-#                 )
-#                 # try:
-#                 #     code = await check_update(bot)
-#                 # except Exception as e:
-#                 #     logger.error(f"更新真寻未知错误 {type(e)}：{e}")
-#                 #     await bot.send_private_msg(
-#                 #         user_id=int(list(bot.config.superusers)[0]),
-#                 #         message=f"更新真寻未知错误 {type(e)}：{e}\n",
-#                 #     )
-#                 # else:
-#                 #     if code == 200:
-#                 #         await bot.send_private_msg(
-#                 #             user_id=int(list(bot.config.superusers)[0]),
-#                 #             message=f"更新完毕，请重启真寻....",
-#                 #         )
+from nonebot.adapters.onebot.v11 import Bot
+from nonebot.permission import SUPERUSER
+from nonebot import on_command
+from .data_source import check_update, get_latest_version_data
+from services.log import logger
+from utils.utils import scheduler, get_bot
+from pathlib import Path
+from configs.config import Config
+from nonebot.rule import to_me
+from nonebot.params import ArgStr
+import platform
+import os
+
+
+__zx_plugin_name__ = "自动更新 [Superuser]"
+__plugin_usage__ = """
+usage：
+    检查更新真寻最新版本，包括了自动更新
+    指令：
+        检查更新真寻
+        重启
+""".strip()
+__plugin_des__ = "就算是真寻也会成长的"
+__plugin_cmd__ = ["检查更新真寻", "重启"]
+__plugin_version__ = 0.1
+__plugin_author__ = "HibiKier"
+__plugin_configs__ = {
+    "AUTO_UPDATE_ZHENXUN": {
+        "value": False,
+        "help": "真寻是否自动检查更新",
+        "default": False,
+    }
+}
+
+update_zhenxun = on_command("检查更新真寻", permission=SUPERUSER, priority=1, block=True)
+
+restart = on_command(
+    "重启",
+    aliases={"restart"},
+    permission=SUPERUSER,
+    rule=to_me(),
+    priority=1,
+    block=True,
+)
+
+
+@update_zhenxun.handle()
+async def _(bot: Bot):
+    try:
+        code, error = await check_update(bot)
+        if error:
+            logger.error(f"更新真寻未知错误 {error}")
+            await bot.send_private_msg(
+                user_id=int(list(bot.config.superusers)[0]), message=f"更新真寻未知错误 {error}"
+            )
+    except Exception as e:
+        logger.error(f"更新真寻未知错误 {type(e)}：{e}")
+        await bot.send_private_msg(
+            user_id=int(list(bot.config.superusers)[0]),
+            message=f"更新真寻未知错误 {type(e)}：{e}",
+        )
+    else:
+        if code == 200:
+            await bot.send_private_msg(
+                user_id=int(list(bot.config.superusers)[0]), message=f"更新完毕，请重启真寻...."
+            )
+
+
+@restart.handle()
+async def _():
+    if str(platform.system()).lower() == "windows":
+        await restart.finish("暂无windows重启脚本...")
+
+
+@restart.got("flag", prompt="确定是否重启真寻？（重启失败咱们将失去联系，请谨慎！）")
+async def _(flag: str = ArgStr("flag")):
+    if flag.lower() in ["true", "是", "好", "确定", "确定是"]:
+        await restart.send("开始重启真寻..请稍等...")
+        open("is_restart", "w")
+        os.system("./restart.sh")
+    else:
+        await restart.send("已取消操作...")
+
+
+@scheduler.scheduled_job(
+    "cron",
+    hour=12,
+    minute=0,
+)
+async def _():
+    if Config.get_config("check_zhenxun_update", "AUTO_UPDATE_ZHENXUN"):
+        _version = "v0.0.0"
+        _version_file = Path() / "__version__"
+        if _version_file.exists():
+            _version = (
+                open(_version_file, "r", encoding="utf8")
+                .readline()
+                .split(":")[-1]
+                .strip()
+            )
+        data = await get_latest_version_data()
+        if data:
+            latest_version = data["name"]
+            if _version != latest_version:
+                bot = get_bot()
+                await bot.send_private_msg(
+                    user_id=int(list(bot.config.superusers)[0]),
+                    message=f"检测到真寻版本更新\n"
+                    f"当前版本：{_version}，最新版本：{latest_version}",
+                )
+                # try:
+                #     code = await check_update(bot)
+                # except Exception as e:
+                #     logger.error(f"更新真寻未知错误 {type(e)}：{e}")
+                #     await bot.send_private_msg(
+                #         user_id=int(list(bot.config.superusers)[0]),
+                #         message=f"更新真寻未知错误 {type(e)}：{e}\n",
+                #     )
+                # else:
+                #     if code == 200:
+                #         await bot.send_private_msg(
+                #             user_id=int(list(bot.config.superusers)[0]),
+                #             message=f"更新完毕，请重启真寻....",
+                #         )