generated from ludeeus/integration_blueprint
-
Notifications
You must be signed in to change notification settings - Fork 10
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add independent sensors and remove attributes from device_tracker (#30)
Add proper sensors and binary_sensors to represent information previously stored as attributes in the device_tracker. Some outstanding issues still, so this is a work in progress.
- Loading branch information
1 parent
0168db7
commit 12b1bb3
Showing
5 changed files
with
503 additions
and
43 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,129 @@ | ||
| """Binary sensor platform that adds support for Leaf Spy.""" | ||
| import logging | ||
| from dataclasses import dataclass, field | ||
| from typing import Any, Callable | ||
|
|
||
| from homeassistant.components.binary_sensor import ( | ||
| BinarySensorDeviceClass, | ||
| BinarySensorEntity, | ||
| BinarySensorEntityDescription, | ||
| ) | ||
| from homeassistant.core import HomeAssistant | ||
| from homeassistant.helpers.dispatcher import async_dispatcher_connect | ||
| from homeassistant.helpers.entity_platform import AddEntitiesCallback | ||
| from homeassistant.helpers.restore_state import RestoreEntity | ||
| from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType | ||
| from homeassistant.util import slugify | ||
|
|
||
| from .const import DOMAIN | ||
|
|
||
| _LOGGER = logging.getLogger(__name__) | ||
|
|
||
| @dataclass(frozen=True) | ||
| class LeafSpyBinarySensorDescription(BinarySensorEntityDescription): | ||
| """Describes Leaf Spy binary sensor.""" | ||
| value_fn: Callable[[dict], Any] = field(default=lambda data: None) | ||
| transform_fn: Callable[[Any], bool] = field(default=lambda x: bool(x)) | ||
|
|
||
| BINARY_SENSOR_TYPES = [ | ||
| LeafSpyBinarySensorDescription( | ||
| key="power switch", | ||
| translation_key="power_switch_state", | ||
| device_class=BinarySensorDeviceClass.POWER, | ||
| value_fn=lambda data: data.get("PwrSw"), | ||
| transform_fn=lambda x: x == 1, | ||
| icon="mdi:power", | ||
| ) | ||
| ] | ||
|
|
||
| async def async_setup_entry( | ||
| hass: HomeAssistant, | ||
| entry: ConfigType, | ||
| async_add_entities: AddEntitiesCallback, | ||
| discovery_info: DiscoveryInfoType | None = None | ||
| ) -> bool: | ||
| """Set up Leaf Spy binary sensors based on a config entry.""" | ||
| if 'binary_sensors' not in hass.data[DOMAIN]: | ||
| hass.data[DOMAIN]['binary_sensors'] = {} | ||
|
|
||
| async def _process_message(context, message): | ||
| """Process incoming sensor messages.""" | ||
| try: | ||
| _LOGGER.debug("Incoming message: %s", message) | ||
| if 'VIN' not in message: | ||
| return | ||
|
|
||
| dev_id = slugify(f'leaf_{message["VIN"]}') | ||
|
|
||
| # Create and update binary sensors for each description | ||
| for description in BINARY_SENSOR_TYPES: | ||
| sensor_id = f"{dev_id}_{description.key}" | ||
| value = description.value_fn(message) | ||
| _LOGGER.debug("Binary Sensor '%s': Raw data=%s, Parsed value=%s", description.key, message, value) | ||
|
|
||
| value = description.transform_fn(value) | ||
| _LOGGER.debug("Binary Sensor '%s': Transformed value=%s", description.key, value) | ||
|
|
||
| if value is not None: | ||
| sensor = hass.data[DOMAIN]['binary_sensors'].get(sensor_id) | ||
| if sensor is not None: | ||
| sensor.update_state(value) | ||
| else: | ||
| sensor = LeafSpyBinarySensor(dev_id, description, value) | ||
| hass.data[DOMAIN]['binary_sensors'][sensor_id] = sensor | ||
| async_add_entities([sensor]) | ||
|
|
||
| except Exception as err: | ||
| _LOGGER.error("Error processing Leaf Spy message: %s", err) | ||
|
|
||
| async_dispatcher_connect(hass, DOMAIN, _process_message) | ||
| return True | ||
|
|
||
|
|
||
| class LeafSpyBinarySensor(BinarySensorEntity, RestoreEntity): | ||
| """Representation of a Leaf Spy binary sensor.""" | ||
|
|
||
| def __init__(self, device_id: str, description: LeafSpyBinarySensorDescription, initial_value): | ||
| """Initialize the binary sensor.""" | ||
| self._device_id = device_id | ||
| self._value = initial_value | ||
| self.entity_description = description | ||
|
|
||
| @property | ||
| def unique_id(self): | ||
| """Return a unique ID.""" | ||
| return f"{self._device_id}_{self.entity_description.key}" | ||
|
|
||
| @property | ||
| def name(self): | ||
| """Return the name of the binary sensor.""" | ||
| return f"Leaf {self.entity_description.key}" | ||
|
|
||
| @property | ||
| def is_on(self): | ||
| """Return true if the binary sensor is on.""" | ||
| return self._value | ||
|
|
||
| @property | ||
| def device_info(self): | ||
| """Return device information.""" | ||
| return { | ||
| "identifiers": {(DOMAIN, self._device_id)}, | ||
| } | ||
|
|
||
| def update_state(self, new_value): | ||
| """Update the binary sensor state.""" | ||
| self._value = new_value | ||
| self.async_write_ha_state() | ||
|
|
||
| async def async_added_to_hass(self): | ||
| """Restore last known state.""" | ||
| await super().async_added_to_hass() | ||
|
|
||
| last_state = await self.async_get_last_state() | ||
| if last_state: | ||
| try: | ||
| transform_fn = self.entity_description.transform_fn | ||
| self._value = transform_fn(last_state.state) | ||
| except (ValueError, TypeError): | ||
| _LOGGER.warning(f"Could not restore state for {self.name}") |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.