Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import uvicorn
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11assignments = {}
12submissions = {}
13next_user_id = 1
14next_assign_id = 1
15next_sub_id = 1
16 
17class SignupRequest(BaseModel):
18    username: str
19    password: str
20 
21class LoginRequest(BaseModel):
22    username: str
23    password: str
24 
25class AssignmentCreate(BaseModel):
26    title: str
27    max_score: float
28 
29class SubmissionCreate(BaseModel):
30    assignment_id: int
31    student_name: str
32    score: float
33    feedback: Optional[str] = ""
34 
35def get_current_user(authorization: Optional[str] = Header(None)):
36    if not authorization:
37        raise HTTPException(status_code=401, detail="Missing auth header")
38    token = authorization.replace("Bearer ", "")
39    if token not in tokens:
40        raise HTTPException(status_code=401, detail="Invalid token")
41    return tokens[token]
42 
43@app.post("/signup")
44def signup(req: SignupRequest):
45    global next_user_id
46    for u in users.values():
47        if u["username"] == req.username:
48            raise HTTPException(status_code=400, detail="Username already exists")
49    user_id = next_user_id
50    next_user_id += 1
51    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
52    token = secrets.token_hex(16)
53    tokens[token] = user_id
54    return {"user_id": user_id, "token": token}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    for u in users.values():
59        if u["username"] == req.username and u["password"] == req.password:
60            token = secrets.token_hex(16)
61            tokens[token] = u["id"]
62            return {"token": token}
63    raise HTTPException(status_code=401, detail="Invalid credentials")
64 
65@app.post("/assignments")
66def create_assignment(assignment: AssignmentCreate, authorization: Optional[str] = Header(None)):
67    user_id = get_current_user(authorization)
68    global next_assign_id
69    assign_id = next_assign_id
70    next_assign_id += 1
71    assignments[assign_id] = {
72        "id": assign_id,
73        "title": assignment.title,
74        "max_score": assignment.max_score,
75        "created_by": user_id
76    }
77    return assignments[assign_id]
78 
79@app.get("/assignments/{assignment_id}")
80def get_assignment(assignment_id: int, authorization: Optional[str] = Header(None)):
81    get_current_user(authorization)
82    if assignment_id not in assignments:
83        raise HTTPException(status_code=404, detail="Assignment not found")
84    return assignments[assignment_id]
85 
86@app.post("/submissions")
87def create_submission(submission: SubmissionCreate, authorization: Optional[str] = Header(None)):
88    user_id = get_current_user(authorization)
89    if submission.assignment_id not in assignments:
90        raise HTTPException(status_code=404, detail="Assignment not found")
91    assign = assignments[submission.assignment_id]
92    if submission.score > assign["max_score"]:
93        raise HTTPException(status_code=400, detail="Score exceeds max score")
94    global next_sub_id
95    sub_id = next_sub_id
96    next_sub_id += 1
97    submissions[sub_id] = {
98        "id": sub_id,
99        "assignment_id": submission.assignment_id,
100        "student_name": submission.student_name,
101        "score": submission.score,
102        "feedback": submission.feedback,
103        "graded_by": user_id
104    }
105    return submissions[sub_id]
106 
107@app.get("/submissions/{submission_id}")
108def get_submission(submission_id: int, authorization: Optional[str] = Header(None)):
109    get_current_user(authorization)
110    if submission_id not in submissions:
111        raise HTTPException(status_code=404, detail="Submission not found")
112    return submissions[submission_id]

1fastapi
2uvicorn