🐛 移除群组数据和聊天数据删除代码 (#2128)

This commit is contained in:
HibiKier
2026-04-27 16:09:33 +08:00
committed by GitHub
parent 98bc636a39
commit 9cb40f0432
2 changed files with 34 additions and 129 deletions
+1 -90
View File
@@ -3,10 +3,7 @@ from pathlib import Path
import nonebot
from nonebot.adapters import Bot
from nonebot.adapters.onebot.v11.exception import NetworkError
from nonebot_plugin_apscheduler import scheduler
from zhenxun.configs.config import Config
from zhenxun.models.chat_history import ChatHistory
from zhenxun.models.group_console import GroupConsole
from zhenxun.services.log import logger
from zhenxun.utils.manager.priority_manager import PriorityLifecycle
@@ -20,16 +17,6 @@ nonebot.load_plugins(str(Path(__file__).parent.resolve()))
driver = nonebot.get_driver()
Config.add_plugin_config(
"auto_clean",
"CLEAN_CHAT_HISTORY",
True,
help="是否自动清理已退出群聊的聊天记录",
default_value=True,
type=bool,
)
@PriorityLifecycle.on_startup(priority=5)
async def _():
register_cache_types()
@@ -56,7 +43,6 @@ async def _(bot: Bot):
"群认证同步",
)
return
current_group_ids = {g.group_id for g in current_group_list}
db_group_list: list[str] = await GroupConsole.all().values_list(
"group_id", flat=True
@@ -83,82 +69,7 @@ async def _(bot: Bot):
for group in fresh:
await GroupMemoryCache.upsert_from_model(group)
all_bots = nonebot.get_bots()
all_visible: set[str] = set(current_group_ids)
for other_bot in all_bots.values():
if other_bot is bot:
continue
if PlatformUtils.get_platform(other_bot) != "qq":
continue
try:
other_groups, _ = await PlatformUtils.get_group_list(other_bot)
all_visible.update(g.group_id for g in other_groups)
except NetworkError as e:
reason = (
f"Bot: {other_bot.self_id} 群列表同步被连接关闭打断,"
f"回退到数据库集合: {e}"
)
logger.debug(
reason,
"群认证同步",
)
all_visible.update(db_group_ids)
break
except Exception:
all_visible.update(db_group_ids)
break
if delete_ids := list(db_group_ids - all_visible):
deleted_count = await GroupConsole.filter(group_id__in=delete_ids).delete()
else:
deleted_count = 0
logger.info(
f"更新Bot: {bot.self_id} 的群认证完成,共创建 {len(create_list)} 条数据,"
f"删除 {deleted_count} 条已退出群组的数据...",
f"更新Bot: {bot.self_id} 的群认证完成,共创建 {len(create_list)} 条数据,",
"群认证同步",
)
if Config.get_config("auto_clean", "CLEAN_CHAT_HISTORY"):
scheduler.add_job(
clean_chat_history,
"cron",
hour=1,
minute=0,
args=(current_group_list,),
id="clean_chat_history",
replace_existing=True,
)
async def clean_chat_history(
group_list: list[GroupConsole],
max_delete: int = 2000,
):
"""清理已退出群组的聊天记录
为避免一次调用删除过多数据,单次调用最多删除 max_delete 条。
"""
# 将传入的对象统一转成 group_id 字符串列表
group_ids: list[str] = [g.group_id for g in group_list]
if not group_ids:
logger.warning("传入群组列表为空,跳过清理", "定时清理群组聊天记录")
return
# 只取最多 max_delete 条记录的 id,然后删除这些记录,避免一次删太多
ids = (
await ChatHistory.filter(group_id__not_in=group_ids)
.order_by("create_time")
.limit(max_delete)
.values_list("id", flat=True)
)
ids = list(ids)
if not ids:
logger.info(
f"群组数 {len(group_ids)},无聊天记录可删除", "定时清理群组聊天记录"
)
return
await ChatHistory.filter(id__in=ids).delete()
logger.success(f"已清理 {len(ids)} 条已退出群组的聊天记录", "定时清理群组聊天记录")