feat: Add current month's budgeted amount for monitored categories

Signed-off-by: wxt9861 <34514130+wxt9861@users.noreply.github.com>

README.md

@@ -6,10 +6,11 @@ This component will retreieve the following data from your YNAB budget
 1. To be budgeted amount
 2. Current month's budgeted amount
 3. Current month's remaining balance of any specified category
-4. Current balance of any specified account
-5. Number of transactions needing approval
-6. Number of uncleared transactions
-7. Number of overspent categories
+4. Current month's budgeted amount of any specific category
+5. Current balance of any specified account
+6. Number of transactions needing approval
+7. Number of uncleared transactions
+8. Number of overspent categories
 
 To keep api usage low, the sensor updates every 5 minutes.
 

custom_components/ynab/__init__.py

@@ -243,9 +243,12 @@ async def update_data(self):
                 self.hass.data[DOMAIN_DATA].update(
                     [(category.name, category.balance / 1000)]
                 )
+                self.hass.data[DOMAIN_DATA].update(
+                    [(category.name + "_budgeted", category.budgeted / 1000)]
+                )
                 _LOGGER.debug(
                     "Received data for categories: %s",
-                    [category.name, category.balance / 1000],
+                    [category.name, category.balance / 1000, category.budgeted / 1000],
                 )
 
 

custom_components/ynab/sensor.py

@@ -65,6 +65,9 @@ async def async_update(self):
                     self.attr[category.replace(" ", "_").lower()] = self.hass.data[
                         DOMAIN_DATA
                     ].get(category)
+                    self.attr[
+                        (category + "_budgeted").replace(" ", "_").lower()
+                    ] = self.hass.data[DOMAIN_DATA].get(category + "_budgeted")
                 else:
                     category_error = CATEGORY_ERROR.format(category=category)
                     _LOGGER.error(category_error)