Class RoverGroupActivity

RoverSign/utils/database/rover_group_activity.py:19–110

群活跃度记录表: 群最后有人使用本插件的时间

class RoverGroupActivity(BaseBotIDModel, table=True):
    """群活跃度记录表: 群最后有人使用本插件的时间"""

    __tablename__ = "RoverGroupActivity"
    __table_args__: Dict[str, Any] = {"extend_existing": True}

    group_id: str = Field(default="", title="群组ID")
    bot_self_id: str = Field(default="", title="机器人自身ID")
    last_active_time: Optional[int] = Field(default=None, title="最后活跃时间")

    @classmethod
    async def update_group_activity(
        cls: Type[T_RoverGroupActivity],
        group_id: str,
        bot_id: str,
        bot_self_id: str,
    ) -> bool:
        try:
            return await cls._do_update_group_activity(group_id, bot_id, bot_self_id)
        except Exception as e:
            if "malformed" in str(e) or "corrupt" in str(e):
                logger.warning(f"[库洛签到·群活跃度] 数据库损坏，跳过更新: {e}")
                return False
            raise

    @classmethod
    @with_lock
    @with_session
    async def _do_update_group_activity(
        cls: Type[T_RoverGroupActivity],
        session: AsyncSession,
        group_id: str,
        bot_id: str,
        bot_self_id: str,
    ) -> bool:
        import time

        current_time = int(time.time())
        sql = select(cls).where(
            and_(
                cls.group_id == group_id,
                cls.bot_id == bot_id,
                cls.bot_self_id == bot_self_id,
            )
        )
        result = await session.execute(sql)
        existing = result.scalars().first()
        if existing:
            existing.last_active_time = current_time
            session.add(existing)
        else:
            session.add(
                cls(
                    group_id=group_id,
                    bot_id=bot_id,
                    bot_self_id=bot_self_id,
                    last_active_time=current_time,
                )