Browse the data

1from fastapi import FastAPI, HTTPException, Header
2import secrets
3import time
4 
5app = FastAPI()
6 
7users = {}
8classes = {}
9questions = {}
10scores = {}
11tokens = {}
12 
13user_id_counter = 1
14class_id_counter = 1
15question_id_counter = 1
16score_id_counter = 1
17 
18def get_current_user(authorization: str = Header(None)):
19    if not authorization:
20        raise HTTPException(401, "No auth token")
21    token = authorization.replace("Bearer ", "")
22    if token not in tokens:
23        raise HTTPException(401, "Invalid token")
24    return tokens[token]
25 
26@app.post("/signup")
27def signup(username: str, password: str):
28    global user_id_counter
29    for u in users.values():
30        if u["username"] == username:
31            raise HTTPException(400, "Username exists")
32    uid = user_id_counter
33    user_id_counter += 1
34    users[uid] = {"id": uid, "username": username, "password": password, "role": "student"}
35    return {"user_id": uid, "username": username}
36 
37@app.post("/login")
38def login(username: str, password: str):
39    for u in users.values():
40        if u["username"] == username and u["password"] == password:
41            token = secrets.token_hex(16)
42            tokens[token] = u["id"]
43            return {"token": token, "user_id": u["id"]}
44    raise HTTPException(401, "Invalid credentials")
45 
46@app.post("/class")
47def create_class(name: str, authorization: str = Header(None)):
48    global class_id_counter
49    user_id = get_current_user(authorization)
50    if users[user_id]["role"] != "student":
51        raise HTTPException(403, "Only students can create classes initially")
52    cid = class_id_counter
53    class_id_counter += 1
54    classes[cid] = {"id": cid, "name": name, "admin": user_id, "quiz_masters": []}
55    users[user_id]["role"] = "admin"
56    return {"class_id": cid, "name": name, "admin": user_id}
57 
58@app.post("/class/{class_id}/promote")
59def promote_to_quiz_master(class_id: int, user_id: int, authorization: str = Header(None)):
60    current_user = get_current_user(authorization)
61    if class_id not in classes:
62        raise HTTPException(404, "Class not found")
63    if classes[class_id]["admin"] != current_user:
64        raise HTTPException(403, "Only admin can promote")
65    if user_id not in users:
66        raise HTTPException(404, "User not found")
67    if user_id not in classes[class_id]["quiz_masters"]:
68        classes[class_id]["quiz_masters"].append(user_id)
69    return {"class_id": class_id, "quiz_masters": classes[class_id]["quiz_masters"]}
70 
71@app.get("/class/{class_id}")
72def get_class(class_id: int):
73    if class_id not in classes:
74        raise HTTPException(404, "Class not found")
75    return classes[class_id]
76 
77@app.get("/user/{user_id}")
78def get_user(user_id: int):
79    if user_id not in users:
80        raise HTTPException(404, "User not found")
81    return {"id": users[user_id]["id"], "username": users[user_id]["username"], "role": users[user_id]["role"]}
82 
83@app.post("/question")
84def create_question(question_text: str, answer: str, time_limit: int, class_id: int, authorization: str = Header(None)):
85    global question_id_counter
86    user_id = get_current_user(authorization)
87    if class_id not in classes:
88        raise HTTPException(404, "Class not found")
89    if user_id != classes[class_id]["admin"] and user_id not in classes[class_id]["quiz_masters"]:
90        raise HTTPException(403, "Only admin or quiz master can create questions")
91    qid = question_id_counter
92    question_id_counter += 1
93    questions[qid] = {
94        "id": qid,
95        "question_text": question_text,
96        "answer": answer,
97        "time_limit": time_limit,
98        "class_id": class_id,
99        "created_by": user_id
100    }
101    return {"question_id": qid}
102 
103@app.get("/question/{question_id}")
104def get_question(question_id: int):
105    if question_id not in questions:
106        raise HTTPException(404, "Question not found")
107    return questions[question_id]
108 
109@app.post("/score")
110def submit_score(user_id: int, question_id: int, score_value: int, authorization: str = Header(None)):
111    global score_id_counter
112    current_user = get_current_user(authorization)
113    if current_user != user_id:
114        raise HTTPException(403, "Can only submit own score")
115    if question_id not in questions:
116        raise HTTPException(404, "Question not found")
117    if user_id not in users:
118        raise HTTPException(404, "User not found")
119    sid = score_id_counter
120    score_id_counter += 1
121    scores[sid] = {
122        "id": sid,
123        "user_id": user_id,
124        "question_id": question_id,
125        "score": score_value,
126        "timestamp": time.time()
127    }
128    return {"score_id": sid}
129 
130@app.get("/score/{score_id}")
131def get_score(score_id: int):
132    if score_id not in scores:
133        raise HTTPException(404, "Score not found")
134    return scores[score_id]

1fastapi
2uvicorn