from __future__ import annotations

from datetime import datetime

from beanie import Document
from pydantic import Field


class TelemetryData(Document):
    device_id: str
    metrics: dict = Field(default_factory=dict)
    timestamp: datetime = Field(default_factory=datetime.utcnow)

    class Settings:
        name = "telemetry_data"
        indexes = [
            "device_id",
            [("timestamp", -1)],
            [("device_id", 1), ("timestamp", -1)],
        ]