from datetime import datetime

from pydantic import BaseModel


class IntervalDataPoint(BaseModel):
    ts: datetime
    generation_kwh: float
    load_kwh: float
    import_kwh: float
    export_kwh: float
    resolution_minutes: int

    def as_row(self, source: str) -> dict:
        return {
            "ts": self.ts,
            "source": source,
            "generation_kwh": self.generation_kwh,
            "load_kwh": self.load_kwh,
            "import_kwh": self.import_kwh,
            "export_kwh": self.export_kwh,
            "resolution_minutes": self.resolution_minutes,
        }