mirror of
https://github.com/elisspace/core.git
synced 2026-10-07 17:48:43 +00:00
Ville Skyttä
2fb3be50ab
* Make async_get_device connections Optional, default None * Remove unnecessary async_get_device connections arg usages Some of these were using an incorrect collection type, which didn't cause issues mostly just due to luck.
88 lines
2.8 KiB
Python
88 lines
2.8 KiB
Python
"""Base class for Acmeda Roller Blinds."""
|
|
import aiopulse
|
|
|
|
from homeassistant.core import callback
|
|
from homeassistant.helpers import entity
|
|
from homeassistant.helpers.device_registry import async_get_registry as get_dev_reg
|
|
from homeassistant.helpers.dispatcher import async_dispatcher_connect
|
|
from homeassistant.helpers.entity_registry import async_get_registry as get_ent_reg
|
|
|
|
from .const import ACMEDA_ENTITY_REMOVE, DOMAIN, LOGGER
|
|
|
|
|
|
class AcmedaBase(entity.Entity):
|
|
"""Base representation of an Acmeda roller."""
|
|
|
|
def __init__(self, roller: aiopulse.Roller):
|
|
"""Initialize the roller."""
|
|
self.roller = roller
|
|
|
|
async def async_remove_and_unregister(self):
|
|
"""Unregister from entity and device registry and call entity remove function."""
|
|
LOGGER.error("Removing %s %s", self.__class__.__name__, self.unique_id)
|
|
|
|
ent_registry = await get_ent_reg(self.hass)
|
|
if self.entity_id in ent_registry.entities:
|
|
ent_registry.async_remove(self.entity_id)
|
|
|
|
dev_registry = await get_dev_reg(self.hass)
|
|
device = dev_registry.async_get_device(identifiers={(DOMAIN, self.unique_id)})
|
|
if device is not None:
|
|
dev_registry.async_update_device(
|
|
device.id, remove_config_entry_id=self.registry_entry.config_entry_id
|
|
)
|
|
|
|
await self.async_remove()
|
|
|
|
async def async_added_to_hass(self):
|
|
"""Entity has been added to hass."""
|
|
self.roller.callback_subscribe(self.notify_update)
|
|
|
|
self.async_on_remove(
|
|
async_dispatcher_connect(
|
|
self.hass,
|
|
ACMEDA_ENTITY_REMOVE.format(self.roller.id),
|
|
self.async_remove_and_unregister,
|
|
)
|
|
)
|
|
|
|
async def async_will_remove_from_hass(self):
|
|
"""Entity being removed from hass."""
|
|
self.roller.callback_unsubscribe(self.notify_update)
|
|
|
|
@callback
|
|
def notify_update(self):
|
|
"""Write updated device state information."""
|
|
LOGGER.debug("Device update notification received: %s", self.name)
|
|
self.async_write_ha_state()
|
|
|
|
@property
|
|
def should_poll(self):
|
|
"""Report that Acmeda entities do not need polling."""
|
|
return False
|
|
|
|
@property
|
|
def unique_id(self):
|
|
"""Return the unique ID of this roller."""
|
|
return self.roller.id
|
|
|
|
@property
|
|
def device_id(self):
|
|
"""Return the ID of this roller."""
|
|
return self.roller.id
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of roller."""
|
|
return self.roller.name
|
|
|
|
@property
|
|
def device_info(self):
|
|
"""Return the device info."""
|
|
return {
|
|
"identifiers": {(DOMAIN, self.unique_id)},
|
|
"name": self.roller.name,
|
|
"manufacturer": "Rollease Acmeda",
|
|
"via_device": (DOMAIN, self.roller.hub.id),
|
|
}
|