From 7d46ed0bf9d09e901a79f894ff0a9d064fc17d80 Mon Sep 17 00:00:00 2001 From: Aaron Bach Date: Fri, 12 Apr 2019 08:03:14 -0600 Subject: [PATCH] Remove expired 17track.net packages from entity registry (#23001) * Remove expired 17track.net packages from entity registry * Reverse order --- homeassistant/components/seventeentrack/sensor.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/seventeentrack/sensor.py b/homeassistant/components/seventeentrack/sensor.py index ff17d1a4c54..8a242b7737d 100644 --- a/homeassistant/components/seventeentrack/sensor.py +++ b/homeassistant/components/seventeentrack/sensor.py @@ -232,7 +232,8 @@ class SeventeenTrackPackageSensor(Entity): return # If the user has elected to not see delivered packages and one gets - # delivered, post a notification and delete the entity: + # delivered, post a notification, remove the entity from the UI, and + # delete it from the entity registry: if package.status == VALUE_DELIVERED and not self._data.show_delivered: _LOGGER.info('Package delivered: %s', self._tracking_number) self.hass.components.persistent_notification.create( @@ -245,6 +246,9 @@ class SeventeenTrackPackageSensor(Entity): title=NOTIFICATION_DELIVERED_TITLE, notification_id=NOTIFICATION_DELIVERED_ID_SCAFFOLD.format( self._tracking_number)) + + reg = self.hass.helpers.entity_registry.async_get_registry() + self.hass.async_create_task(reg.async_remove(self.entity_id)) self.hass.async_create_task(self.async_remove()) return