"""爬虫脚本相关的Pydantic模型"""

from datetime import datetime
from typing import Optional, Dict, Any
from pydantic import BaseModel


class ScriptBase(BaseModel):
    """脚本基础字段"""
    script_name: str
    script_content: str
    config_content: Optional[Dict[str, Any]] = None


class ScriptCreate(ScriptBase):
    """创建脚本请求"""
    university_id: int


class ScriptResponse(ScriptBase):
    """脚本响应"""
    id: int
    university_id: int
    version: int
    status: str
    error_message: Optional[str] = None
    created_at: datetime
    updated_at: datetime

    class Config:
        from_attributes = True


class GenerateScriptRequest(BaseModel):
    """生成脚本请求"""
    university_url: str
    university_name: Optional[str] = None


class GenerateScriptResponse(BaseModel):
    """生成脚本响应"""
    success: bool
    university_id: int
    script_id: Optional[int] = None
    message: str
    status: str  # analyzing, completed, failed