Browse the data

1import random
2import string
3from fastapi import FastAPI, HTTPException, Header
4from pydantic import BaseModel
5from typing import Optional, List, Dict
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11students = {}
12quizzes = {}
13quiz_questions = {}
14completions = {}
15 
16class SignupRequest(BaseModel):
17    username: str
18    password: str
19 
20class LoginRequest(BaseModel):
21    username: str
22    password: str
23 
24class QuizCreate(BaseModel):
25    course_id: int
26    title: str
27 
28class QuestionCreate(BaseModel):
29    question_type: str
30    prompt: str
31    options: Optional[List[str]] = None
32    correct_answer: str
33 
34class StudentAnswer(BaseModel):
35    answer: str
36 
37def generate_token():
38    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
39 
40def get_current_user(authorization: str = Header(None)):
41    if not authorization:
42        raise HTTPException(status_code=401, detail="Missing auth token")
43    token = authorization.replace("Bearer ", "")
44    if token not in tokens:
45        raise HTTPException(status_code=401, detail="Invalid token")
46    return tokens[token]
47 
48@app.post("/signup")
49def signup(req: SignupRequest):
50    if req.username in users:
51        raise HTTPException(status_code=400, detail="User already exists")
52    user_id = len(users) + 1
53    users[req.username] = {"id": user_id, "username": req.username, "password": req.password}
54    students[user_id] = {"id": user_id, "username": req.username}
55    return {"id": user_id, "username": req.username}
56 
57@app.post("/login")
58def login(req: LoginRequest):
59    if req.username not in users or users[req.username]["password"] != req.password:
60        raise HTTPException(status_code=401, detail="Invalid credentials")
61    token = generate_token()
62    tokens[token] = users[req.username]["id"]
63    return {"token": token}
64 
65@app.post("/quiz")
66def create_quiz(quiz: QuizCreate, authorization: str = Header(None)):
67    user_id = get_current_user(authorization)
68    quiz_id = len(quizzes) + 1
69    quizzes[quiz_id] = {"id": quiz_id, "course_id": quiz.course_id, "title": quiz.title, "creator_id": user_id}
70    return quizzes[quiz_id]
71 
72@app.get("/quiz/{quiz_id}")
73def get_quiz(quiz_id: int, authorization: str = Header(None)):
74    get_current_user(authorization)
75    if quiz_id not in quizzes:
76        raise HTTPException(status_code=404, detail="Quiz not found")
77    return quizzes[quiz_id]
78 
79@app.post("/quiz/{quiz_id}/question")
80def add_question(quiz_id: int, question: QuestionCreate, authorization: str = Header(None)):
81    get_current_user(authorization)
82    if quiz_id not in quizzes:
83        raise HTTPException(status_code=404, detail="Quiz not found")
84    if quiz_id not in quiz_questions:
85        quiz_questions[quiz_id] = []
86    q_id = len(quiz_questions[quiz_id]) + 1
87    q_data = {"id": q_id, "quiz_id": quiz_id, "question_type": question.question_type, "prompt": question.prompt, "options": question.options, "correct_answer": question.correct_answer}
88    quiz_questions[quiz_id].append(q_data)
89    return q_data
90 
91@app.get("/quiz/{course_id}/{question_num}.xml")
92def get_quiz_question(course_id: int, question_num: int, authorization: str = Header(None)):
93    user_id = get_current_user(authorization)
94    for qid, quiz in quizzes.items():
95        if quiz["course_id"] == course_id:
96            if qid not in quiz_questions:
97                raise HTTPException(status_code=404, detail="No questions for this quiz")
98            for q in quiz_questions[qid]:
99                if q["id"] == question_num:
100                    return {
101                        "question": {
102                            "type": q["question_type"],
103                            "prompt": q["prompt"],
104                            "options": q.get("options"),
105                            "id": q["id"]
106                        }
107                    }
108    raise HTTPException(status_code=404, detail="Question not found")
109 
110@app.post("/quiz/{course_id}/{question_num}/answer")
111def answer_question(course_id: int, question_num: int, answer: StudentAnswer, authorization: str = Header(None)):
112    user_id = get_current_user(authorization)
113    for qid, quiz in quizzes.items():
114        if quiz["course_id"] == course_id:
115            if qid not in quiz_questions:
116                raise HTTPException(status_code=404, detail="No questions for this quiz")
117            for q in quiz_questions[qid]:
118                if q["id"] == question_num:
119                    correct = q["correct_answer"] == answer.answer
120                    if user_id not in completions:
121                        completions[user_id] = {}
122                    if course_id not in completions[user_id]:
123                        completions[user_id][course_id] = []
124                    completions[user_id][course_id].append({
125                        "question_id": question_num,
126                        "answer": answer.answer,
127                        "correct": correct
128                    })
129                    return {"correct": correct}
130    raise HTTPException(status_code=404, detail="Question not found")
131 
132@app.get("/completion/{user_id}/{course_id}")
133def get_completion(user_id: int, course_id: int, authorization: str = Header(None)):
134    get_current_user(authorization)
135    if user_id not in completions or course_id not in completions[user_id]:
136        return {"answers": []}
137    return {"answers": completions[user_id][course_id]}

1fastapi
2uvicorn