Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict
4import secrets
5 
6app = FastAPI()
7 
8users: Dict[int, dict] = {}
9next_user_id = 1
10tokens: Dict[str, int] = {}
11 
12grades: Dict[int, dict] = {}
13next_grade_id = 1
14 
15class SignupRequest(BaseModel):
16    username: str
17    password: str
18 
19class LoginRequest(BaseModel):
20    username: str
21    password: str
22 
23class GradeCreate(BaseModel):
24    student_name: str
25    subject: str
26    score: float
27 
28class GradeUpdate(BaseModel):
29    student_name: Optional[str] = None
30    subject: Optional[str] = None
31    score: Optional[float] = None
32 
33def get_user_id_from_token(authorization: str = Header(...)):
34    if not authorization.startswith("Bearer "):
35        raise HTTPException(status_code=401, detail="Invalid token format")
36    token = authorization[7:]
37    user_id = tokens.get(token)
38    if user_id is None:
39        raise HTTPException(status_code=401, detail="Invalid token")
40    return user_id
41 
42@app.post("/signup")
43def signup(req: SignupRequest):
44    global next_user_id
45    user_id = next_user_id
46    next_user_id += 1
47    users[user_id] = {"username": req.username, "password": req.password}
48    return {"id": user_id, "username": req.username}
49 
50@app.post("/login")
51def login(req: LoginRequest):
52    for uid, u in users.items():
53        if u["username"] == req.username and u["password"] == req.password:
54            token = secrets.token_hex(16)
55            tokens[token] = uid
56            return {"token": token}
57    raise HTTPException(status_code=401, detail="Invalid credentials")
58 
59@app.post("/grades")
60def create_grade(grade: GradeCreate, authorization: str = Header(...)):
61    get_user_id_from_token(authorization)
62    global next_grade_id
63    grade_id = next_grade_id
64    next_grade_id += 1
65    grades[grade_id] = {
66        "id": grade_id,
67        "student_name": grade.student_name,
68        "subject": grade.subject,
69        "score": grade.score
70    }
71    return grades[grade_id]
72 
73@app.get("/grades/{grade_id}")
74def get_grade(grade_id: int, authorization: str = Header(...)):
75    get_user_id_from_token(authorization)
76    grade = grades.get(grade_id)
77    if grade is None:
78        raise HTTPException(status_code=404, detail="Grade not found")
79    return grade
80 
81@app.put("/grades/{grade_id}")
82def update_grade(grade_id: int, grade: GradeUpdate, authorization: str = Header(...)):
83    get_user_id_from_token(authorization)
84    if grade_id not in grades:
85        raise HTTPException(status_code=404, detail="Grade not found")
86    existing = grades[grade_id]
87    if grade.student_name is not None:
88        existing["student_name"] = grade.student_name
89    if grade.subject is not None:
90        existing["subject"] = grade.subject
91    if grade.score is not None:
92        existing["score"] = grade.score
93    return existing
94 
95@app.delete("/grades/{grade_id}")
96def delete_grade(grade_id: int, authorization: str = Header(...)):
97    get_user_id_from_token(authorization)
98    if grade_id not in grades:
99        raise HTTPException(status_code=404, detail="Grade not found")
100    del grades[grade_id]
101    return {"ok": True}
102 
103@app.get("/grades")
104def list_grades(authorization: str = Header(...)):
105    get_user_id_from_token(authorization)
106    return list(grades.values())

1fastapi
2uvicorn