Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9grades = {}
10courses = {}
11user_ids = 1
12grade_ids = 1
13course_ids = 1
14 
15class SignupRequest(BaseModel):
16    username: str
17    password: str
18    role: str = "student"
19 
20class LoginRequest(BaseModel):
21    username: str
22    password: str
23 
24class GradeCreate(BaseModel):
25    student_id: int
26    course_id: int
27    grade: float
28 
29class CourseCreate(BaseModel):
30    name: str
31 
32def get_current_user(authorization: str = Header(None)):
33    if not authorization:
34        raise HTTPException(status_code=401, detail="No auth header")
35    token = authorization.replace("Bearer ", "")
36    if token not in tokens:
37        raise HTTPException(status_code=401, detail="Invalid token")
38    return tokens[token]
39 
40@app.post("/signup")
41def signup(req: SignupRequest):
42    global user_ids
43    for u in users.values():
44        if u["username"] == req.username:
45            raise HTTPException(status_code=400, detail="Username exists")
46    user_id = user_ids
47    user_ids += 1
48    users[user_id] = {"id": user_id, "username": req.username, "password": req.password, "role": req.role}
49    return {"id": user_id, "username": req.username, "role": req.role}
50 
51@app.post("/login")
52def login(req: LoginRequest):
53    for u in users.values():
54        if u["username"] == req.username and u["password"] == req.password:
55            token = secrets.token_hex(16)
56            tokens[token] = u
57            return {"token": token}
58    raise HTTPException(status_code=401, detail="Invalid credentials")
59 
60@app.post("/courses")
61def create_course(req: CourseCreate, authorization: str = Header(None)):
62    user = get_current_user(authorization)
63    if user["role"] != "teacher":
64        raise HTTPException(status_code=403, detail="Only teachers can create courses")
65    global course_ids
66    course_id = course_ids
67    course_ids += 1
68    courses[course_id] = {"id": course_id, "name": req.name, "teacher_id": user["id"]}
69    return courses[course_id]
70 
71@app.get("/courses/{course_id}")
72def get_course(course_id: int):
73    if course_id not in courses:
74        raise HTTPException(status_code=404, detail="Course not found")
75    return courses[course_id]
76 
77@app.post("/grades")
78def create_grade(req: GradeCreate, authorization: str = Header(None)):
79    user = get_current_user(authorization)
80    if user["role"] != "teacher":
81        raise HTTPException(status_code=403, detail="Only teachers can post grades")
82    if req.course_id not in courses:
83        raise HTTPException(status_code=404, detail="Course not found")
84    if req.student_id not in users:
85        raise HTTPException(status_code=404, detail="Student not found")
86    global grade_ids
87    grade_id = grade_ids
88    grade_ids += 1
89    grades[grade_id] = {"id": grade_id, "student_id": req.student_id, "course_id": req.course_id, "grade": req.grade}
90    return grades[grade_id]
91 
92@app.get("/grades/{grade_id}")
93def get_grade(grade_id: int):
94    if grade_id not in grades:
95        raise HTTPException(status_code=404, detail="Grade not found")
96    return grades[grade_id]
97 
98@app.get("/grades")
99def list_my_grades(authorization: str = Header(None)):
100    user = get_current_user(authorization)
101    if user["role"] != "student":
102        raise HTTPException(status_code=403, detail="Only students can view their grades")
103    student_grades = [g for g in grades.values() if g["student_id"] == user["id"]]
104    return student_grades

1fastapi
2uvicorn