from __future__ import annotations

from datetime import datetime

from pydantic import BaseModel


class HaState(BaseModel):
    entity_id: str
    state: str
    attributes: dict[str, object] | None = None
    last_changed: str | None = None
    last_updated: str | None = None


class HaEntitySummary(BaseModel):
    entity_id: str
    domain: str
    state: str | None = None
    last_changed: datetime | None = None
    state_class: str | None = None
    device_class: str | None = None
    unit_of_measurement: str | None = None
    friendly_name: str | None = None
    area_id: str | None = None
    area_name: str | None = None
    device_id: str | None = None
    device_name: str | None = None