Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3 
4app = FastAPI()
5 
6users = {}
7courses = {}
8registrations = {}
9tokens = {}
10user_id_counter = 1
11course_id_counter = 1
12registration_id_counter = 1
13token_counter = 1
14 
15def get_user_from_token(authorization: Optional[str] = Header(None)):
16    if not authorization:
17        raise HTTPException(status_code=401, detail="No auth header")
18    token = authorization.replace("Bearer ", "")
19    if token not in tokens:
20        raise HTTPException(status_code=401, detail="Invalid token")
21    return tokens[token]
22 
23@app.post("/signup")
24def signup(name: str, email: str):
25    global user_id_counter
26    user_id = user_id_counter
27    user_id_counter += 1
28    users[user_id] = {"id": user_id, "name": name, "email": email}
29    return {"id": user_id, "name": name, "email": email}
30 
31@app.post("/login")
32def login(email: str, name: str):
33    for uid, u in users.items():
34        if u["email"] == email and u["name"] == name:
35            global token_counter
36            token = f"tok_{token_counter}"
37            token_counter += 1
38            tokens[token] = uid
39            return {"token": token}
40    raise HTTPException(status_code=401, detail="Invalid credentials")
41 
42@app.post("/courses")
43def create_course(name: str, capacity: int, authorization: Optional[str] = Header(None)):
44    get_user_from_token(authorization)
45    global course_id_counter
46    course_id = course_id_counter
47    course_id_counter += 1
48    courses[course_id] = {"id": course_id, "name": name, "capacity": capacity}
49    return {"id": course_id, "name": name, "capacity": capacity}
50 
51@app.post("/register")
52def register(student_name: str, course_id: int, authorization: Optional[str] = Header(None)):
53    get_user_from_token(authorization)
54    if course_id not in courses:
55        raise HTTPException(status_code=404, detail="Course not found")
56    course = courses[course_id]
57    count = sum(1 for r in registrations.values() if r["course_id"] == course_id)
58    if count >= course["capacity"]:
59        raise HTTPException(status_code=400, detail="Course full")
60    global registration_id_counter
61    reg_id = registration_id_counter
62    registration_id_counter += 1
63    registrations[reg_id] = {"id": reg_id, "student_name": student_name, "course_id": course_id}
64    return {"id": reg_id, "student_name": student_name, "course_id": course_id}
65 
66@app.get("/courses")
67def list_courses(authorization: Optional[str] = Header(None)):
68    get_user_from_token(authorization)
69    result = []
70    for cid, course in courses.items():
71        count = sum(1 for r in registrations.values() if r["course_id"] == cid)
72        result.append({"id": cid, "name": course["name"], "capacity": course["capacity"], "enrolled": count})
73    return result
74 
75@app.get("/courses/{course_id}")
76def get_course(course_id: int, authorization: Optional[str] = Header(None)):
77    get_user_from_token(authorization)
78    if course_id not in courses:
79        raise HTTPException(status_code=404, detail="Course not found")
80    course = courses[course_id]
81    count = sum(1 for r in registrations.values() if r["course_id"] == course_id)
82    return {"id": course_id, "name": course["name"], "capacity": course["capacity"], "enrolled": count}
83 
84@app.get("/users/{user_id}")
85def get_user(user_id: int, authorization: Optional[str] = Header(None)):
86    get_user_from_token(authorization)
87    if user_id not in users:
88        raise HTTPException(status_code=404, detail="User not found")
89    return users[user_id]
90 
91@app.get("/registrations/{reg_id}")
92def get_registration(reg_id: int, authorization: Optional[str] = Header(None)):
93    get_user_from_token(authorization)
94    if reg_id not in registrations:
95        raise HTTPException(status_code=404, detail="Registration not found")
96    return registrations[reg_id]

1fastapi
2uvicorn