feat✨: 更新内置插件

zhenxun/builtin_plugins/scheduler/__init__.py

@@ -0,0 +1,5 @@
+from pathlib import Path
+
+import nonebot
+
+nonebot.load_plugins(str(Path(__file__).parent.resolve()))

zhenxun/builtin_plugins/scheduler/auto_backup.py

@@ -0,0 +1,62 @@
+import shutil
+from pathlib import Path
+
+from nonebot_plugin_apscheduler import scheduler
+
+from zhenxun.configs.config import Config
+from zhenxun.services.log import logger
+
+Config.add_plugin_config(
+    "_backup",
+    "BACKUP_FLAG",
+    True,
+    help="是否开启文件备份",
+    default_value=True,
+    type=bool,
+)
+
+Config.add_plugin_config(
+    "_backup",
+    "BACKUP_DIR_OR_FILE",
+    [
+        "data/black_word",
+        "data/configs",
+        "data/statistics",
+        "data/word_bank",
+        "data/manager",
+        "configs",
+    ],
+    help="备份的文件夹或文件",
+    default_value=[],
+    type=list[str],
+)
+
+
+# 自动备份
+@scheduler.scheduled_job(
+    "cron",
+    hour=3,
+    minute=25,
+)
+async def _():
+    if Config.get_config("_backup", "BACKUP_FLAG"):
+        _backup_path = Path() / "backup"
+        _backup_path.mkdir(exist_ok=True, parents=True)
+        if backup_dir_or_file := Config.get_config("_backup", "BACKUP_DIR_OR_FILE"):
+            for path_file in backup_dir_or_file:
+                try:
+                    path = Path(path_file)
+                    _p = _backup_path / path_file
+                    if path.exists():
+                        if path.is_dir():
+                            if _p.exists():
+                                shutil.rmtree(_p, ignore_errors=True)
+                            shutil.copytree(path_file, _p)
+                        else:
+                            if _p.exists():
+                                _p.unlink()
+                            shutil.copy(path_file, _p)
+                        logger.debug(f"已完成自动备份：{path_file}", "自动备份")
+                except Exception as e:
+                    logger.error(f"自动备份文件 {path_file} 发生错误", "自动备份", e=e)
+        logger.info("自动备份成功...", "自动备份")

zhenxun/builtin_plugins/scheduler/auto_update_group.py

@@ -0,0 +1,68 @@
+import nonebot
+from nonebot_plugin_apscheduler import scheduler
+
+from zhenxun.models.friend_user import FriendUser
+from zhenxun.models.group_console import GroupConsole
+from zhenxun.services.log import logger
+
+# TODO: 其他平台更新
+
+
+# 自动更新群组信息
+@scheduler.scheduled_job(
+    "cron",
+    hour=3,
+    minute=1,
+)
+async def _():
+    bots = nonebot.get_bots()
+    _used_group = []
+    for bot in bots.values():
+        try:
+            group_list = await bot.get_group_list()
+            gl = [g["group_id"] for g in group_list if g["group_id"] not in _used_group]
+            for g in gl:
+                _used_group.append(g)
+                group_info = await bot.get_group_info(group_id=g)
+                await GroupConsole.update_or_create(
+                    group_id=str(group_info["group_id"]),
+                    defaults={
+                        "group_name": group_info["group_name"],
+                        "max_member_count": group_info["max_member_count"],
+                        "member_count": group_info["member_count"],
+                        "group_flag": 1,
+                    },
+                )
+                logger.debug("自动更新群组信息成功", "自动更新群组", group_id=g)
+        except Exception as e:
+            logger.error(f"Bot: {bot.self_id} 自动更新群组信息", e=e)
+    logger.info("自动更新群组成员信息成功...")
+
+
+# 自动更新好友信息
+@scheduler.scheduled_job(
+    "cron",
+    hour=3,
+    minute=1,
+)
+async def _():
+    bots = nonebot.get_bots()
+    for key in bots:
+        try:
+            bot = bots[key]
+            fl = await bot.get_friend_list()
+            for f in fl:
+                if FriendUser.exists(user_id=str(f["user_id"])):
+                    await FriendUser.create(
+                        user_id=str(f["user_id"]), user_name=f["nickname"]
+                    )
+                    logger.debug(
+                        f"更新好友信息成功", "自动更新好友", session=f["user_id"]
+                    )
+                else:
+                    logger.debug(
+                        f"好友信息已存在", "自动更新好友", session=f["user_id"]
+                    )
+        except Exception as e:
+            logger.error(f"自动更新好友信息错误", "自动更新好友", e=e)
+    logger.info("自动更新好友信息成功...")

zhenxun/builtin_plugins/scheduler/morning.py

@@ -0,0 +1,28 @@
+from nonebot_plugin_apscheduler import scheduler
+
+# TODO: 消息发送
+
+# # 早上好
+# @scheduler.scheduled_job(
+#     "cron",
+#     hour=6,
+#     minute=1,
+# )
+# async def _():
+#     img = image(IMAGE_PATH / "zhenxun" / "zao.jpg")
+#     await broadcast_group("[[_task|zwa]]早上好" + img, log_cmd="被动早晚安")
+#     logger.info("每日早安发送...")
+
+
+# # 睡觉了
+# @scheduler.scheduled_job(
+#     "cron",
+#     hour=23,
+#     minute=59,
+# )
+# async def _():
+#     img = image(IMAGE_PATH / "zhenxun" / "sleep.jpg")
+#     await broadcast_group(
+#         f"[[_task|zwa]]{NICKNAME}要睡觉了，你们也要早点睡呀" + img, log_cmd="被动早晚安"
+#     )
+#     logger.info("每日晚安发送...")