Update sensor.py

Correct rounding so money remains 2 decimal places
hif2k1 · Aug 17, 2022 · 01f62cf · 01f62cf
1 parent 1e6b411
commit 01f62cf
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/custom_components/battery_sim/sensor.py b/custom_components/battery_sim/sensor.py
@@ -146,8 +146,11 @@ def device_info(self):
 
     @property
     def native_value(self):
-        """Return the state of the sensor."""
-        return round(float(self._handle._sensors[self._sensor_type]),3)
+        """Return the state of the sensor."""        
+        if (self._sensor_type == ATTR_MONEY_SAVED):
+            return round(float(self._handle._sensors[self._sensor_type]),2)
+        else:
+            return round(float(self._handle._sensors[self._sensor_type]),3)
 
     @property
     def device_class(self):
@@ -203,7 +206,10 @@ def icon(self):
     @property
     def state(self):
         """Return the state of the sensor."""
-        return round(float(self._handle._sensors[self._sensor_type]),3)
+        if (self._sensor_type == ATTR_MONEY_SAVED):
+            return round(float(self._handle._sensors[self._sensor_type]),2)
+        else:
+            return round(float(self._handle._sensors[self._sensor_type]),3)
 
     def update(self):
         """Not used"""