zhenxun_bot/zhenxun/builtin_plugins/scheduler/chat_check.py

from datetime import datetime, timedelta

import pytz
import nonebot
from nonebot_plugin_apscheduler import scheduler

from zhenxun.services.log import logger
from zhenxun.configs.config import Config
from zhenxun.models.task_info import TaskInfo
from zhenxun.utils.platform import PlatformUtils
from zhenxun.models.chat_history import ChatHistory
from zhenxun.models.group_console import GroupConsole

Config.add_plugin_config(
    "chat_check",
    "STATUS",
    True,
    help="是否开启群组两日内未发送任何消息，关闭该群全部被动",
    default_value=True,
    type=bool,
)


@scheduler.scheduled_job(
    "cron",
    hour=4,
    minute=40,
)
async def _():
    if not Config.get_config("chat_history", "FLAG"):
        logger.debug("未开启历史发言记录，过滤群组发言检测...")
        return
    if not Config.get_config("chat_check", "STATUS"):
        logger.debug("未开启群组聊天时间检查，过滤群组发言检测...")
        return
    """检测群组发言时间并禁用全部被动"""
    update_list = []
    if modules := await TaskInfo.annotate().values_list("module", flat=True):
        for bot in nonebot.get_bots().values():
            group_list, _ = await PlatformUtils.get_group_list(bot)
            group_list = [g for g in group_list if g.channel_id is None]
            for group in group_list:
                try:
                    last_message = (
                        await ChatHistory.filter(group_id=group.group_id)
                        .annotate()
                        .order_by("-create_time")
                        .first()
                    )
                    if last_message:
                        now = datetime.now(pytz.timezone("Asia/Shanghai"))
                        if now - timedelta(days=2) > last_message.create_time:
                            _group, _ = await GroupConsole.get_or_create(
                                group_id=group.group_id, channel_id__isnull=True
                            )
                            _group.block_task = ",".join(modules) + ","  # type: ignore
                            update_list.append(_group)
                            logger.info(
                                "群组两日内未发送任何消息，关闭该群全部被动",
                                "Chat检测",
                                target=_group.group_id,
                            )
                except Exception:
                    logger.error(
                        "检测群组发言时间失败...", "Chat检测", target=group.group_id
                    )
    if update_list:
        await GroupConsole.bulk_update(update_list, ["block_task"], 10)