300-add-esp-temperature (#318)

* Update sensor.py Add CelsiusSensorEntity class * Update powerstream.py Added 20_1.espTempsensor as "ESP Temperature"
tolwi · Sep 5, 2024 · 1e03006 · 1e03006
1 parent c1a4915
commit 1e03006
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/custom_components/ecoflow_cloud/devices/public/powerstream.py b/custom_components/ecoflow_cloud/devices/public/powerstream.py
@@ -5,14 +5,16 @@
 from ...number import MinBatteryLevelEntity, MaxBatteryLevelEntity, BrightnessLevelEntity, DeciChargingPowerEntity
 from ...select import PowerDictSelectEntity
 from ...sensor import StatusSensorEntity, InWattsSolarSensorEntity, DecivoltSensorEntity, CentivoltSensorEntity, \
-    DeciampSensorEntity, DecicelsiusSensorEntity, MiscSensorEntity, LevelSensorEntity, DeciwattsSensorEntity, \
+    DeciampSensorEntity, CelsiusSensorEntity, DecicelsiusSensorEntity, MiscSensorEntity, LevelSensorEntity, DeciwattsSensorEntity, \
     AmpSensorEntity, RemainSensorEntity, DecihertzSensorEntity
 
 
 class PowerStream(BaseDevice):
 
     def sensors(self, client: EcoflowApiClient) -> list[BaseSensorEntity]:
         return [
+            CelsiusSensorEntity(client, self, "20_1.espTempsensor", "ESP Temperature"),
+
             InWattsSolarSensorEntity(client, self, "20_1.pv1InputWatts", "Solar 1 Watts"),
             DecivoltSensorEntity(client, self, "20_1.pv1InputVolt", "Solar 1 Input Potential"),
             CentivoltSensorEntity(client, self, "20_1.pv1OpVolt", "Solar 1 Op Potential"),

diff --git a/custom_components/ecoflow_cloud/sensor.py b/custom_components/ecoflow_cloud/sensor.py
@@ -107,6 +107,9 @@ class TempSensorEntity(BaseSensorEntity):
     _attr_state_class = SensorStateClass.MEASUREMENT
     _attr_native_value = -1
 
+class CelsiusSensorEntity(TempSensorEntity):
+    def _update_value(self, val: Any) -> bool:
+        return super()._update_value(int(val))
 
 class DecicelsiusSensorEntity(TempSensorEntity):
     def _update_value(self, val: Any) -> bool: