Initial commit

routers/connections/arbor_router.py

@@ -0,0 +1,29 @@
+from fastapi import APIRouter, HTTPException
+import os
+import requests
+from base64 import b64decode
+
+router = APIRouter()
+
+def get_basic_auth_header(token: str) -> dict:
+    """Decode the base64 token and return the appropriate header."""
+    decoded_token = b64decode(token).decode('utf-8')
+    return {"Authorization": f"Basic {token}"}
+
+@router.get("/data/{id}")
+async def fetch_arbor_data(id: int, token: str):
+    url_mapping = {
+        1: os.environ["KS3_COURSE_CLASS_MEMBERSHIP_URL"],
+        2: os.environ["TEACHING_GROUP_MEMBERSHIPS_2023_2024_URL"],
+        3: os.environ["SCHEDULED_TIMETABLE_SLOTS_URL"],
+        4: os.environ["BEHAVIOURAL_INCIDENTS_REPORTING_URL"],
+        5: os.environ["Y7_LESSON_TIMETABLE_URL"]
+    }
+    if id not in url_mapping:
+        raise HTTPException(status_code=404, detail="Data ID not supported")
+
+    headers = get_basic_auth_header(token)
+    response = requests.get(url_mapping[id], headers=headers)
+    if response.status_code != 200:
+        raise HTTPException(status_code=response.status_code, detail="Failed to fetch data from Arbor")
+    return response.json()