Merge pull request #177 from dipamsen/cache-calendar-2

Cache Academic Calendar

Author: Devansh-bit

.gitignore

@@ -11,7 +11,7 @@ venv
 .env
 .python-version
 
-ACADEMIC_CALENDAR_*.pdf
+*.pdf
 Academic_Cal-j/**
 final.json
 

Dockerfile

@@ -20,4 +20,6 @@ RUN chmod +x ./postinstall.sh
 
 COPY . .
 
+RUN python download-calendar.py
+
 CMD [ "./postinstall.sh", "gunicorn", "--bind", "0.0.0.0:8000", "wsgi:app" ]

app.py

@@ -213,7 +213,7 @@ def download_ics():
 
         courses = get_courses(session, sso_token, roll_number)
 
-        ics_content = generate_ics(courses, "")
+        ics_content = generate_ics(courses, "", is_web=True)
 
         # Create an in-memory file-like object for the ics content
         ics_file = io.BytesIO()
@@ -256,7 +256,7 @@ def image_parser():
 
         courses = build_courses_from_image(data)
 
-        ics_content = generate_ics(courses, "")
+        ics_content = generate_ics(courses, "", is_web=True)
 
         # Create an in-memory file-like object for the ics content
         ics_file = io.BytesIO()

download-calendar.py

@@ -0,0 +1,21 @@
+from utils.academic_calander_handler import get_latest_calendar
+import time
+
+MAX_RETRIES = 10
+WAIT_TIME = 2
+
+def main():
+	for attempt in range(1, MAX_RETRIES + 1):
+		print(f"Attempt {attempt} to download academic calendar...")
+		success = get_latest_calendar()
+		if success:
+			print("Academic calendar downloaded successfully.")
+			return
+		else:
+			print(f"Download failed. Retrying in {WAIT_TIME} seconds...")
+			time.sleep(WAIT_TIME)
+	print("Failed to download academic calendar after multiple attempts.")
+
+if __name__ == "__main__":
+	main()
+

timetable/generate_ics.py

@@ -8,7 +8,7 @@
 WORKING_DAYS = dates.get_dates()
 
 
-def generate_ics(courses: list[Course], output_filename):
+def generate_ics(courses: list[Course], output_filename, is_web=False):
     """
     Creates an ICS file `timetable.ics` with the timetable data present inside the 'timetable' parameter.
     """
@@ -52,7 +52,7 @@ def generate_ics(courses: list[Course], output_filename):
         event.add("dtend", holiday[1] + timedelta(days=1))
         cal.add_component(event)
 
-    for entry in academic_calander_handler.get_academic_calendar():
+    for entry in academic_calander_handler.get_academic_calendar(is_web):
         event = Event()
         event.add("summary", entry.event)
         event.add("dtstart",entry.start_date)

utils/academic_calander_handler.py

@@ -64,19 +64,27 @@ def delete_file(file):
 
 
 # fetch the latest academic calendar from the iitkgp website
-def get_latest_calendar():
+def get_latest_calendar(is_web=False):
     filename = get_latest_calendar_name()
     url = 'https://www.iitkgp.ac.in/assets/pdf/' + filename
 
-    ## delete any old academic calander pdf if exists
-    if (is_file_present(filename)):
+    if is_web and is_file_present(filename):
+        return True
+
+    # delete any old academic calendar pdf if exists
+    if is_file_present(filename):
         delete_file(filename)
 
-    with open(filename, "wb") as file:
-        response = requests.get(url)
-        file.write(response.content)
+    try:
+        with open(filename, "wb") as file:
+            response = requests.get(url)
+            response.raise_for_status()
+            file.write(response.content)
+    except Exception as e:
+        print(f"Error downloading calendar: {e}")
+        return False
 
-    if (is_file_present(filename)):
+    if is_file_present(filename):
         return True
     return False
 
@@ -153,8 +161,8 @@ def clean_temp_files():
             continue
 
 
-def get_academic_calendar() -> list[DataEntry]:
-    get_latest_calendar()
+def get_academic_calendar(is_web = False) -> list[DataEntry]:
+    get_latest_calendar(is_web)
     export_json()
 
     all_dates = merge_json()