from pydantic import Field
from pydantic_settings import BaseSettings, SettingsConfigDict
from typing import Dict


class TaskChineseNameConfig(BaseSettings):
    """冷启动配置"""

    # 分类映射
    category_map: Dict[str, str] = Field(
        default_factory=lambda: {
            "title_rewrite": "标题重写",
            "crawler_gzh_articles": "抓取公众号文章",
            "crawler_account_manager": "抓取账号管理",
            "article_pool_category_generation": "文章池品类生成",
            "candidate_account_quality_analysis": "候选账号质量分析",
            "article_pool_cold_start": "文章路冷启动",
            "crawler_toutiao": "头条抓取",
            "task_processing_monitor": "协程监测",
            "update_root_source_id": "更新今日root_source_id",
            "daily_publish_articles_recycle": "回收今日发文",
            "inner_article_monitor": "账号发文违规监测",
            "outside_article_monitor": "外部服务号发文监测",
            "get_off_videos": "自动下架视频",
            "check_publish_video_audit_status": "校验发布视频状态",
            "check_kimi_balance": "检验kimi余额",
            "account_category_analysis": "账号品类分析",
            "mini_program_detail_process": "更新小程序信息",
            "crawler_detail_analysis": "抓取详情分析",
            "limited_account_analysis": "限流账号分析处理",
            "auto_follow_account": "自动关注账号",
            "update_account_open_rate_avg": "更新账号平均打开率",
            "update_limited_account_info": "更新限流账号信息",
            "update_account_read_avg": "更新账号平均阅读率",
            "get_follow_result": "获取自动关注回复",
            "extract_reply_result": "解析自动回复结果",
        }
    )

    model_config = SettingsConfigDict(
        env_prefix="TASK_CHINESE_NAME_",
        env_file=".env",
        case_sensitive=False,
        extra="ignore",
    )