new attribute category

FaserF · Sep 22, 2022 · f7557dd · f7557dd
1 parent 8f223de
commit f7557dd
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/custom_components/rewe/sensor.py b/custom_components/rewe/sensor.py
@@ -34,6 +34,7 @@
     ATTR_BASE_PRICE,
     ATTR_PICTURE,
     ATTR_VALID_DATE,
+    ATTR_CATEGORY,
 
     DOMAIN,
 )
@@ -103,7 +104,7 @@ def state(self) -> Optional[str]:
     @property
     def unit_of_measurement(self):
         """Return unit of measurement."""
-        return "date"
+        return "items"
 
     @property
     def session(self):
@@ -159,7 +160,7 @@ async def async_update(self):
                     discounts = []
                     n = 0
                     for category in data['categories']:
-                        _LOGGER.debug(f"Processing category: '{category}")
+                        _LOGGER.debug(f"Processing category: '{category['title']}")
                         if 'PAYBACK' in category['title']:  # ignore payback offers
                             n += 1
                             continue
@@ -175,6 +176,7 @@ async def async_update(self):
                                     ATTR_DISCOUNT_PRICE: item['priceData'],
                                     #ATTR_BASE_PRICE: item['subtitle']
                                     ATTR_PICTURE: item['images'],
+                                    ATTR_CATEGORY: category['title']
                                 }
                             )
                         n += 1