Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9students = {}
10practice_buddies = {}
11schedules = {}
12rooms = {}
13 
14user_id_counter = 1
15student_id_counter = 1
16schedule_id_counter = 1
17room_id_counter = 1
18promotion_id_counter = 1
19 
20def get_current_user(authorization: Optional[str] = Header(None)):
21    if not authorization:
22        raise HTTPException(status_code=401, detail="Missing auth header")
23    token = authorization.replace("Bearer ", "")
24    if token not in tokens:
25        raise HTTPException(status_code=401, detail="Invalid token")
26    return tokens[token]
27 
28@app.post("/signup")
29def signup(email: str, password: str, name: str):
30    global user_id_counter
31    if email in users:
32        raise HTTPException(status_code=400, detail="Email already exists")
33    user_id = user_id_counter
34    user_id_counter += 1
35    users[email] = {"id": user_id, "email": email, "password": password, "name": name}
36    return {"id": user_id, "email": email, "name": name}
37 
38@app.post("/login")
39def login(email: str, password: str):
40    user = users.get(email)
41    if not user or user["password"] != password:
42        raise HTTPException(status_code=401, detail="Invalid credentials")
43    token = secrets.token_hex(16)
44    tokens[token] = user["id"]
45    return {"token": token, "user_id": user["id"]}
46 
47@app.get("/student/{student_id}")
48def get_student(student_id: int, authorization: Optional[str] = Header(None)):
49    get_current_user(authorization)
50    student = students.get(student_id)
51    if not student:
52        raise HTTPException(status_code=404, detail="Student not found")
53    return student
54 
55@app.post("/student")
56def create_student(name: str, teacher_id: int, authorization: Optional[str] = Header(None)):
57    global student_id_counter
58    get_current_user(authorization)
59    student_id = student_id_counter
60    student_id_counter += 1
61    students[student_id] = {"id": student_id, "name": name, "teacher_id": teacher_id, "is_buddy": False}
62    return students[student_id]
63 
64@app.get("/schedule/{schedule_id}")
65def get_schedule(schedule_id: int, authorization: Optional[str] = Header(None)):
66    get_current_user(authorization)
67    schedule = schedules.get(schedule_id)
68    if not schedule:
69        raise HTTPException(status_code=404, detail="Schedule not found")
70    return schedule
71 
72@app.post("/schedule")
73def create_schedule(student_id: int, room_id: int, day: str, time: str, authorization: Optional[str] = Header(None)):
74    global schedule_id_counter
75    get_current_user(authorization)
76    schedule_id = schedule_id_counter
77    schedule_id_counter += 1
78    schedules[schedule_id] = {"id": schedule_id, "student_id": student_id, "room_id": room_id, "day": day, "time": time}
79    return schedules[schedule_id]
80 
81@app.get("/room/{room_id}")
82def get_room(room_id: int, authorization: Optional[str] = Header(None)):
83    get_current_user(authorization)
84    room = rooms.get(room_id)
85    if not room:
86        raise HTTPException(status_code=404, detail="Room not found")
87    return room
88 
89@app.post("/room")
90def create_room(name: str, capacity: int, authorization: Optional[str] = Header(None)):
91    global room_id_counter
92    get_current_user(authorization)
93    room_id = room_id_counter
94    room_id_counter += 1
95    rooms[room_id] = {"id": room_id, "name": name, "capacity": capacity}
96    return rooms[room_id]
97 
98@app.post("/promote")
99def promote_to_buddy(student_id: int, authorization: Optional[str] = Header(None)):
100    current_user_id = get_current_user(authorization)
101    student = students.get(student_id)
102    if not student:
103        raise HTTPException(status_code=404, detail="Student not found")
104    if student["teacher_id"] != current_user_id:
105        raise HTTPException(status_code=403, detail="Only the teacher can promote")
106    student["is_buddy"] = True
107    practice_buddies[student_id] = student
108    return {"message": f"Student {student_id} promoted to practice buddy", "student": student}

1fastapi
2uvicorn