From e23cac8befb080a3e7e1a4a31f9c9a7d7bc7c052 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Fri, 20 Jun 2025 10:23:41 +0200
Subject: [PATCH] Simplify remove listener in kodi (#147183)

---
 homeassistant/components/kodi/__init__.py | 12 ++----------
 homeassistant/components/kodi/const.py    |  1 -
 2 files changed, 2 insertions(+), 11 deletions(-)

diff --git a/homeassistant/components/kodi/__init__.py b/homeassistant/components/kodi/__init__.py
index d3c7d4da724..5400d142f28 100644
--- a/homeassistant/components/kodi/__init__.py
+++ b/homeassistant/components/kodi/__init__.py
@@ -17,13 +17,7 @@ from homeassistant.const import (
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
 
-from .const import (
-    CONF_WS_PORT,
-    DATA_CONNECTION,
-    DATA_KODI,
-    DATA_REMOVE_LISTENER,
-    DOMAIN,
-)
+from .const import CONF_WS_PORT, DATA_CONNECTION, DATA_KODI, DOMAIN
 
 _LOGGER = logging.getLogger(__name__)
 PLATFORMS = [Platform.MEDIA_PLAYER]
@@ -58,13 +52,12 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     async def _close(event):
         await conn.close()
 
-    remove_stop_listener = hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, _close)
+    entry.async_on_unload(hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, _close))
 
     hass.data.setdefault(DOMAIN, {})
     hass.data[DOMAIN][entry.entry_id] = {
         DATA_CONNECTION: conn,
         DATA_KODI: kodi,
-        DATA_REMOVE_LISTENER: remove_stop_listener,
     }
 
     await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
@@ -78,6 +71,5 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     if unload_ok:
         data = hass.data[DOMAIN].pop(entry.entry_id)
         await data[DATA_CONNECTION].close()
-        data[DATA_REMOVE_LISTENER]()
 
     return unload_ok
diff --git a/homeassistant/components/kodi/const.py b/homeassistant/components/kodi/const.py
index 479b02e0fb5..167ea2a4725 100644
--- a/homeassistant/components/kodi/const.py
+++ b/homeassistant/components/kodi/const.py
@@ -6,7 +6,6 @@ CONF_WS_PORT = "ws_port"
 
 DATA_CONNECTION = "connection"
 DATA_KODI = "kodi"
-DATA_REMOVE_LISTENER = "remove_listener"
 
 DEFAULT_PORT = 8080
 DEFAULT_SSL = False