Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import time
6 
7app = FastAPI()
8 
9# In-memory stores
10users = {}
11tokens = {}
12quizzes = {}
13questions = {}
14answers = {}
15scores = {}
16 
17# ID counters
18user_id_counter = 1
19quiz_id_counter = 1
20question_id_counter = 1
21answer_id_counter = 1
22score_id_counter = 1
23 
24# Models
25class UserCreate(BaseModel):
26    username: str
27    password: str
28 
29class UserLogin(BaseModel):
30    username: str
31    password: str
32 
33class QuizCreate(BaseModel):
34    title: str
35    teacher_id: int
36 
37class QuestionCreate(BaseModel):
38    quiz_id: int
39    question_type: str  # "multiple_choice" or "text"
40    question_text: str
41    correct_answer: str
42    options: Optional[list] = None
43 
44class AnswerSubmission(BaseModel):
45    student_id: int
46    quiz_id: int
47    answers: list  # list of {"question_id": int, "answer": str}
48 
49# Auth helper
50def get_user_from_token(authorization: str = Header(None)):
51    if not authorization:
52        raise HTTPException(status_code=401, detail="No token provided")
53    token = authorization.replace("Bearer ", "")
54    if token not in tokens:
55        raise HTTPException(status_code=401, detail="Invalid token")
56    return tokens[token]
57 
58# Endpoints
59@app.post("/signup")
60def signup(user: UserCreate):
61    global user_id_counter
62    for u in users.values():
63        if u["username"] == user.username:
64            raise HTTPException(status_code=400, detail="Username already exists")
65    uid = user_id_counter
66    user_id_counter += 1
67    users[uid] = {"id": uid, "username": user.username, "password": user.password, "role": "student"}
68    return {"user_id": uid, "message": "User created"}
69 
70@app.post("/login")
71def login(user: UserLogin):
72    for uid, u in users.items():
73        if u["username"] == user.username and u["password"] == user.password:
74            token = secrets.token_hex(16)
75            tokens[token] = uid
76            return {"token": token, "user_id": uid}
77    raise HTTPException(status_code=401, detail="Invalid credentials")
78 
79@app.get("/users/{user_id}")
80def get_user(user_id: int, authorization: str = Header(None)):
81    get_user_from_token(authorization)
82    if user_id not in users:
83        raise HTTPException(status_code=404, detail="User not found")
84    return users[user_id]
85 
86@app.post("/quizzes")
87def create_quiz(quiz: QuizCreate, authorization: str = Header(None)):
88    teacher_id = get_user_from_token(authorization)
89    global quiz_id_counter
90    qid = quiz_id_counter
91    quiz_id_counter += 1
92    quizzes[qid] = {"id": qid, "title": quiz.title, "teacher_id": teacher_id, "questions": []}
93    return {"quiz_id": qid}
94 
95@app.get("/quizzes/{quiz_id}")
96def get_quiz(quiz_id: int, authorization: str = Header(None)):
97    get_user_from_token(authorization)
98    if quiz_id not in quizzes:
99        raise HTTPException(status_code=404, detail="Quiz not found")
100    return quizzes[quiz_id]
101 
102@app.post("/questions")
103def create_question(question: QuestionCreate, authorization: str = Header(None)):
104    teacher_id = get_user_from_token(authorization)
105    if question.quiz_id not in quizzes:
106        raise HTTPException(status_code=404, detail="Quiz not found")
107    if quizzes[question.quiz_id]["teacher_id"] != teacher_id:
108        raise HTTPException(status_code=403, detail="Not your quiz")
109    global question_id_counter
110    qid = question_id_counter
111    question_id_counter += 1
112    questions[qid] = {
113        "id": qid,
114        "quiz_id": question.quiz_id,
115        "question_type": question.question_type,
116        "question_text": question.question_text,
117        "correct_answer": question.correct_answer,
118        "options": question.options
119    }
120    quizzes[question.quiz_id]["questions"].append(qid)
121    return {"question_id": qid}
122 
123@app.get("/questions/{question_id}")
124def get_question(question_id: int, authorization: str = Header(None)):
125    get_user_from_token(authorization)
126    if question_id not in questions:
127        raise HTTPException(status_code=404, detail="Question not found")
128    return questions[question_id]
129 
130@app.post("/answers")
131def submit_answers(submission: AnswerSubmission, authorization: str = Header(None)):
132    student_id = get_user_from_token(authorization)
133    if submission.quiz_id not in quizzes:
134        raise HTTPException(status_code=404, detail="Quiz not found")
135    if student_id != submission.student_id:
136        raise HTTPException(status_code=403, detail="Cannot submit for another student")
137    
138    quiz = quizzes[submission.quiz_id]
139    total_questions = len(quiz["questions"])
140    correct_count = 0
141    
142    answer_records = []
143    for ans in submission.answers:
144        qid = ans["question_id"]
145        if qid not in questions or questions[qid]["quiz_id"] != submission.quiz_id:
146            continue
147        question = questions[qid]
148        is_correct = ans["answer"].strip().lower() == question["correct_answer"].strip().lower()
149        if is_correct:
150            correct_count += 1
151        answer_records.append({
152            "question_id": qid,
153            "student_answer": ans["answer"],
154            "is_correct": is_correct
155        })
156    
157    global answer_id_counter
158    aid = answer_id_counter
159    answer_id_counter += 1
160    answers[aid] = {
161        "id": aid,
162        "student_id": student_id,
163        "quiz_id": submission.quiz_id,
164        "answers": answer_records
165    }
166    
167    score = (correct_count / total_questions * 100) if total_questions > 0 else 0
168    global score_id_counter
169    sid = score_id_counter
170    score_id_counter += 1
171    scores[sid] = {
172        "id": sid,
173        "student_id": student_id,
174        "quiz_id": submission.quiz_id,
175        "score": score,
176        "correct": correct_count,
177        "total": total_questions
178    }
179    
180    return {"answer_id": aid, "score_id": sid, "score": score}
181 
182@app.get("/answers/{answer_id}")
183def get_answer(answer_id: int, authorization: str = Header(None)):
184    get_user_from_token(authorization)
185    if answer_id not in answers:
186        raise HTTPException(status_code=404, detail="Answer not found")
187    return answers[answer_id]
188 
189@app.get("/scores/{score_id}")
190def get_score(score_id: int, authorization: str = Header(None)):
191    get_user_from_token(authorization)
192    if score_id not in scores:
193        raise HTTPException(status_code=404, detail="Score not found")
194    return scores[score_id]

1fastapi
2uvicorn