Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import hashlib
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11questions = {}
12next_user_id = 1
13next_question_id = 1
14 
15def get_user_from_token(authorization: str = Header(None)):
16    if not authorization:
17        raise HTTPException(status_code=401, detail="No auth token")
18    token = authorization.replace("Bearer ", "")
19    if token not in tokens:
20        raise HTTPException(status_code=401, detail="Invalid token")
21    return tokens[token]
22 
23class SignupRequest(BaseModel):
24    username: str
25    password: str
26 
27class LoginRequest(BaseModel):
28    username: str
29    password: str
30 
31class QuestionCreate(BaseModel):
32    question: str
33    answer: str
34    difficulty: Optional[str] = "medium"
35    category: Optional[str] = "general"
36 
37class QuestionUpdate(BaseModel):
38    question: Optional[str] = None
39    answer: Optional[str] = None
40    difficulty: Optional[str] = None
41    category: Optional[str] = None
42 
43@app.post("/signup")
44def signup(req: SignupRequest):
45    global next_user_id
46    if any(u["username"] == req.username for u in users.values()):
47        raise HTTPException(status_code=400, detail="Username taken")
48    user_id = next_user_id
49    next_user_id += 1
50    users[user_id] = {
51        "id": user_id,
52        "username": req.username,
53        "password": hashlib.sha256(req.password.encode()).hexdigest()
54    }
55    token = secrets.token_hex(16)
56    tokens[token] = user_id
57    return {"user_id": user_id, "token": token}
58 
59@app.post("/login")
60def login(req: LoginRequest):
61    for uid, u in users.items():
62        if u["username"] == req.username and u["password"] == hashlib.sha256(req.password.encode()).hexdigest():
63            token = secrets.token_hex(16)
64            tokens[token] = uid
65            return {"token": token}
66    raise HTTPException(status_code=401, detail="Invalid credentials")
67 
68@app.get("/questions/{question_id}")
69def get_question(question_id: int, authorization: str = Header(None)):
70    get_user_from_token(authorization)
71    if question_id not in questions:
72        raise HTTPException(status_code=404, detail="Question not found")
73    return questions[question_id]
74 
75@app.post("/questions")
76def create_question(req: QuestionCreate, authorization: str = Header(None)):
77    global next_question_id
78    get_user_from_token(authorization)
79    qid = next_question_id
80    next_question_id += 1
81    questions[qid] = {
82        "id": qid,
83        "question": req.question,
84        "answer": req.answer,
85        "difficulty": req.difficulty,
86        "category": req.category
87    }
88    return questions[qid]
89 
90@app.patch("/questions/{question_id}")
91def update_question(question_id: int, req: QuestionUpdate, authorization: str = Header(None)):
92    get_user_from_token(authorization)
93    if question_id not in questions:
94        raise HTTPException(status_code=404, detail="Question not found")
95    q = questions[question_id]
96    if req.question is not None:
97        q["question"] = req.question
98    if req.answer is not None:
99        q["answer"] = req.answer
100    if req.difficulty is not None:
101        q["difficulty"] = req.difficulty
102    if req.category is not None:
103        q["category"] = req.category
104    return q

1fastapi
2uvicorn