Add restart function to local speaker

AlexxIT · Apr 16, 2024 · 732ab66 · 732ab66
1 parent 29b592d
commit 732ab66
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/custom_components/yandex_station/core/yandex_station.py b/custom_components/yandex_station/core/yandex_station.py
@@ -1,3 +1,4 @@
+import asyncio
 import binascii
 import json
 import logging
@@ -821,9 +822,6 @@ async def async_play_media(
             elif media_type == "json":
                 payload = json.loads(media_id)
 
-            elif RE_MUSIC_ID.match(media_id):
-                payload = {"command": "playMusic", "id": media_id, "type": media_type}
-
             elif media_type == "shopping_list":
                 coro = shopping_list.shopping_sync(self.hass, self.glagol)
                 await self.hass.async_create_background_task(coro, self.name)
@@ -835,6 +833,15 @@ async def async_play_media(
                 await self.response(card, request_id)
                 return
 
+            elif media_type == "restart":
+                await self.glagol.stop()
+                await asyncio.sleep(float(media_id))
+                await self.glagol.start_or_restart()
+                return
+
+            elif RE_MUSIC_ID.match(media_id):
+                payload = {"command": "playMusic", "id": media_id, "type": media_type}
+
             else:
                 _LOGGER.warning(f"Unsupported local media: {media_id}")
                 return