69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
from __future__ import annotations
|
|
|
|
from homeassistant.components.button import ButtonEntity
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
|
|
from .const import DOMAIN, SERVICE_NEXT_SLIDE, SERVICE_REFRESH_ALBUM, ATTR_ENTRY_ID
|
|
from .coordinator import AlbumCoordinator
|
|
|
|
|
|
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None:
|
|
coordinator: AlbumCoordinator = hass.data[DOMAIN][entry.entry_id]["coordinator"]
|
|
async_add_entities(
|
|
[
|
|
NextSlideButton(hass, entry, coordinator),
|
|
RefreshAlbumButton(hass, entry, coordinator),
|
|
]
|
|
)
|
|
|
|
|
|
class _BaseButton(ButtonEntity):
|
|
_attr_has_entity_name = True
|
|
|
|
def __init__(self, hass: HomeAssistant, entry: ConfigEntry, coordinator: AlbumCoordinator) -> None:
|
|
self.hass = hass
|
|
self.entry = entry
|
|
self.coordinator = coordinator
|
|
|
|
@property
|
|
def device_info(self):
|
|
return {
|
|
"identifiers": {(DOMAIN, self.entry.entry_id)},
|
|
"name": f"Album Slideshow {self.entry.title}",
|
|
"manufacturer": "Album Slideshow",
|
|
}
|
|
|
|
|
|
class NextSlideButton(_BaseButton):
|
|
def __init__(self, hass: HomeAssistant, entry: ConfigEntry, coordinator: AlbumCoordinator) -> None:
|
|
super().__init__(hass, entry, coordinator)
|
|
self._attr_unique_id = f"{entry.entry_id}_next_button"
|
|
self._attr_name = "Next slide"
|
|
self._attr_icon = "mdi:skip-next"
|
|
|
|
async def async_press(self) -> None:
|
|
await self.hass.services.async_call(
|
|
DOMAIN,
|
|
SERVICE_NEXT_SLIDE,
|
|
{ATTR_ENTRY_ID: self.entry.entry_id},
|
|
blocking=False,
|
|
)
|
|
|
|
|
|
class RefreshAlbumButton(_BaseButton):
|
|
def __init__(self, hass: HomeAssistant, entry: ConfigEntry, coordinator: AlbumCoordinator) -> None:
|
|
super().__init__(hass, entry, coordinator)
|
|
self._attr_unique_id = f"{entry.entry_id}_refresh_button"
|
|
self._attr_name = "Refresh album"
|
|
self._attr_icon = "mdi:refresh"
|
|
|
|
async def async_press(self) -> None:
|
|
await self.hass.services.async_call(
|
|
DOMAIN,
|
|
SERVICE_REFRESH_ALBUM,
|
|
{ATTR_ENTRY_ID: self.entry.entry_id},
|
|
blocking=False,
|
|
)
|