fix: Handle temperature and retrieved at times not being provided for heat pumps (15 minutes dev time)

Author: BottlecapDave

custom_components/octopus_energy/api_client/heat_pump.py

@@ -11,9 +11,9 @@ class Connectivity(BaseModel):
 
 
 class Telemetry(BaseModel):
-    temperatureInCelsius: float
+    temperatureInCelsius: Optional[float]
     humidityPercentage: Optional[float]
-    retrievedAt: str
+    retrievedAt: Optional[str]
 
 
 class Sensor(BaseModel):

custom_components/octopus_energy/heat_pump/__init__.py

@@ -44,9 +44,9 @@ def mock_heat_pump_status_and_configuration():
             "retrievedAt": (now - timedelta(seconds=random.randrange(1, 120))).strftime("%Y-%m-%dT%H:%M:%S.%f%z")
           },
           "telemetry": {
-            "temperatureInCelsius": -273 + (random.randrange(1, 20) * 0.1),
+            "temperatureInCelsius": None,
             "humidityPercentage": None,
-            "retrievedAt": (now - timedelta(seconds=random.randrange(1, 120))).strftime("%Y-%m-%dT%H:%M:%S.%f%z")
+            "retrievedAt": None
           }
         },
         {

custom_components/octopus_energy/heat_pump/sensor_humidity.py

@@ -92,7 +92,7 @@ def _handle_coordinator_update(self) -> None:
       for sensor in sensors:
         if sensor.code == self._sensor.code and sensor.telemetry is not None:
           self._state = sensor.telemetry.humidityPercentage
-          self._attributes["retrieved_at"] = datetime.fromisoformat(sensor.telemetry.retrievedAt)
+          self._attributes["retrieved_at"] = datetime.fromisoformat(sensor.telemetry.retrievedAt) if sensor.telemetry.retrievedAt is not None else None
 
       self._last_updated = current
 

custom_components/octopus_energy/heat_pump/sensor_temperature.py

@@ -93,7 +93,7 @@ def _handle_coordinator_update(self) -> None:
       for sensor in sensors:
         if sensor.code == self._sensor.code and sensor.telemetry is not None:
           self._state = sensor.telemetry.temperatureInCelsius
-          self._attributes["retrieved_at"] = datetime.fromisoformat(sensor.telemetry.retrievedAt)
+          self._attributes["retrieved_at"] = datetime.fromisoformat(sensor.telemetry.retrievedAt) if sensor.telemetry.retrievedAt is not None else None
 
       self._last_updated = current
 

tests/unit/api_client/test_heat_pump.py

@@ -36,9 +36,9 @@ def test_when_valid_dictionary_returned_then_it_can_be_parsed_into_heat_pump_obj
             "retrievedAt": "2025-05-09T17:28:51.555000+00:00"
           },
           "telemetry": {
-            "temperatureInCelsius": -90.3,
+            "temperatureInCelsius": None,
             "humidityPercentage": None,
-            "retrievedAt": "2025-05-09T17:28:44.152000+00:00"
+            "retrievedAt": None
           }
         },
         {