Add all-time kWh sensor and enable voltage sensor by default due to c…

…hange in power balancer logic
TeepCo · Jul 15, 2024 · e8e6fd2 · e8e6fd2
1 parent d6b3f1d
commit e8e6fd2
Showing 1 changed file with 43 additions and 1 deletion.
diff --git a/custom_components/chuck_control/sensor.py b/custom_components/chuck_control/sensor.py
@@ -444,7 +444,7 @@ def unique_id(self) -> str | None:
 
     @property
     def entity_registry_enabled_default(self) -> bool:
-        return False
+        return True
 
 
 class ConnectorPower(SensorEntity):
@@ -646,6 +646,48 @@ def state_attributes(self) -> dict[str, Any]:
         return {"connector_count ": self.chargebox.get_connectors_count()}
 
 
+class ChargeBoxTotalEnergy(SensorEntity):
+    def __init__(self, chargebox) -> None:
+        self.chargebox = chargebox
+        self.friendly_name_appendix = "Total energy"
+        self.friendly_name = get_friendly_name(self)
+
+    @property
+    def name(self) -> str:
+        return self.friendly_name
+
+    @property
+    def device_info(self) -> DeviceInfo:
+        return self.chargebox.get_device_info()
+
+    @property
+    def state(self) -> Any:
+        return round(self.chargebox.get_energy_total() / 1000, 4)
+
+    @property
+    def unique_id(self) -> str:
+        return f"{self.chargebox.info['serialNumber']}_energy_total"
+
+    @property
+    def device_class(self) -> SensorDeviceClass:
+        return SensorDeviceClass.ENERGY
+
+    @property
+    def state_class(self) -> SensorStateClass:
+        return SensorStateClass.MEASUREMENT
+
+    @property
+    def unit_of_measurement(self) -> str:
+        return UnitOfEnergy.KILO_WATT_HOUR
+
+    def update(self) -> None:
+        self.chargebox.update()
+
+    @property
+    def state_attributes(self) -> dict[str, Any]:
+        return {"connector_count ": self.chargebox.get_connectors_count()}
+
+
 class ConnectorInternalTemp(SensorEntity):
     def __init__(self, chargebox, connector_id) -> None:
         self.chargebox = chargebox