Work on main Schedule, support for holidays

data/schedule.py

@@ -0,0 +1,128 @@
+import json
+from dacite import from_dict
+from dataclasses import dataclass, field
+from datetime import datetime, timedelta
+from enums.platforms import Platforms
+from functions.config import get
+from typing import Dict, Optional, List
+
+from functions.timeFormatters import fromArray
+
+
+@dataclass
+class Holiday:
+    start_list: List[int]
+    end_list: List[int]
+    start_date: datetime = field(init=False)
+    end_date: datetime = field(init=False)
+    duration: timedelta = field(init=False)
+
+    def __post_init__(self):
+        self.start_date = fromArray(self.start_list)
+        self.end_date = fromArray(self.end_list)
+        self.duration = self.end_date - self.start_date
+
+    def has_passed(self, current_day: datetime) -> bool:
+        """
+        Check if a holiday has passed already
+        """
+        return current_day > self.end_date
+
+
+@dataclass
+class Course:
+    day: str
+    week: int
+    course_dict: Dict
+
+
+@dataclass
+class Location:
+    campus: str
+    building: str
+    room: str
+
+
+@dataclass
+class Timeslot:
+    course: Course
+    start_time: int
+    end_time: int
+    canceled: bool = False
+    is_special: bool = False
+    location: Optional[Location] = None
+    online_link: Optional[str] = None
+    online_platform: Optional[Platforms] = None
+
+
+@dataclass
+class Schedule:
+    day: datetime
+    schedule_dict: Dict = field(init=False)
+    start_date: datetime
+    end_date: datetime
+    semester_over: bool = False
+    holiday_offset: int = 0
+    current_holiday: Optional[Holiday] = None
+
+    def __post_init__(self):
+        self.schedule_dict: Dict = self.load_schedule_file()
+        self.start_date = fromArray(self.schedule_dict["semester_start"])
+        self.end_date = fromArray(self.schedule_dict["semester_end"])
+
+        # Semester is over
+        if self.end_date <= self.day:
+            self.semester_over = True
+            return
+
+        self.check_holidays()
+
+        # Show schedule for after holidays
+        if self.current_holiday is not None:
+            self.day = self.current_holiday.end_date + timedelta(days=1)
+
+    def check_holidays(self):
+        """
+        Do all holiday-related stuff here to avoid multiple loops
+        """
+        for hol_entry in self.schedule_dict.get("holidays", []):
+            holiday: Holiday = from_dict(Holiday, hol_entry)
+
+            # Hasn't happened yet, don't care
+            if holiday.start_date > self.day:
+                continue
+
+            # In the past: add the offset
+            if holiday.has_passed(self.day):
+                self.holiday_offset += (self.day - holiday.end_date) // 7
+            elif holiday.start_date <= self.day <= holiday.end_date:
+                self.current_holiday = holiday
+
+    def load_schedule_file(self) -> Dict:
+        """
+        Load the schedule from the JSON file
+        """
+        semester = get("semester")
+        year = get("year")
+
+        with open(f"files/{year}{semester}.json", "r") as fp:
+            return json.load(fp)
+
+    def get_week(self) -> int:
+        """
+        Get the current week of the semester
+        """
+        diff: timedelta = self.day - self.start_date
+
+        # Hasn't started yet, show week 1
+        if diff.days < 0:
+            return 1
+
+        # Add +1 at the end because week 1 would be 0 as it's not over yet
+        return (diff.days // 7) + self.holiday_offset + 1
+
+    def create_schedule(self):
+        """
+        Create the schedule for the current week
+        """
+        week: int = self.get_week()

functions/timeFormatters.py

@@ -1,8 +1,12 @@
 import datetime
+from typing import List
+
 import dateutil.relativedelta
 import pytz
 import time
 
+from functions import stringFormatters
+
 
 def epochToDate(epochTimeStamp, strFormat="%d/%m/%Y om %H:%M:%S"):
     now = dateTimeNow()
@@ -147,8 +151,16 @@ def intToWeekday(day):
     return ["Maandag", "Dinsdag", "Woensdag", "Donderdag", "Vrijdag", "Zaterdag", "Zondag"][day]
 
 
-def fromString(timeString: str, formatString="%d/%m/%Y"):
+def fromString(timeString: str, formatString="%d/%m/%Y", tzinfo=pytz.timezone("Europe/Brussels")):
     """
     Constructs a datetime object from an input string
     """
-    return datetime.datetime.strptime(timeString, formatString)
+    return datetime.datetime.strptime(timeString, formatString).replace(tzinfo=tzinfo)
+
+
+def fromArray(data: List[int]) -> datetime:
+    day = stringFormatters.leadingZero(str(data[0]))
+    month = stringFormatters.leadingZero(str(data[1]))
+    year = str(data[2])
+
+    return fromString(f"{day}/{month}/{year}")

requirements.txt

@@ -15,3 +15,5 @@ feedparser==6.0.2
 googletrans==4.0.0rc1
 quart==0.15.1
 Quart-CORS==0.5.0
+attrs~=21.2.0
+dacite~=1.6.0