Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import random
4import string
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10courses = {}
11enrollments = {}
12next_user_id = 1
13next_course_id = 1
14next_enrollment_id = 1
15 
16def generate_token():
17    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
18 
19def get_current_user(authorization: Optional[str] = Header(None)):
20    if not authorization:
21        raise HTTPException(status_code=401, detail="Missing authorization header")
22    token = authorization.replace("Bearer ", "")
23    if token not in tokens:
24        raise HTTPException(status_code=401, detail="Invalid token")
25    return tokens[token]
26 
27@app.post("/signup")
28def signup(username: str, password: str, role: str = "student"):
29    global next_user_id
30    user_id = next_user_id
31    next_user_id += 1
32    users[user_id] = {"id": user_id, "username": username, "password": password, "role": role}
33    return {"id": user_id, "username": username, "role": role}
34 
35@app.post("/login")
36def login(username: str, password: str):
37    for user in users.values():
38        if user["username"] == username and user["password"] == password:
39            token = generate_token()
40            tokens[token] = user
41            return {"token": token}
42    raise HTTPException(status_code=401, detail="Invalid credentials")
43 
44@app.post("/courses")
45def create_course(name: str, instructor_id: int, authorization: Optional[str] = Header(None)):
46    user = get_current_user(authorization)
47    if user["role"] != "instructor":
48        raise HTTPException(status_code=403, detail="Only instructors can create courses")
49    global next_course_id
50    course_id = next_course_id
51    next_course_id += 1
52    courses[course_id] = {"id": course_id, "name": name, "instructor_id": instructor_id}
53    return courses[course_id]
54 
55@app.get("/courses/{course_id}")
56def get_course(course_id: int, authorization: Optional[str] = Header(None)):
57    get_current_user(authorization)
58    if course_id not in courses:
59        raise HTTPException(status_code=404, detail="Course not found")
60    return courses[course_id]
61 
62@app.post("/enrollments")
63def create_enrollment(student_id: int, course_id: int, authorization: Optional[str] = Header(None)):
64    user = get_current_user(authorization)
65    if user["role"] != "student":
66        raise HTTPException(status_code=403, detail="Only students can enroll")
67    if course_id not in courses:
68        raise HTTPException(status_code=404, detail="Course not found")
69    global next_enrollment_id
70    enrollment_id = next_enrollment_id
71    next_enrollment_id += 1
72    enrollments[enrollment_id] = {"id": enrollment_id, "student_id": student_id, "course_id": course_id, "grade": None}
73    return enrollments[enrollment_id]
74 
75@app.get("/enrollments/{enrollment_id}")
76def get_enrollment(enrollment_id: int, authorization: Optional[str] = Header(None)):
77    get_current_user(authorization)
78    if enrollment_id not in enrollments:
79        raise HTTPException(status_code=404, detail="Enrollment not found")
80    return enrollments[enrollment_id]
81 
82@app.patch("/enrollments/{enrollment_id}/grade")
83def set_grade(enrollment_id: int, grade: str, authorization: Optional[str] = Header(None)):
84    user = get_current_user(authorization)
85    if user["role"] != "instructor":
86        raise HTTPException(status_code=403, detail="Only instructors can set grades")
87    if enrollment_id not in enrollments:
88        raise HTTPException(status_code=404, detail="Enrollment not found")
89    enrollments[enrollment_id]["grade"] = grade
90    return enrollments[enrollment_id]
91 
92@app.get("/courses/{course_id}/enrollments")
93def get_enrollments_by_course(course_id: int, authorization: Optional[str] = Header(None)):
94    get_current_user(authorization)
95    if course_id not in courses:
96        raise HTTPException(status_code=404, detail="Course not found")
97    result = []
98    for e in enrollments.values():
99        if e["course_id"] == course_id:
100            result.append(e)
101    return result

1fastapi
2uvicorn