Attributo per arrotondamenti

virtualdj · Dec 10, 2022 · d239dae · d239dae
1 parent beeb66e
commit d239dae
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/custom_components/pun_sensor/sensor.py b/custom_components/pun_sensor/sensor.py
@@ -8,6 +8,7 @@
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.typing import DiscoveryInfoType
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
+from typing import Any, Dict
 
 from . import PUNDataUpdateCoordinator
 from .const import (
@@ -17,6 +18,7 @@
     PUN_FASCIA_F2,
     PUN_FASCIA_F3,
 )
+ATTR_ROUNDED_DECIMALS = "rounded_decimals"
 
 import logging
 _LOGGER = logging.getLogger(__name__)
@@ -111,6 +113,14 @@ def name(self) -> str:
         else:
             return None
 
+    @property
+    def extra_state_attributes(self) -> Dict[str, Any]:
+        """Restituisce gli attributi di stato"""
+        state_attr = {
+            ATTR_ROUNDED_DECIMALS: round(self.native_value, 3)
+        }
+        return state_attr
+
 class FasciaPUNSensorEntity(CoordinatorEntity, SensorEntity):
     """Sensore che rappresenta la fascia PUN corrente"""
 
@@ -205,4 +215,12 @@ def entity_id(self) -> str:
     @property
     def name(self) -> str:
         """Restituisce il nome del sensore"""
-        return "Prezzo fascia corrente"
+        return "Prezzo fascia corrente"
+
+    @property
+    def extra_state_attributes(self) -> Dict[str, Any]:
+        """Restituisce gli attributi di stato"""
+        state_attr = {
+            ATTR_ROUNDED_DECIMALS: round(self.native_value, 3)
+        }
+        return state_attr