diff --git a/homeassistant/components/plaato/__init__.py b/homeassistant/components/plaato/__init__.py index af93cfd51722f4..9857ef47b1ce8c 100644 --- a/homeassistant/components/plaato/__init__.py +++ b/homeassistant/components/plaato/__init__.py @@ -33,6 +33,7 @@ ATTR_BATCH_VOLUME = 'batch_volume' SENSOR_UPDATE = '{}_sensor_update'.format(DOMAIN) +SENSOR_DATA_KEY = '{}.{}'.format(DOMAIN, SENSOR) WEBHOOK_SCHEMA = vol.Schema({ vol.Required(ATTR_DEVICE_NAME): cv.string, @@ -74,6 +75,7 @@ async def async_setup_entry(hass, entry): async def async_unload_entry(hass, entry): """Unload a config entry.""" hass.components.webhook.async_unregister(entry.data[CONF_WEBHOOK_ID]) + hass.data[SENSOR_DATA_KEY]() await hass.config_entries.async_forward_entry_unload(entry, SENSOR) return True diff --git a/homeassistant/components/plaato/sensor.py b/homeassistant/components/plaato/sensor.py index 307d6a28793cc9..4a55e10444d760 100644 --- a/homeassistant/components/plaato/sensor.py +++ b/homeassistant/components/plaato/sensor.py @@ -8,13 +8,10 @@ from . import ( ATTR_ABV, ATTR_BATCH_VOLUME, ATTR_BPM, ATTR_CO2_VOLUME, ATTR_TEMP, ATTR_TEMP_UNIT, ATTR_VOLUME_UNIT, DOMAIN as PLAATO_DOMAIN, - PLAATO_DEVICE_ATTRS, PLAATO_DEVICE_SENSORS, SENSOR as SENSOR_DOMAIN, - SENSOR_UPDATE) + PLAATO_DEVICE_ATTRS, PLAATO_DEVICE_SENSORS, SENSOR_DATA_KEY, SENSOR_UPDATE) _LOGGER = logging.getLogger(__name__) -DATA_KEY = '{}.{}'.format(PLAATO_DOMAIN, SENSOR_DOMAIN) - async def async_setup_platform(hass, config, async_add_entities, discovery_info=None): @@ -50,7 +47,7 @@ async def _update_sensor(device_id): for entity in devices[device_id]: entity.async_schedule_update_ha_state() - hass.data[DATA_KEY] = async_dispatcher_connect( + hass.data[SENSOR_DATA_KEY] = async_dispatcher_connect( hass, SENSOR_UPDATE, _update_sensor )