From 4d60c10e5ce0ad8b04ce21622bb15b8a93c4f0c3 Mon Sep 17 00:00:00 2001
From: soloam <11949987+soloam@users.noreply.github.com>
Date: Fri, 3 May 2024 16:55:19 +0100
Subject: [PATCH] Fix only return bills if requested

---
 .../fireflyiii_integration/calendar.py           | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/custom_components/fireflyiii_integration/calendar.py b/custom_components/fireflyiii_integration/calendar.py
index d083c1b..ef21f49 100644
--- a/custom_components/fireflyiii_integration/calendar.py
+++ b/custom_components/fireflyiii_integration/calendar.py
@@ -43,18 +43,20 @@ async def async_setup_entry(
     if config_entry.options:
         config.update(config_entry.options)
 
-    coordinator = config[COORDINATOR]
+    coordinator: FireflyiiiCoordinator = config[COORDINATOR]
 
     user_locale = get_hass_locale(hass)
 
     bills = []
-    obj = FireflyiiiBillCalendarEntity(
-        coordinator,
-        FIREFLYIII_SENSOR_DESCRIPTIONS[FireflyiiiObjectType.BILLS],
-        locale=user_locale,
-    )
 
-    bills.append(obj)
+    if coordinator.user_data.get_bills:
+        obj = FireflyiiiBillCalendarEntity(
+            coordinator,
+            FIREFLYIII_SENSOR_DESCRIPTIONS[FireflyiiiObjectType.BILLS],
+            locale=user_locale,
+        )
+
+        bills.append(obj)
 
     calendars = []
     calendars.extend(bills)