Update monthly spent sensor.

Author: dvejsada

custom_components/rohlikcz/const.py

@@ -34,6 +34,7 @@
 ICON_CALENDAR_REMOVE = "mdi:calendar-remove"
 ICON_INFO = "mdi:information-outline"
 ICON_DELIVERY_TIME = "mdi:timer-sand"
+ICON_MONTHLY_SPENT = "mdi:cash-register"
 
 """ Service attributes """
 ATTR_CONFIG_ENTRY_ID = "config_entry_id"

custom_components/rohlikcz/rohlik_api.py

@@ -191,7 +191,8 @@ async def get_data(self):
             "last_order": "/api/v3/orders/delivered?offset=0&limit=1",
             "premium_profile": "/services/frontend-service/premium/profile",
             "next_delivery_slot": "/services/frontend-service/timeslots-api/",
-            "delivery_announcements": "/services/frontend-service/announcements/delivery"
+            "delivery_announcements": "/services/frontend-service/announcements/delivery",
+            "delivered_orders": "/api/v3/orders/delivered?offset=0&limit=50"
         }
 
         result["login"] = await self.login(session)

custom_components/rohlikcz/sensor.py

@@ -16,10 +16,10 @@
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from .const import DOMAIN, ICON_UPDATE, ICON_CREDIT, ICON_NO_LIMIT, ICON_FREE_EXPRESS, ICON_DELIVERY, ICON_BAGS, \
     ICON_CART, ICON_ACCOUNT, ICON_EMAIL, ICON_PHONE, ICON_PREMIUM_DAYS, ICON_LAST_ORDER, ICON_NEXT_ORDER_SINCE, \
-    ICON_NEXT_ORDER_TILL, ICON_INFO, ICON_DELIVERY_TIME
+    ICON_NEXT_ORDER_TILL, ICON_INFO, ICON_DELIVERY_TIME, ICON_MONTHLY_SPENT
 from .entity import BaseEntity
 from .hub import RohlikAccount
-from .utils import extract_delivery_datetime
+from .utils import extract_delivery_datetime, calculate_current_month_orders_total
 
 SCAN_INTERVAL = timedelta(seconds=600)
 
@@ -48,7 +48,8 @@ async def async_setup_entry(
         NextOrderTill(rohlik_hub),
         NextOrderSince(rohlik_hub),
         DeliveryInfo(rohlik_hub),
-        DeliveryTime(rohlik_hub)
+        DeliveryTime(rohlik_hub),
+        MonthlySpent(rohlik_hub)
     ]
 
     if rohlik_hub.has_address:
@@ -423,6 +424,29 @@ async def async_will_remove_from_hass(self) -> None:
         self._rohlik_account.remove_callback(self.async_write_ha_state)
 
 
+class MonthlySpent(BaseEntity, SensorEntity):
+    """Sensor for amount spent in current month."""
+
+    _attr_translation_key = "monthly_spent"
+    _attr_should_poll = False
+
+    @property
+    def native_value(self) -> float | None:
+        """Returns amount spend within last month."""
+        _LOGGER.error(self._rohlik_account.data.get('delivered_orders', "[]"))
+        return calculate_current_month_orders_total(self._rohlik_account.data.get('delivered_orders', []))
+
+    @property
+    def icon(self) -> str:
+        return ICON_MONTHLY_SPENT
+
+    async def async_added_to_hass(self) -> None:
+        self._rohlik_account.register_callback(self.async_write_ha_state)
+
+    async def async_will_remove_from_hass(self) -> None:
+        self._rohlik_account.remove_callback(self.async_write_ha_state)
+
+
 class NoLimitOrders(BaseEntity, SensorEntity):
     """Sensor for remaining no limit orders."""
 

custom_components/rohlikcz/translations/cs.json

@@ -128,6 +128,10 @@
       },
       "delivery_time": {
         "name": "Čas doručení"
+      },
+      "monthly_spent": {
+        "name": "Měsíční útrata",
+        "unit_of_measurement": "Kč"
       }
     },
     "todo": {

custom_components/rohlikcz/translations/en.json

@@ -128,6 +128,10 @@
       },
       "delivery_time": {
         "name": "Delivery Time"
+      },
+      "monthly_spent": {
+        "name": "Spent This Month",
+        "unit_of_measurement": "CZK"
       }
     },
     "todo": {

custom_components/rohlikcz/utils.py

@@ -1,7 +1,45 @@
 from datetime import timedelta, datetime, time
+import json
 from zoneinfo import ZoneInfo
 import re
 
+
+def calculate_current_month_orders_total(orders: list) -> float|None:
+    """
+    Calculate the total amount of orders for the current month from JSON data.
+
+    Args:
+        orders (list): List of order dictionaries, each containing an 'orderTime' and 'priceComposition'
+
+    Returns:
+        float: Total amount of orders for the current month. Returns 0.0 if no orders
+               are found for the current month or None if the JSON is invalid.
+
+    """
+    try:
+
+        # Get current month pattern (e.g., "2025-06-")
+        current_month_pattern = datetime.now().strftime("%Y-%m-")
+
+        # Filter orders from current month and calculate sum
+        total_amount = 0.0
+
+        for order in orders:
+            try:
+                # Simple string check for current month
+                if current_month_pattern in order['orderTime']:
+                    amount = float(order['priceComposition']['total']['amount'])
+                    total_amount += amount
+            except (KeyError, ValueError, TypeError):
+                # Skip invalid orders
+                continue
+
+        return total_amount
+
+    except (json.JSONDecodeError, TypeError):
+        return None
+
+
 def extract_delivery_datetime(text: str) -> datetime | None:
     """
     Extract delivery time information from various formatted strings and return a datetime object.