CheckInApp/apps/backend/schemas/task.py

from datetime import datetime
from typing import Optional
from pydantic import BaseModel, ConfigDict, Field, field_validator


class TaskBase(BaseModel):
    """打卡任务基础 Schema"""

    payload_config: str = Field(
        ..., description="完整的 payload 配置 JSON（包含 ThreadId 和所有字段）"
    )
    name: Optional[str] = Field("", max_length=100, description="任务名称（用户自定义）")
    is_active: Optional[bool] = Field(True, description="是否启用自动打卡")

    @field_validator("payload_config")
    @classmethod
    def validate_payload_config(cls, v: str) -> str:
        """
        验证 payload_config 是否为有效的 JSON，并且包含必需的 ThreadId 字段
        """
        from backend.utils.json_helpers import safe_parse_json, extract_thread_id

        if not v or not v.strip():
            raise ValueError("payload_config 不能为空")

        payload = safe_parse_json(v)
        if payload is None:
            raise ValueError("payload_config 必须是有效的 JSON 格式")

        # 检查是否为字典类型
        if not isinstance(payload, dict):
            raise ValueError("payload_config 必须是 JSON 对象（字典）")

        # 检查必需字段 ThreadId
        thread_id = extract_thread_id(v)
        if not thread_id or not str(thread_id).strip():
            raise ValueError("payload_config 必须包含有效的 ThreadId 字段")

        return v


class TaskCreate(TaskBase):
    """创建打卡任务 Schema"""

    cron_expression: Optional[str] = Field(
        None,
        max_length=100,
        description="Crontab 表达式（例如 '0 20 * * *' 表示每天 20:00）。NULL 表示禁用定时打卡",
    )

    @field_validator("cron_expression")
    @classmethod
    def validate_cron_expression(cls, v: Optional[str]) -> Optional[str]:
        """验证 Crontab 表达式格式"""
        if v is None:
            return v  # NULL 允许（表示禁用定时打卡）

        if not v.strip():
            raise ValueError("cron_expression 不能为空字符串，应该使用 NULL")

        try:
            from croniter import croniter

            if not croniter.is_valid(v):
                raise ValueError(f"无效的 Crontab 表达式: '{v}'")
        except Exception as e:
            raise ValueError(f"Crontab 表达式验证失败: {str(e)}")

        return v


class TaskUpdate(BaseModel):
    """更新打卡任务 Schema"""

    payload_config: Optional[str] = None
    name: Optional[str] = None
    is_active: Optional[bool] = None
    cron_expression: Optional[str] = Field(
        None, max_length=100, description="Crontab 表达式。NULL 表示禁用定时打卡"
    )

    @field_validator("payload_config")
    @classmethod
    def validate_payload_config(cls, v: Optional[str]) -> Optional[str]:
        """
        验证 payload_config 是否为有效的 JSON（如果提供的话）
        """
        from backend.utils.json_helpers import safe_parse_json, extract_thread_id

        if v is None:
            return v

        if not v.strip():
            raise ValueError("payload_config 不能为空字符串")

        payload = safe_parse_json(v)
        if payload is None:
            raise ValueError("payload_config 必须是有效的 JSON 格式")

        # 检查是否为字典类型
        if not isinstance(payload, dict):
            raise ValueError("payload_config 必须是 JSON 对象（字典）")

        # 检查必需字段 ThreadId
        thread_id = extract_thread_id(v)
        if not thread_id or not str(thread_id).strip():
            raise ValueError("payload_config 必须包含有效的 ThreadId 字段")

        return v

    @field_validator("cron_expression")
    @classmethod
    def validate_cron_expression(cls, v: Optional[str]) -> Optional[str]:
        """验证 Crontab 表达式（与 TaskCreate 相同）"""
        if v is None:
            return v

        if not v.strip():
            raise ValueError("cron_expression 不能为空字符串，应该使用 NULL")

        try:
            from croniter import croniter

            if not croniter.is_valid(v):
                raise ValueError(f"无效的 Crontab 表达式: '{v}'")
        except Exception as e:
            raise ValueError(f"Crontab 表达式验证失败: {str(e)}")

        return v


class TaskResponse(TaskBase):
    """打卡任务响应 Schema"""

    model_config = ConfigDict(from_attributes=True)

    id: int
    user_id: int
    created_at: datetime
    updated_at: Optional[datetime] = None
    cron_expression: Optional[str] = Field(
        None, description="当前 Crontab 表达式（NULL = 禁用定时打卡）"
    )
    is_scheduled_enabled: Optional[bool] = Field(None, description="是否启用了定时打卡")

    # 新增字段：最后一次打卡信息
    last_check_in_time: Optional[datetime] = Field(None, description="最后一次打卡时间")
    last_check_in_status: Optional[str] = Field(None, description="最后一次打卡状态")
    thread_id: Optional[str] = Field(None, description="接龙 ID（从 payload_config 中提取）")