38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
"""Support for WeMo binary sensors."""
|
|
import asyncio
|
|
import logging
|
|
|
|
from homeassistant.components.binary_sensor import BinarySensorEntity
|
|
from homeassistant.helpers.dispatcher import async_dispatcher_connect
|
|
|
|
from .const import DOMAIN as WEMO_DOMAIN
|
|
from .entity import WemoSubscriptionEntity
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
async def async_setup_entry(hass, config_entry, async_add_entities):
|
|
"""Set up WeMo binary sensors."""
|
|
|
|
async def _discovered_wemo(device):
|
|
"""Handle a discovered Wemo device."""
|
|
async_add_entities([WemoBinarySensor(device)])
|
|
|
|
async_dispatcher_connect(hass, f"{WEMO_DOMAIN}.binary_sensor", _discovered_wemo)
|
|
|
|
await asyncio.gather(
|
|
*[
|
|
_discovered_wemo(device)
|
|
for device in hass.data[WEMO_DOMAIN]["pending"].pop("binary_sensor")
|
|
]
|
|
)
|
|
|
|
|
|
class WemoBinarySensor(WemoSubscriptionEntity, BinarySensorEntity):
|
|
"""Representation a WeMo binary sensor."""
|
|
|
|
def _update(self, force_update=True):
|
|
"""Update the sensor state."""
|
|
with self._wemo_exception_handler("update status"):
|
|
self._state = self.wemo.get_state(force_update)
|