from pydantic import BaseModel
from datetime import datetime
from enum import Enum
from pydantic import  ConfigDict
class TaskStatus(str, Enum):
    PENDING = "pending"
    RUNNING = "running"
    COMPLETED = "completed"
    FAILED = "failed"

class ImportTaskResponse(BaseModel):
    id: int
    task_type: str
    status: TaskStatus
    details: dict | None
    created_at: datetime
    updated_at: datetime

class ResponseModel(BaseModel):
    success: bool
    message: str
    data: dict | list | None = None
    model_config = ConfigDict(from_attributes=True)