diff --git a/custom_components/tgtg/sensor.py b/custom_components/tgtg/sensor.py index a58ae99..0377ac9 100644 --- a/custom_components/tgtg/sensor.py +++ b/custom_components/tgtg/sensor.py @@ -29,6 +29,8 @@ ATTR_TOTAL_QUANTITY_ORDERED = "total_quantity_ordered" ATTR_PICKUP_WINDOW_CHANGED = "pickup_window_changed" ATTR_CANCEL_UNTIL = "cancel_until" +ATTR_LOGO_URL = "logo_url" +ATTR_COVER_URL = "cover_url" _LOGGER = logging.getLogger(DOMAIN) @@ -178,6 +180,11 @@ def extra_state_attributes(self) -> dict | None: + " " + self.tgtg_answer["item"]["item_value"]["code"] ) + + if "logo_picture" in self.tgtg_answer["item"]: + data[ATTR_LOGO_URL] = self.tgtg_answer["item"]["logo_picture"]["current_url"] + if "cover_picture" in self.tgtg_answer["item"]: + data[ATTR_COVER_URL] = self.tgtg_answer["item"]["cover_picture"]["current_url"] if "pickup_interval" in self.tgtg_answer: if "start" in self.tgtg_answer["pickup_interval"]: data[ATTR_PICKUP_START] = self.tgtg_answer["pickup_interval"]["start"]