PydanticTaskModels.py

from typing import Dict, List, Any, Optional, Union, Tuple
from typing_extensions import Annotated
from pydantic import BaseModel, Field
from datetime import datetime, timedelta
from enum import Enum
import json


def custom_json_serializer(obj):
    if isinstance(obj, datetime):
        return obj.isoformat()
    elif isinstance(obj, timedelta):
        return str(obj)
    elif isinstance(obj, Enum):
        return obj.value
    elif isinstance(obj, tuple):
        return list(obj)
    raise TypeError(f"Object of type {obj.__class__.__name__} is not JSON serializable")

class CustomJSONEncoder(json.JSONEncoder):
    def default(self, obj):
        try:
            return custom_json_serializer(obj)
        except TypeError:
            return super().default(obj)

class BaseModelWithCustomJSON(BaseModel):
    class Config:
        json_encoders = {
            datetime: lambda v: v.isoformat(),
            timedelta: str,
            Enum: lambda v: v.value,
            tuple: list,
        }

    def json(self, **kwargs):
        return json.dumps(self.dict(), cls=CustomJSONEncoder, **kwargs)

# Pydantic models
class TodoCreate(BaseModelWithCustomJSON):
    content: str = Field(..., min_length=1, max_length=1000)

class TodoOut(BaseModelWithCustomJSON):
    id: str
    content: str
    createdAt: datetime
    updatedAt: datetime

# Simplified Pydantic models for OKR
class OKRCreate(BaseModelWithCustomJSON):
    title: str = Field(..., min_length=1, max_length=200)
    description: str = Field(..., min_length=1, max_length=1000)

class OKROut(BaseModelWithCustomJSON):
    id: str
    title: str
    description: str
    createdAt: datetime
    updatedAt: datetime

class OKROutList(BaseModelWithCustomJSON):
    okrs: List[OKROut]

class TaskCreate(BaseModelWithCustomJSON):
    name: str
    description: Optional[str] = None
    estimated_time_mins: Optional[int] = None
    priority: Optional[int] = None
    tags: Optional[List[str]] = None
    scheduled_date_utc: Optional[int] = None  # Add this line

class TaskOut(BaseModelWithCustomJSON):
    id: str
    name: str
    description: Optional[str] = None
    estimated_time_mins: Optional[int] = None
    priority: Optional[int] = None
    tags: Optional[List[str]] = None
    scheduled_date_utc: Optional[int] = None  # Add this line
    createdAt: datetime
    updatedAt: datetime
    
class TaskList(BaseModelWithCustomJSON):
    tasks: List[TaskOut]

class NullModel(BaseModelWithCustomJSON):
    value: None = None

class DirectoryInput(BaseModelWithCustomJSON):
    directory: str

class FileList(BaseModelWithCustomJSON):
    files: List[str]
    
class CurrentDateTime(BaseModelWithCustomJSON):
    current_datetime: datetime

class InputDatetimePlaintext(BaseModelWithCustomJSON):
    input_dt: str

class DatetimeMillis(BaseModelWithCustomJSON):
    datetime_millis: str

class DatetimeSeconds(BaseModelWithCustomJSON):
    datetime_seconds: str

class DeleteTaskInput(BaseModelWithCustomJSON):
    id: str    

class TaskId(BaseModelWithCustomJSON):
    id: str    

class UpdateTaskInput(BaseModelWithCustomJSON):
    id: str
    name: Optional[str] = None
    description: Optional[str] = None
    estimated_time_mins: Optional[int] = None
    priority: Optional[int] = None
    tags: Optional[List[str]] = None
    scheduled_date_utc: Optional[int] = None    


class UTCSecondsList(BaseModelWithCustomJSON):
    utc_seconds: List[int]

class HumanReadableDateList(BaseModelWithCustomJSON):
    dates: List[str]

class HNStory(BaseModelWithCustomJSON):
    title: str
    url: str

class HNSearchResult(BaseModelWithCustomJSON):
    hits: List[HNStory]

class HNBlob(BaseModelWithCustomJSON):
    search_result: HNSearchResult

class DadJoke(BaseModelWithCustomJSON):
    joke_id: str
    joke_contents: str