Add automatic quarter detection for Drexel course scraper

Co-authored-by: zohair.ul.hasan <[email protected]>

Author: cursoragent

README.md

@@ -44,7 +44,23 @@ python src/main.py
 
 The scraper will output a JSON file called `data.json` in the same directory as the scraper.
 
-You can modify the scraper to scrape other terms by changing the `year`, `quarter`, and `college_code` variables in `src/config.py`.
+#### Automatic Quarter Detection
+
+The scraper now automatically detects the current Drexel quarter based on the current date. You no longer need to manually update the `year` and `quarter` values in `src/config.py`. The system uses the following schedule:
+
+- **Fall Quarter (15)**: September 23 - December 31
+- **Winter Quarter (25)**: January 1 - March 22
+- **Spring Quarter (35)**: March 23 - June 14
+- **Summer Quarter (45)**: June 15 - September 22
+
+If you need to override the automatic detection (e.g., for testing or scraping a specific past/future quarter), you can set the `DREXEL_YEAR` and `DREXEL_QUARTER` environment variables:
+
+```bash
+export DREXEL_YEAR=2024
+export DREXEL_QUARTER=35  # Spring quarter
+```
+
+You can still modify the `college_code` variable in `src/config.py` to scrape a specific college.
 
 To view all the options that the scraper supports, run `python3 src/main.py --help` on Mac/Linux, or `python src/main.py --help` on Windows.  
 

src/config.py

@@ -1,12 +1,22 @@
 import os
 import sys
+from quarter_utils import get_current_quarter_and_year, get_quarter_name
 
-# in format YYYY (e.g. 2022)
-# example value: 2022
-year = "2024"
-# 15 for Fall, 25 for Winter, 35 for Spring, 45 for Summer
-# example value: 45
-quarter = "45"
+# Automatically determine the current quarter and year
+# Can be overridden with DREXEL_YEAR and DREXEL_QUARTER environment variables
+if "DREXEL_YEAR" in os.environ and "DREXEL_QUARTER" in os.environ:
+    year = os.environ["DREXEL_YEAR"]
+    quarter = os.environ["DREXEL_QUARTER"]
+    print(f"Using manually configured {get_quarter_name(quarter)} {year} quarter (code: {quarter})")
+else:
+    year, quarter = get_current_quarter_and_year()
+    print(f"Using auto-detected {get_quarter_name(quarter)} {year} quarter (code: {quarter})")
+
+# Note: These values are now automatically determined based on the current date
+# Fall (15): Late September - Mid December
+# Winter (25): Early January - Late March
+# Spring (35): Late March/Early April - Mid June
+# Summer (45): Late June - Early September
 # check college code by going to the tms website and selecting your college from the left sidebar
 # the URL bar should update and it should end with something like collCode=CI
 # the characters after the = sign is your college code

src/quarter_utils.py

@@ -0,0 +1,57 @@
+from datetime import datetime
+from typing import Tuple
+
+
+def get_current_quarter_and_year() -> Tuple[str, str]:
+    """
+    Determine the current Drexel quarter and year based on the current date.
+    
+    Drexel quarters:
+    - Fall (15): Late September - Mid December
+    - Winter (25): Early January - Late March
+    - Spring (35): Late March/Early April - Mid June
+    - Summer (45): Late June - Early September
+    
+    Returns:
+        Tuple of (year, quarter_code) as strings
+    """
+    now = datetime.now()
+    month = now.month
+    day = now.day
+    year = now.year
+    
+    # Determine quarter based on month and day
+    if month == 1 or month == 2 or (month == 3 and day <= 22):
+        # Winter quarter
+        quarter = "25"
+    elif (month == 3 and day > 22) or month == 4 or month == 5 or (month == 6 and day <= 14):
+        # Spring quarter
+        quarter = "35"
+    elif (month == 6 and day > 14) or month == 7 or month == 8 or (month == 9 and day <= 22):
+        # Summer quarter
+        quarter = "45"
+    else:
+        # Fall quarter (Sept 23 - Dec 31)
+        quarter = "15"
+    
+    # For Fall quarter, if we're in September-December, it's the current year
+    # For other quarters in January-August, they belong to the current year
+    return str(year), quarter
+
+
+def get_quarter_name(quarter_code: str) -> str:
+    """Get the human-readable name for a quarter code."""
+    quarter_names = {
+        "15": "Fall",
+        "25": "Winter",
+        "35": "Spring",
+        "45": "Summer"
+    }
+    return quarter_names.get(quarter_code, "Unknown")
+
+
+if __name__ == "__main__":
+    # Test the function
+    year, quarter = get_current_quarter_and_year()
+    quarter_name = get_quarter_name(quarter)
+    print(f"Current quarter: {quarter_name} {year} (code: {quarter})")