Skip to content

Commit 57b2e35

Browse files
authored
Refactor time change listeners to use _register_listener
1 parent 0c0ff3c commit 57b2e35

File tree

1 file changed

+15
-6
lines changed

1 file changed

+15
-6
lines changed

custom_components/yidcal/tehilim_daily_pupa_sensor.py

Lines changed: 15 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -103,19 +103,28 @@ def __init__(self, hass: HomeAssistant, yidcal_helper) -> None:
103103
self.entity_id = "sensor.yidcal_tehilim_daily_pupa"
104104
self._state: str | None = None
105105

106+
async def async_added_to_hass(self) -> None:
107+
await super().async_added_to_hass()
108+
106109
# initial compute
107110
self.update()
108111

109112
# refresh after midnight (civil)
110-
async_track_time_change(
111-
hass, self._handle_midnight, hour=0, minute=0, second=1
113+
self._register_listener(
114+
async_track_time_change(
115+
self.hass, self._handle_midnight, hour=0, minute=0, second=1
116+
)
112117
)
113118

114-
async_track_state_change_event(
115-
hass, "binary_sensor.yidcal_no_melucha", self._handle_midnight
119+
self._register_listener(
120+
async_track_state_change_event(
121+
self.hass, "binary_sensor.yidcal_no_melucha", self._handle_midnight
122+
)
116123
)
117-
async_track_state_change_event(
118-
hass, "sensor.yidcal_holiday", self._handle_midnight
124+
self._register_listener(
125+
async_track_state_change_event(
126+
self.hass, "sensor.yidcal_holiday", self._handle_midnight
127+
)
119128
)
120129

121130
async def _handle_midnight(self, *_):

0 commit comments

Comments
 (0)