Make battery low sensor restore

Author: andrew-codechimp

custom_components/battery_notes/binary_sensor.py

@@ -24,6 +24,7 @@
     config_validation as cv,
     device_registry as dr,
     entity_registry as er,
+    template,
 )
 from homeassistant.helpers.event import (
     EventStateChangedData,
@@ -32,7 +33,6 @@
     async_track_template_result,
 )
 
-from homeassistant.helpers import template
 from homeassistant.helpers.template import (
     Template,
     TemplateStateFromEntityId,
@@ -51,6 +51,7 @@
     async_track_entity_registry_updated_event,
 )
 from homeassistant.helpers.reload import async_setup_reload_service
+from homeassistant.helpers.restore_state import RestoreEntity
 
 from homeassistant.const import (
     CONF_NAME,
@@ -313,7 +314,7 @@ def handle_result(
 
 
 class BatteryNotesBatteryLowTemplateSensor(
-    BinarySensorEntity, CoordinatorEntity[BatteryNotesCoordinator]
+    BinarySensorEntity, CoordinatorEntity[BatteryNotesCoordinator], RestoreEntity
 ):
     """Represents a low battery threshold binary sensor."""
 

custom_components/battery_notes/coordinator.py

@@ -58,7 +58,7 @@ class BatteryNotesCoordinator(DataUpdateCoordinator):
     battery_type: str
     battery_quantity: int
     battery_low_threshold: int
-    battery_low_template: str
+    battery_low_template: str | None
     wrapped_battery: RegistryEntry
     _current_battery_level: str = None
     enable_replaced: bool = True