evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Return unsubscribe callback from start.async_at_start (#69083)

Browse source 
* Return unsubscribe callback from start.async_at_start

* Update calling code
This commit is contained in:
Erik Montnemery 2022-04-01 23:34:36 +02:00 • committed by GitHub
parent 4b5996c5ed
commit ae9c2df691
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

4
homeassistant/components/group/__init__.py
View file

@ -454,7 +454,7 @@ class GroupEntity(Entity):
self.async_update_group_state() self.async_update_group_state()
self.async_write_ha_state() self.async_write_ha_state()
start.async_at_start(self.hass, _update_at_start) self.async_on_remove(start.async_at_start(self.hass, _update_at_start))
@callback @callback
def async_defer_or_update_ha_state(self) -> None: def async_defer_or_update_ha_state(self) -> None:
@ -689,7 +689,7 @@ class Group(Entity):
async def async_added_to_hass(self): async def async_added_to_hass(self):
"""Handle addition to Home Assistant.""" """Handle addition to Home Assistant."""
start.async_at_start(self.hass, self._async_start) self.async_on_remove(start.async_at_start(self.hass, self._async_start))
async def async_will_remove_from_hass(self): async def async_will_remove_from_hass(self):
"""Handle removal from Home Assistant.""" """Handle removal from Home Assistant."""

2
homeassistant/components/here_travel_time/sensor.py
View file

@ -247,7 +247,7 @@ class HERETravelTimeSensor(SensorEntity, CoordinatorEntity):
async def _update_at_start(_): async def _update_at_start(_):
await self.async_update() await self.async_update()
async_at_start(self.hass, _update_at_start) self.async_on_remove(async_at_start(self.hass, _update_at_start))
@property @property
def native_value(self) -> str | None: def native_value(self) -> str | None:

2
homeassistant/components/statistics/sensor.py
View file

@ -302,7 +302,7 @@ class StatisticsSensor(SensorEntity):
if "recorder" in self.hass.config.components: if "recorder" in self.hass.config.components:
self.hass.async_create_task(self._initialize_from_database()) self.hass.async_create_task(self._initialize_from_database())
async_at_start(self.hass, async_stats_sensor_startup) self.async_on_remove(async_at_start(self.hass, async_stats_sensor_startup))
def _add_state_to_queue(self, new_state: State) -> None: def _add_state_to_queue(self, new_state: State) -> None:
"""Add the state to the queue.""" """Add the state to the queue."""

8
homeassistant/helpers/start.py
View file

@ -4,13 +4,13 @@ from __future__ import annotations
from collections.abc import Awaitable, Callable from collections.abc import Awaitable, Callable
from homeassistant.const import EVENT_HOMEASSISTANT_START from homeassistant.const import EVENT_HOMEASSISTANT_START
from homeassistant.core import Event, HassJob, HomeAssistant, callback from homeassistant.core import CALLBACK_TYPE, Event, HassJob, HomeAssistant, callback
@callback @callback
def async_at_start( def async_at_start(
hass: HomeAssistant, at_start_cb: Callable[[HomeAssistant], Awaitable[None] | None] hass: HomeAssistant, at_start_cb: Callable[[HomeAssistant], Awaitable[None] | None]
) -> None: ) -> CALLBACK_TYPE:
"""Execute something when Home Assistant is started. """Execute something when Home Assistant is started.
Will execute it now if Home Assistant is already started. Will execute it now if Home Assistant is already started.
@ -18,10 +18,10 @@ def async_at_start(
at_start_job = HassJob(at_start_cb) at_start_job = HassJob(at_start_cb)
if hass.is_running: if hass.is_running:
hass.async_run_hass_job(at_start_job, hass) hass.async_run_hass_job(at_start_job, hass)
return return lambda: None
async def _matched_event(event: Event) -> None: async def _matched_event(event: Event) -> None:
"""Call the callback when Home Assistant started.""" """Call the callback when Home Assistant started."""
hass.async_run_hass_job(at_start_job, hass) hass.async_run_hass_job(at_start_job, hass)
hass.bus.async_listen_once(EVENT_HOMEASSISTANT_START, _matched_event) return hass.bus.async_listen_once(EVENT_HOMEASSISTANT_START, _matched_event)