Now actually implements the offsets

RogerSelwyn · Sep 1, 2020 · 3fc4c9a · 3fc4c9a
1 parent debf019
commit 3fc4c9a
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/custom_components/o365/calendar.py b/custom_components/o365/calendar.py
@@ -96,7 +96,7 @@ def __init__(self, hass, account, calendar_id, entity, entity_id):
         self.end_offset = entity.get(CONF_HOURS_FORWARD_TO_GET)
         self.search = entity.get(CONF_SEARCH)
         self.data = O365CalendarData(
-            account, calendar_id, self.search, self.max_results
+            account, calendar_id, self.search, self.max_results, self.start_offset, self.end_offset
         )
         self._event = {}
         self._name = entity.get(CONF_NAME)
@@ -140,10 +140,12 @@ async def async_update(self):
 
 
 class O365CalendarData:
-    def __init__(self, account, calendar_id, search=None, limit=999):
+    def __init__(self, account, calendar_id, search=None, limit=999, start_offset=None, end_offset=None):
         self.account = account
         self.calendar_id = calendar_id
         self.limit = limit
+        self.start_offset = start_offset
+        self.end_offset = end_offset
         self.schedule = self.account.schedule()
         self.calendar = self.schedule.get_calendar(calendar_id=self.calendar_id)
         self.search = search
@@ -174,6 +176,17 @@ async def async_get_events(self, hass, start_date, end_date):
 
     @Throttle(MIN_TIME_BETWEEN_UPDATES)
     async def async_update(self, hass):
+        if self.start_offset is None:
+            start = dt.start_of_local_day()
+        else:
+            start = dt.now() + timedelta(hours=self.start_offset)
+        if self.end_offset is None:
+            end = start + timedelta(days=1)
+        else:
+            end = dt.now() + timedelta(hours=self.end_offset)
+
+
+
         results = await hass.async_add_executor_job(self.o365_get_events,
             dt.start_of_local_day(), dt.start_of_local_day() + timedelta(days=1)
         )