122 lines
4.3 KiB
Python
122 lines
4.3 KiB
Python
"""Entity representing a Sonos player."""
|
|
from __future__ import annotations
|
|
|
|
from abc import abstractmethod
|
|
import datetime
|
|
import logging
|
|
|
|
import soco.config as soco_config
|
|
from soco.core import SoCo
|
|
from soco.exceptions import SoCoException
|
|
|
|
from homeassistant.components import persistent_notification
|
|
import homeassistant.helpers.device_registry as dr
|
|
from homeassistant.helpers.dispatcher import async_dispatcher_connect
|
|
from homeassistant.helpers.entity import DeviceInfo, Entity
|
|
|
|
from .const import (
|
|
DATA_SONOS,
|
|
DOMAIN,
|
|
SONOS_FAVORITES_UPDATED,
|
|
SONOS_POLL_UPDATE,
|
|
SONOS_STATE_UPDATED,
|
|
)
|
|
from .speaker import SonosSpeaker
|
|
|
|
SUB_FAIL_URL = "https://www.home-assistant.io/integrations/sonos/#network-requirements"
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class SonosEntity(Entity):
|
|
"""Representation of a Sonos entity."""
|
|
|
|
_attr_should_poll = False
|
|
|
|
def __init__(self, speaker: SonosSpeaker) -> None:
|
|
"""Initialize a SonosEntity."""
|
|
self.speaker = speaker
|
|
|
|
async def async_added_to_hass(self) -> None:
|
|
"""Handle common setup when added to hass."""
|
|
self.hass.data[DATA_SONOS].entity_id_mappings[self.entity_id] = self.speaker
|
|
self.async_on_remove(
|
|
async_dispatcher_connect(
|
|
self.hass,
|
|
f"{SONOS_POLL_UPDATE}-{self.soco.uid}",
|
|
self.async_poll,
|
|
)
|
|
)
|
|
self.async_on_remove(
|
|
async_dispatcher_connect(
|
|
self.hass,
|
|
f"{SONOS_STATE_UPDATED}-{self.soco.uid}",
|
|
self.async_write_ha_state,
|
|
)
|
|
)
|
|
self.async_on_remove(
|
|
async_dispatcher_connect(
|
|
self.hass,
|
|
f"{SONOS_FAVORITES_UPDATED}-{self.soco.household_id}",
|
|
self.async_write_ha_state,
|
|
)
|
|
)
|
|
|
|
async def async_will_remove_from_hass(self) -> None:
|
|
"""Clean up when entity is removed."""
|
|
del self.hass.data[DATA_SONOS].entity_id_mappings[self.entity_id]
|
|
|
|
async def async_poll(self, now: datetime.datetime) -> None:
|
|
"""Poll the entity if subscriptions fail."""
|
|
if not self.speaker.subscriptions_failed:
|
|
if soco_config.EVENT_ADVERTISE_IP:
|
|
listener_msg = f"{self.speaker.subscription_address} (advertising as {soco_config.EVENT_ADVERTISE_IP})"
|
|
else:
|
|
listener_msg = self.speaker.subscription_address
|
|
message = f"{self.speaker.zone_name} cannot reach {listener_msg}, falling back to polling, functionality may be limited"
|
|
log_link_msg = f", see {SUB_FAIL_URL} for more details"
|
|
notification_link_msg = f'.\n\nSee <a href="{SUB_FAIL_URL}">Sonos documentation</a> for more details.'
|
|
_LOGGER.warning(message + log_link_msg)
|
|
persistent_notification.async_create(
|
|
self.hass,
|
|
message + notification_link_msg,
|
|
"Sonos networking issue",
|
|
"sonos_subscriptions_failed",
|
|
)
|
|
self.speaker.subscriptions_failed = True
|
|
await self.speaker.async_unsubscribe()
|
|
try:
|
|
await self._async_poll()
|
|
except (OSError, SoCoException) as ex:
|
|
_LOGGER.debug("Error connecting to %s: %s", self.entity_id, ex)
|
|
|
|
@abstractmethod
|
|
async def _async_poll(self) -> None:
|
|
"""Poll the specific functionality. Should be implemented by platforms if needed."""
|
|
|
|
@property
|
|
def soco(self) -> SoCo:
|
|
"""Return the speaker SoCo instance."""
|
|
return self.speaker.soco
|
|
|
|
@property
|
|
def device_info(self) -> DeviceInfo:
|
|
"""Return information about the device."""
|
|
return DeviceInfo(
|
|
identifiers={(DOMAIN, self.soco.uid)},
|
|
name=self.speaker.zone_name,
|
|
model=self.speaker.model_name.replace("Sonos ", ""),
|
|
sw_version=self.speaker.version,
|
|
connections={
|
|
(dr.CONNECTION_NETWORK_MAC, self.speaker.mac_address),
|
|
(dr.CONNECTION_UPNP, f"uuid:{self.speaker.uid}"),
|
|
},
|
|
manufacturer="Sonos",
|
|
suggested_area=self.speaker.zone_name,
|
|
configuration_url=f"http://{self.soco.ip_address}:1400/support/review",
|
|
)
|
|
|
|
@property
|
|
def available(self) -> bool:
|
|
"""Return whether this device is available."""
|
|
return self.speaker.available
|