from pydantic import BaseModel
from typing import Set

class Config(BaseModel):
    disabled_groups: Set[int] = set()  # 默认禁用的群聊列表
    enabled_ban_groups: Set[int] = set()  # 默认允许禁言的群聊列表
    rolling_ban: bool = False  # 是否禁言随机时间
    ban_duration: int = 600  # 默认禁言时间（秒）

# 重建模型
Config.model_rebuild()