"""Nello.io lock platform.""" from itertools import filterfalse import logging import voluptuous as vol import homeassistant.helpers.config_validation as cv from homeassistant.components.lock import (LockDevice, PLATFORM_SCHEMA) from homeassistant.const import (CONF_PASSWORD, CONF_USERNAME) _LOGGER = logging.getLogger(__name__) ATTR_ADDRESS = 'address' ATTR_LOCATION_ID = 'location_id' EVENT_DOOR_BELL = 'nello_bell_ring' PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({ vol.Required(CONF_USERNAME): cv.string, vol.Required(CONF_PASSWORD): cv.string }) def setup_platform(hass, config, add_entities, discovery_info=None): """Set up the Nello lock platform.""" from pynello.private import Nello nello = Nello(config.get(CONF_USERNAME), config.get(CONF_PASSWORD)) add_entities([NelloLock(lock) for lock in nello.locations], True) class NelloLock(LockDevice): """Representation of a Nello lock.""" def __init__(self, nello_lock): """Initialize the lock.""" self._nello_lock = nello_lock self._device_attrs = None self._activity = None self._name = None @property def name(self): """Return the name of the lock.""" return self._name @property def is_locked(self): """Return true if lock is locked.""" return True @property def device_state_attributes(self): """Return the device specific state attributes.""" return self._device_attrs def update(self): """Update the nello lock properties.""" self._nello_lock.update() # Location identifiers location_id = self._nello_lock.location_id short_id = self._nello_lock.short_id address = self._nello_lock.address self._name = 'Nello {}'.format(short_id) self._device_attrs = { ATTR_ADDRESS: address, ATTR_LOCATION_ID: location_id } # Process recent activity activity = self._nello_lock.activity if self._activity: # Filter out old events new_activity = list( filterfalse(lambda x: x in self._activity, activity)) if new_activity: for act in new_activity: activity_type = act.get('type') if activity_type == 'bell.ring.denied': event_data = { 'address': address, 'date': act.get('date'), 'description': act.get('description'), 'location_id': location_id, 'short_id': short_id } self.hass.bus.fire(EVENT_DOOR_BELL, event_data) # Save the activity history so that we don't trigger an event twice self._activity = activity def unlock(self, **kwargs): """Unlock the device.""" if not self._nello_lock.open_door(): _LOGGER.error("Failed to unlock")