app/widget_collection/model.py

"""MongoDB document model for analytics_widget_collection."""
from typing import Any, Dict, Optional
from pydantic import BaseModel


class DataConfig(BaseModel):
    source: str
    params: Dict[str, Any] = {}


class WidgetCollectionDocument(BaseModel):
    widget_id: str
    type: str = "kpi"
    title: str
    category: str
    unit: Optional[str] = None
    description: Optional[str] = None
    drill_down_url: Optional[str] = None
    data_config: DataConfig

    class Config:
        populate_by_name = True