Cleanup homekit callbacks and jobs

Author: bdraco

homeassistant/components/homekit/__init__.py

@@ -551,7 +551,7 @@ def _start(self, bridged_states):
             )
 
         _LOGGER.debug("Driver start for %s", self._name)
-        self.hass.async_add_executor_job(self.driver.start)
+        self.hass.add_job(self.driver.start)
         self.status = STATUS_RUNNING
 
     async def async_stop(self, *args):
@@ -561,7 +561,7 @@ async def async_stop(self, *args):
         self.status = STATUS_STOPPED
 
         _LOGGER.debug("Driver stop for %s", self._name)
-        self.hass.async_add_executor_job(self.driver.stop)
+        self.hass.add_job(self.driver.stop)
 
     @callback
     def _async_configure_linked_battery_sensors(self, ent_reg, device_lookup, state):

homeassistant/components/homekit/accessories.py

@@ -302,8 +302,10 @@ async def run_handler(self):
         Run inside the Home Assistant event loop.
         """
         state = self.hass.states.get(self.entity_id)
-        self.hass.async_add_executor_job(self.update_state_callback, None, None, state)
-        async_track_state_change(self.hass, self.entity_id, self.update_state_callback)
+        await self.async_update_state_callback(None, None, state)
+        async_track_state_change(
+            self.hass, self.entity_id, self.async_update_state_callback
+        )
 
         battery_charging_state = None
         battery_state = None
@@ -316,7 +318,7 @@ async def run_handler(self):
                 ATTR_BATTERY_CHARGING
             )
             async_track_state_change(
-                self.hass, self.linked_battery_sensor, self.update_linked_battery
+                self.hass, self.linked_battery_sensor, self.async_update_linked_battery
             )
         else:
             battery_state = state.attributes.get(ATTR_BATTERY_LEVEL)
@@ -328,7 +330,7 @@ async def run_handler(self):
             async_track_state_change(
                 self.hass,
                 self.linked_battery_charging_sensor,
-                self.update_linked_battery_charging,
+                self.async_update_linked_battery_charging,
             )
         elif battery_charging_state is None:
             battery_charging_state = state.attributes.get(ATTR_BATTERY_CHARGING)
@@ -338,8 +340,9 @@ async def run_handler(self):
                 self.update_battery, battery_state, battery_charging_state
             )
 
-    @ha_callback
-    def update_state_callback(self, entity_id=None, old_state=None, new_state=None):
+    async def async_update_state_callback(
+        self, entity_id=None, old_state=None, new_state=None
+    ):
         """Handle state change listener callback."""
         _LOGGER.debug("New_state: %s", new_state)
         if new_state is None:
@@ -357,28 +360,28 @@ def update_state_callback(self, entity_id=None, old_state=None, new_state=None):
         ):
             battery_charging_state = new_state.attributes.get(ATTR_BATTERY_CHARGING)
         if battery_state is not None or battery_charging_state is not None:
-            self.hass.async_add_executor_job(
+            await self.hass.async_add_executor_job(
                 self.update_battery, battery_state, battery_charging_state
             )
-        self.hass.async_add_executor_job(self.update_state, new_state)
+        await self.hass.async_add_executor_job(self.update_state, new_state)
 
-    @ha_callback
-    def update_linked_battery(self, entity_id=None, old_state=None, new_state=None):
+    async def async_update_linked_battery(
+        self, entity_id=None, old_state=None, new_state=None
+    ):
         """Handle linked battery sensor state change listener callback."""
         if self.linked_battery_charging_sensor:
             battery_charging_state = None
         else:
             battery_charging_state = new_state.attributes.get(ATTR_BATTERY_CHARGING)
-        self.hass.async_add_executor_job(
+        await self.hass.async_add_executor_job(
             self.update_battery, new_state.state, battery_charging_state,
         )
 
-    @ha_callback
-    def update_linked_battery_charging(
+    async def async_update_linked_battery_charging(
         self, entity_id=None, old_state=None, new_state=None
     ):
         """Handle linked battery charging sensor state change listener callback."""
-        self.hass.async_add_executor_job(
+        await self.hass.async_add_executor_job(
             self.update_battery, None, new_state.state == STATE_ON
         )