Remove expired 17track.net packages from entity registry (#23001)

* Remove expired 17track.net packages from entity registry

* Reverse order
This commit is contained in:
Aaron Bach 2019-04-12 08:03:14 -06:00 committed by GitHub
parent 3d441dffad
commit 7d46ed0bf9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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