From 118997b1ae505663dc4c6c819ddb9515755ce1a1 Mon Sep 17 00:00:00 2001
From: Maximilian Giller <m.giller.dev@gmail.com>
Date: Sun, 3 Mar 2019 13:53:07 +0100
Subject: [PATCH] Lowered abstraction level to keep generic sorting centralized

---
 Calendar/CalendarInterface.py | 36 ++++++++++++++++++++++++++++++-----
 1 file changed, 31 insertions(+), 5 deletions(-)

diff --git a/Calendar/CalendarInterface.py b/Calendar/CalendarInterface.py
index 588c3d0..cd7a659 100644
--- a/Calendar/CalendarInterface.py
+++ b/Calendar/CalendarInterface.py
@@ -1,18 +1,44 @@
 from DataSourceInterface import DataSourceInterface
+from datetime import datetime, timezone, timedelta
 
 class CalendarInterface (DataSourceInterface):
     """Interface for fetching and processing calendar event information."""
-    def get_upcoming_events(self):
+    def __init__(self):
+        self.loaded_events = self.__get_events__()
+        self.__sort_events__()
+
+    def __sort_events__(self):
+        self.loaded_events.sort(key=lambda x : x.begin_datetime)
+
+    def __get_events__(self):
         raise NotImplementedError("Functions needs to be implemented")
 
+    def get_upcoming_events(self):
+        return self.__get_events_to_filter__(lambda x : (x.begin_datetime - datetime.now(timezone.utc)) > timedelta(days=-1))
+
     def get_today_events(self):
-        raise NotImplementedError("Functions needs to be implemented")
+        return self.get_day_events(datetime.now(timezone.utc))
 
     def get_day_events(self, date):
-        raise NotImplementedError("Functions needs to be implemented")
+        return self.__get_events_to_filter__(lambda x : x.begin_datetime.strftime('%d-%m-%y') == date.strftime('%d-%m-%y'))
 
     def get_month_events(self, month = -1):
-        raise NotImplementedError("Functions needs to be implemented")
+        if month < 0:
+            month = datetime.now().month
+        return self.__get_events_to_filter__(lambda x : x.begin_datetime.month == month)
 
     def get_week_events(self, week = -1):
-        raise NotImplementedError("Functions needs to be implemented")
\ No newline at end of file
+        if week < 0 and week_starts_on == "Monday":
+            week = int(datetime.now().strftime('%W')) + 1
+        elif week < 0:
+            week = int(datetime.now().strftime('%U')) + 1
+
+        if week_starts_on == "Monday":
+            return self.__get_events_to_filter__(lambda x : int(x.begin_datetime.strftime('%W')) + 1 == week)
+        else:
+            return self.__get_events_to_filter__(lambda x : int(x.begin_datetime.strftime('%U')) + 1 == week)
+
+    def __get_events_to_filter__(self, event_filter):
+        if self.loaded_events is None:
+            return []
+        return [event for event in self.loaded_events if event_filter(event)]
\ No newline at end of file