mirror of
https://github.com/home-assistant/core.git
synced 2025-11-09 19:09:32 +00:00
d65e704823
Co-authored-by: J. Nick Koston <nick@home-assistant.io> Co-authored-by: J. Nick Koston <nick@koston.org>
25 lines
702 B
Python
25 lines
702 B
Python
"""Models for the usage prediction integration."""
|
|
|
|
from dataclasses import dataclass, field
|
|
from datetime import datetime
|
|
|
|
from homeassistant.util import dt as dt_util
|
|
|
|
|
|
@dataclass
|
|
class EntityUsagePredictions:
|
|
"""Prediction which entities are likely to be used in each time category."""
|
|
|
|
morning: list[str] = field(default_factory=list)
|
|
afternoon: list[str] = field(default_factory=list)
|
|
evening: list[str] = field(default_factory=list)
|
|
night: list[str] = field(default_factory=list)
|
|
|
|
|
|
@dataclass
|
|
class EntityUsageDataCache:
|
|
"""Data model for entity usage prediction."""
|
|
|
|
predictions: EntityUsagePredictions
|
|
timestamp: datetime = field(default_factory=dt_util.utcnow)
|