Fix cloud streams for HA 2024.4 (#523)

dext0r · Apr 8, 2024 · 28e67aa · 28e67aa
1 parent 886f314
commit 28e67aa
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/custom_components/yandex_smart_home/cloud_stream.py b/custom_components/yandex_smart_home/cloud_stream.py
@@ -1,4 +1,5 @@
 """Implement the Yandex Smart Home cloud connection manager for video streaming."""
+
 import asyncio
 from datetime import timedelta
 import logging
@@ -29,6 +30,11 @@
 from pydantic import BaseModel
 import yarl
 
+try:
+    from homeassistant.helpers.http import KEY_HASS
+except ImportError:
+    KEY_HASS = "hass"  # type: ignore
+
 from .const import CLOUD_STREAM_BASE_URL
 
 _LOGGER = logging.getLogger(__name__)
@@ -58,7 +64,7 @@ class WebRequest:
 
     def __init__(self, hass: HomeAssistant, url: yarl.URL):
         """Initialize web request from url."""
-        self.app = {"hass": hass}
+        self.app = {KEY_HASS: hass}
         self._url = url
 
     @property