Handle partial price data better

If today or tomorrow has only partial price data, don't return any averages
or highest/lowest price, because it's not possible to return something that
makes sense if we don't have prices for the whole day.

Author: slovdahl

spothinta_api/models.py

@@ -59,6 +59,7 @@ class Electricity:
     """Object representing electricity data."""
 
     prices: dict[datetime, float]
+    resolution: timedelta
     time_zone: ZoneInfo
 
     @property
@@ -257,12 +258,19 @@ def prices_today(self) -> dict[datetime, float]:
 
         """
         today = self.now_in_timezone().astimezone().date()
-        return {
+        prices = {
             timestamp: price
             for timestamp, price in self.prices.items()
             if timestamp.date() == today
         }
 
+        if (self.resolution == timedelta(minutes=15) and len(prices) == 96) or (
+            self.resolution == timedelta(minutes=60) and len(prices) == 24
+        ):
+            return prices
+
+        return {}
+
     def prices_tomorrow(self) -> dict[datetime, float]:
         """Return the prices for tomorrow.
 
@@ -272,12 +280,19 @@ def prices_tomorrow(self) -> dict[datetime, float]:
 
         """
         tomorrow = (self.now_in_timezone() + timedelta(days=1)).astimezone().date()
-        return {
+        prices = {
             timestamp: price
             for timestamp, price in self.prices.items()
             if timestamp.date() == tomorrow
         }
 
+        if (self.resolution == timedelta(minutes=15) and len(prices) == 96) or (
+            self.resolution == timedelta(minutes=60) and len(prices) == 24
+        ):
+            return prices
+
+        return {}
+
     def now_in_timezone(self) -> datetime:
         """Return the current timestamp in the current timezone.
 
@@ -329,13 +344,15 @@ def price_at_time(self, moment: datetime) -> float | None:
     def from_dict(
         cls: type[Electricity],
         data: list[dict[str, Any]],
+        resolution: timedelta,
         time_zone: ZoneInfo,
     ) -> Electricity:
         """Create an Electricity object from a dictionary.
 
         Args:
         ----
             data: A dictionary with the data from the API.
+            resolution: The price resolution.
             time_zone: The timezone to use for determining "today" and "tomorrow".
 
         Returns:
@@ -350,5 +367,6 @@ def from_dict(
             ]
         return cls(
             prices=prices,
+            resolution=resolution,
             time_zone=time_zone,
         )

spothinta_api/spothinta.py

@@ -161,7 +161,7 @@ async def energy_prices(
             raise SpotHintaNoDataError(msg)
 
         time_zone = await async_get_time_zone(REGION_TO_TIMEZONE[region])
-        return Electricity.from_dict(data, time_zone=time_zone)
+        return Electricity.from_dict(data, resolution=resolution, time_zone=time_zone)
 
     async def close(self) -> None:
         """Close open client session."""