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


class TaskBase(BaseModel):
    command: str
    args: Any
    status: Optional[int] = 0


class TaskCreate(TaskBase):
    pass


class TaskUpdate(BaseModel):
    result: Optional[Any] = None
    status: Optional[int] = None


class TaskOut(TaskBase):
    id: int
    result: Optional[Any] = None
    create_at: datetime
    update_at: datetime

    model_config = {
        "from_attributes": True
    }