mirror of
https://github.com/home-assistant/core.git
synced 2025-07-26 22:57:17 +00:00
Move ForkedDaapdUpdater to separate module (#137654)
* Move ForkedDaapdUpdater to separate module * Remove moved constants
This commit is contained in:
epenet
GitHub
parent
8ddae828f7
commit
58eb8e1598
142
homeassistant/components/forked_daapd/coordinator.py
Normal file
142
homeassistant/components/forked_daapd/coordinator.py
Normal file
@ -0,0 +1,142 @@
|
|||||||
|
"""Support forked_daapd media player."""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import asyncio
|
||||||
|
import logging
|
||||||
|
|
||||||
|
from homeassistant.exceptions import PlatformNotReady
|
||||||
|
from homeassistant.helpers.dispatcher import async_dispatcher_send
|
||||||
|
|
||||||
|
from .const import (
|
||||||
|
SIGNAL_ADD_ZONES,
|
||||||
|
SIGNAL_UPDATE_DATABASE,
|
||||||
|
SIGNAL_UPDATE_MASTER,
|
||||||
|
SIGNAL_UPDATE_OUTPUTS,
|
||||||
|
SIGNAL_UPDATE_PLAYER,
|
||||||
|
SIGNAL_UPDATE_QUEUE,
|
||||||
|
)
|
||||||
|
|
||||||
|
_LOGGER = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
WS_NOTIFY_EVENT_TYPES = ["player", "outputs", "volume", "options", "queue", "database"]
|
||||||
|
WEBSOCKET_RECONNECT_TIME = 30 # seconds
|
||||||
|
|
||||||
|
|
||||||
|
class ForkedDaapdUpdater:
|
||||||
|
"""Manage updates for the forked-daapd device."""
|
||||||
|
|
||||||
|
def __init__(self, hass, api, entry_id):
|
||||||
|
"""Initialize."""
|
||||||
|
self.hass = hass
|
||||||
|
self._api = api
|
||||||
|
self.websocket_handler = None
|
||||||
|
self._all_output_ids = set()
|
||||||
|
self._entry_id = entry_id
|
||||||
|
|
||||||
|
async def async_init(self):
|
||||||
|
"""Perform async portion of class initialization."""
|
||||||
|
if not (server_config := await self._api.get_request("config")):
|
||||||
|
raise PlatformNotReady
|
||||||
|
if websocket_port := server_config.get("websocket_port"):
|
||||||
|
self.websocket_handler = asyncio.create_task(
|
||||||
|
self._api.start_websocket_handler(
|
||||||
|
websocket_port,
|
||||||
|
WS_NOTIFY_EVENT_TYPES,
|
||||||
|
self._update,
|
||||||
|
WEBSOCKET_RECONNECT_TIME,
|
||||||
|
self._disconnected_callback,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
_LOGGER.error("Invalid websocket port")
|
||||||
|
|
||||||
|
async def _disconnected_callback(self):
|
||||||
|
"""Send update signals when the websocket gets disconnected."""
|
||||||
|
async_dispatcher_send(
|
||||||
|
self.hass, SIGNAL_UPDATE_MASTER.format(self._entry_id), False
|
||||||
|
)
|
||||||
|
async_dispatcher_send(
|
||||||
|
self.hass, SIGNAL_UPDATE_OUTPUTS.format(self._entry_id), []
|
||||||
|
)
|
||||||
|
|
||||||
|
async def _update(self, update_types):
|
||||||
|
"""Private update method."""
|
||||||
|
update_types = set(update_types)
|
||||||
|
update_events = {}
|
||||||
|
_LOGGER.debug("Updating %s", update_types)
|
||||||
|
if (
|
||||||
|
"queue" in update_types
|
||||||
|
): # update queue, queue before player for async_play_media
|
||||||
|
if queue := await self._api.get_request("queue"):
|
||||||
|
update_events["queue"] = asyncio.Event()
|
||||||
|
async_dispatcher_send(
|
||||||
|
self.hass,
|
||||||
|
SIGNAL_UPDATE_QUEUE.format(self._entry_id),
|
||||||
|
queue,
|
||||||
|
update_events["queue"],
|
||||||
|
)
|
||||||
|
# order of below don't matter
|
||||||
|
if not {"outputs", "volume"}.isdisjoint(update_types): # update outputs
|
||||||
|
if outputs := await self._api.get_request("outputs"):
|
||||||
|
outputs = outputs["outputs"]
|
||||||
|
update_events["outputs"] = (
|
||||||
|
asyncio.Event()
|
||||||
|
) # only for master, zones should ignore
|
||||||
|
async_dispatcher_send(
|
||||||
|
self.hass,
|
||||||
|
SIGNAL_UPDATE_OUTPUTS.format(self._entry_id),
|
||||||
|
outputs,
|
||||||
|
update_events["outputs"],
|
||||||
|
)
|
||||||
|
self._add_zones(outputs)
|
||||||
|
if not {"database"}.isdisjoint(update_types):
|
||||||
|
pipes, playlists = await asyncio.gather(
|
||||||
|
self._api.get_pipes(), self._api.get_playlists()
|
||||||
|
)
|
||||||
|
update_events["database"] = asyncio.Event()
|
||||||
|
async_dispatcher_send(
|
||||||
|
self.hass,
|
||||||
|
SIGNAL_UPDATE_DATABASE.format(self._entry_id),
|
||||||
|
pipes,
|
||||||
|
playlists,
|
||||||
|
update_events["database"],
|
||||||
|
)
|
||||||
|
if not {"update", "config"}.isdisjoint(update_types): # not supported
|
||||||
|
_LOGGER.debug("update/config notifications neither requested nor supported")
|
||||||
|
if not {"player", "options", "volume"}.isdisjoint(
|
||||||
|
update_types
|
||||||
|
): # update player
|
||||||
|
if player := await self._api.get_request("player"):
|
||||||
|
update_events["player"] = asyncio.Event()
|
||||||
|
if update_events.get("queue"):
|
||||||
|
await update_events[
|
||||||
|
"queue"
|
||||||
|
].wait() # make sure queue done before player for async_play_media
|
||||||
|
async_dispatcher_send(
|
||||||
|
self.hass,
|
||||||
|
SIGNAL_UPDATE_PLAYER.format(self._entry_id),
|
||||||
|
player,
|
||||||
|
update_events["player"],
|
||||||
|
)
|
||||||
|
if update_events:
|
||||||
|
await asyncio.wait(
|
||||||
|
[asyncio.create_task(event.wait()) for event in update_events.values()]
|
||||||
|
) # make sure callbacks done before update
|
||||||
|
async_dispatcher_send(
|
||||||
|
self.hass, SIGNAL_UPDATE_MASTER.format(self._entry_id), True
|
||||||
|
)
|
||||||
|
|
||||||
|
def _add_zones(self, outputs):
|
||||||
|
outputs_to_add = []
|
||||||
|
for output in outputs:
|
||||||
|
if output["id"] not in self._all_output_ids:
|
||||||
|
self._all_output_ids.add(output["id"])
|
||||||
|
outputs_to_add.append(output)
|
||||||
|
if outputs_to_add:
|
||||||
|
async_dispatcher_send(
|
||||||
|
self.hass,
|
||||||
|
SIGNAL_ADD_ZONES.format(self._entry_id),
|
||||||
|
self._api,
|
||||||
|
outputs_to_add,
|
||||||
|
)
|
||||||
@ -31,7 +31,6 @@ from homeassistant.components.spotify import (
|
|||||||
from homeassistant.config_entries import ConfigEntry
|
from homeassistant.config_entries import ConfigEntry
|
||||||
from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_PORT
|
from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_PORT
|
||||||
from homeassistant.core import HomeAssistant, callback
|
from homeassistant.core import HomeAssistant, callback
|
||||||
from homeassistant.exceptions import PlatformNotReady
|
|
||||||
from homeassistant.helpers.aiohttp_client import async_get_clientsession
|
from homeassistant.helpers.aiohttp_client import async_get_clientsession
|
||||||
from homeassistant.helpers.dispatcher import (
|
from homeassistant.helpers.dispatcher import (
|
||||||
async_dispatcher_connect,
|
async_dispatcher_connect,
|
||||||
@ -75,12 +74,10 @@ from .const import (
|
|||||||
SUPPORTED_FEATURES_ZONE,
|
SUPPORTED_FEATURES_ZONE,
|
||||||
TTS_TIMEOUT,
|
TTS_TIMEOUT,
|
||||||
)
|
)
|
||||||
|
from .coordinator import ForkedDaapdUpdater
|
||||||
|
|
||||||
_LOGGER = logging.getLogger(__name__)
|
_LOGGER = logging.getLogger(__name__)
|
||||||
|
|
||||||
WS_NOTIFY_EVENT_TYPES = ["player", "outputs", "volume", "options", "queue", "database"]
|
|
||||||
WEBSOCKET_RECONNECT_TIME = 30 # seconds
|
|
||||||
|
|
||||||
|
|
||||||
async def async_setup_entry(
|
async def async_setup_entry(
|
||||||
hass: HomeAssistant,
|
hass: HomeAssistant,
|
||||||
@ -897,122 +894,3 @@ class ForkedDaapdMaster(MediaPlayerEntity):
|
|||||||
if url := result.get("artwork_url"):
|
if url := result.get("artwork_url"):
|
||||||
return await self._async_fetch_image(self.api.full_url(url))
|
return await self._async_fetch_image(self.api.full_url(url))
|
||||||
return None, None
|
return None, None
|
||||||
|
|
||||||
|
|
||||||
class ForkedDaapdUpdater:
|
|
||||||
"""Manage updates for the forked-daapd device."""
|
|
||||||
|
|
||||||
def __init__(self, hass, api, entry_id):
|
|
||||||
"""Initialize."""
|
|
||||||
self.hass = hass
|
|
||||||
self._api = api
|
|
||||||
self.websocket_handler = None
|
|
||||||
self._all_output_ids = set()
|
|
||||||
self._entry_id = entry_id
|
|
||||||
|
|
||||||
async def async_init(self):
|
|
||||||
"""Perform async portion of class initialization."""
|
|
||||||
if not (server_config := await self._api.get_request("config")):
|
|
||||||
raise PlatformNotReady
|
|
||||||
if websocket_port := server_config.get("websocket_port"):
|
|
||||||
self.websocket_handler = asyncio.create_task(
|
|
||||||
self._api.start_websocket_handler(
|
|
||||||
websocket_port,
|
|
||||||
WS_NOTIFY_EVENT_TYPES,
|
|
||||||
self._update,
|
|
||||||
WEBSOCKET_RECONNECT_TIME,
|
|
||||||
self._disconnected_callback,
|
|
||||||
)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
_LOGGER.error("Invalid websocket port")
|
|
||||||
|
|
||||||
async def _disconnected_callback(self):
|
|
||||||
"""Send update signals when the websocket gets disconnected."""
|
|
||||||
async_dispatcher_send(
|
|
||||||
self.hass, SIGNAL_UPDATE_MASTER.format(self._entry_id), False
|
|
||||||
)
|
|
||||||
async_dispatcher_send(
|
|
||||||
self.hass, SIGNAL_UPDATE_OUTPUTS.format(self._entry_id), []
|
|
||||||
)
|
|
||||||
|
|
||||||
async def _update(self, update_types):
|
|
||||||
"""Private update method."""
|
|
||||||
update_types = set(update_types)
|
|
||||||
update_events = {}
|
|
||||||
_LOGGER.debug("Updating %s", update_types)
|
|
||||||
if (
|
|
||||||
"queue" in update_types
|
|
||||||
): # update queue, queue before player for async_play_media
|
|
||||||
if queue := await self._api.get_request("queue"):
|
|
||||||
update_events["queue"] = asyncio.Event()
|
|
||||||
async_dispatcher_send(
|
|
||||||
self.hass,
|
|
||||||
SIGNAL_UPDATE_QUEUE.format(self._entry_id),
|
|
||||||
queue,
|
|
||||||
update_events["queue"],
|
|
||||||
)
|
|
||||||
# order of below don't matter
|
|
||||||
if not {"outputs", "volume"}.isdisjoint(update_types): # update outputs
|
|
||||||
if outputs := await self._api.get_request("outputs"):
|
|
||||||
outputs = outputs["outputs"]
|
|
||||||
update_events["outputs"] = (
|
|
||||||
asyncio.Event()
|
|
||||||
) # only for master, zones should ignore
|
|
||||||
async_dispatcher_send(
|
|
||||||
self.hass,
|
|
||||||
SIGNAL_UPDATE_OUTPUTS.format(self._entry_id),
|
|
||||||
outputs,
|
|
||||||
update_events["outputs"],
|
|
||||||
)
|
|
||||||
self._add_zones(outputs)
|
|
||||||
if not {"database"}.isdisjoint(update_types):
|
|
||||||
pipes, playlists = await asyncio.gather(
|
|
||||||
self._api.get_pipes(), self._api.get_playlists()
|
|
||||||
)
|
|
||||||
update_events["database"] = asyncio.Event()
|
|
||||||
async_dispatcher_send(
|
|
||||||
self.hass,
|
|
||||||
SIGNAL_UPDATE_DATABASE.format(self._entry_id),
|
|
||||||
pipes,
|
|
||||||
playlists,
|
|
||||||
update_events["database"],
|
|
||||||
)
|
|
||||||
if not {"update", "config"}.isdisjoint(update_types): # not supported
|
|
||||||
_LOGGER.debug("update/config notifications neither requested nor supported")
|
|
||||||
if not {"player", "options", "volume"}.isdisjoint(
|
|
||||||
update_types
|
|
||||||
): # update player
|
|
||||||
if player := await self._api.get_request("player"):
|
|
||||||
update_events["player"] = asyncio.Event()
|
|
||||||
if update_events.get("queue"):
|
|
||||||
await update_events[
|
|
||||||
"queue"
|
|
||||||
].wait() # make sure queue done before player for async_play_media
|
|
||||||
async_dispatcher_send(
|
|
||||||
self.hass,
|
|
||||||
SIGNAL_UPDATE_PLAYER.format(self._entry_id),
|
|
||||||
player,
|
|
||||||
update_events["player"],
|
|
||||||
)
|
|
||||||
if update_events:
|
|
||||||
await asyncio.wait(
|
|
||||||
[asyncio.create_task(event.wait()) for event in update_events.values()]
|
|
||||||
) # make sure callbacks done before update
|
|
||||||
async_dispatcher_send(
|
|
||||||
self.hass, SIGNAL_UPDATE_MASTER.format(self._entry_id), True
|
|
||||||
)
|
|
||||||
|
|
||||||
def _add_zones(self, outputs):
|
|
||||||
outputs_to_add = []
|
|
||||||
for output in outputs:
|
|
||||||
if output["id"] not in self._all_output_ids:
|
|
||||||
self._all_output_ids.add(output["id"])
|
|
||||||
outputs_to_add.append(output)
|
|
||||||
if outputs_to_add:
|
|
||||||
async_dispatcher_send(
|
|
||||||
self.hass,
|
|
||||||
SIGNAL_ADD_ZONES.format(self._entry_id),
|
|
||||||
self._api,
|
|
||||||
outputs_to_add,
|
|
||||||
)
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user