LongArticleTaskServer/app/domains/recommend/i2i_recommend/_utils.py

from typing import Dict

from app.infra.external import feishu_robot

from ._const import I2IRecommendDataSyncConst


class I2IRecommendDataSyncUtil(I2IRecommendDataSyncConst):

    @staticmethod
    def filter_accounts_to_sync(account_version_list: list[dict]) -> list[dict]:
        """
        筛选需要切换的账号：最大version的status为OFFLINE(0)
        """
        return [
            row for row in account_version_list
            if row["status"] == I2IRecommendDataSyncConst.VersionStatus.INIT
        ]

    # 飞书通知
    async def bot(self, title: str, detail: Dict, mention: bool=False):
        return await feishu_robot.bot(
            title=title,
            detail=detail,
            env=self.RANK_BOT,
            mention=mention,
        )