schemas/template.py

from pydantic import BaseModel
from datetime import datetime
from typing import Optional, List

class TemplateBase(BaseModel):
    key: str
    name: str
    description: Optional[str] = None
    color: str
    features: Optional[List[str]] = None
    applicable_scene: Optional[str] = None
    regions: Optional[List[str]] = None

class TemplateCreate(TemplateBase):
    pass

class TemplateUpdate(BaseModel):
    name: Optional[str] = None
    description: Optional[str] = None
    color: Optional[str] = None
    features: Optional[List[str]] = None
    applicable_scene: Optional[str] = None
    regions: Optional[List[str]] = None

class TemplateResponse(TemplateBase):
    id: int
    select_count: int = 0
    use_count: int = 0
    status: int = 0
    created_at: datetime
    updated_at: Optional[datetime] = None
    
    class Config:
        from_attributes = True


class TemplateStatusUpdate(BaseModel):
    status: int