From 020f1294b3f6579eb2c4c193e219690a879285d2 Mon Sep 17 00:00:00 2001 From: ov3rk1ll Date: Tue, 12 Nov 2024 09:37:12 +0100 Subject: [PATCH] add attribute for logo and cover image --- custom_components/tgtg/sensor.py | 7 +++++++ 1 file changed, 7 insertions(+) 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"]