Lowered abstraction level to keep generic sorting centralized
This commit is contained in:
parent
385ae3da11
commit
118997b1ae
1 changed files with 31 additions and 5 deletions
|
@ -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")
|
||||
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)]
|
Loading…
Reference in a new issue