Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import uuid
5 
6app = FastAPI()
7 
8users = {}
9decks = {}
10tokens = {}
11next_user_id = 1
12next_deck_id = 1
13 
14class SignupRequest(BaseModel):
15    username: str
16    password: str
17 
18class LoginRequest(BaseModel):
19    username: str
20    password: str
21 
22class CreateDeckRequest(BaseModel):
23    name: str
24    description: Optional[str] = ""
25    is_public: Optional[bool] = False
26    difficulty: Optional[str] = "easy"
27 
28class UpdateDeckRequest(BaseModel):
29    name: Optional[str] = None
30    description: Optional[str] = None
31    is_public: Optional[bool] = None
32    difficulty: Optional[str] = None
33 
34def get_user_from_token(authorization: str = Header(...)):
35    token = authorization.replace("Bearer ", "")
36    user_id = tokens.get(token)
37    if not user_id:
38        raise HTTPException(status_code=401, detail="Invalid token")
39    return user_id
40 
41@app.post("/signup")
42def signup(req: SignupRequest):
43    global next_user_id
44    for u in users.values():
45        if u["username"] == req.username:
46            raise HTTPException(status_code=400, detail="Username taken")
47    user_id = next_user_id
48    next_user_id += 1
49    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
50    token = str(uuid.uuid4())
51    tokens[token] = user_id
52    return {"user_id": user_id, "token": token}
53 
54@app.post("/login")
55def login(req: LoginRequest):
56    for u in users.values():
57        if u["username"] == req.username and u["password"] == req.password:
58            token = str(uuid.uuid4())
59            tokens[token] = u["id"]
60            return {"user_id": u["id"], "token": token}
61    raise HTTPException(status_code=401, detail="Invalid credentials")
62 
63@app.get("/decks/{deck_id}")
64def get_deck(deck_id: int, authorization: str = Header(...)):
65    get_user_from_token(authorization)
66    deck = decks.get(deck_id)
67    if not deck:
68        raise HTTPException(status_code=404, detail="Deck not found")
69    return deck
70 
71@app.post("/decks")
72def create_deck(req: CreateDeckRequest, authorization: str = Header(...)):
73    global next_deck_id
74    user_id = get_user_from_token(authorization)
75    deck_id = next_deck_id
76    next_deck_id += 1
77    decks[deck_id] = {
78        "id": deck_id,
79        "name": req.name,
80        "description": req.description,
81        "is_public": req.is_public,
82        "difficulty": req.difficulty,
83        "owner_id": user_id
84    }
85    return decks[deck_id]
86 
87@app.patch("/decks/{deck_id}")
88def update_deck(deck_id: int, req: UpdateDeckRequest, authorization: str = Header(...)):
89    user_id = get_user_from_token(authorization)
90    deck = decks.get(deck_id)
91    if not deck:
92        raise HTTPException(status_code=404, detail="Deck not found")
93    if deck["owner_id"] != user_id:
94        raise HTTPException(status_code=403, detail="Not your deck")
95    if req.name is not None:
96        deck["name"] = req.name
97    if req.description is not None:
98        deck["description"] = req.description
99    if req.is_public is not None:
100        deck["is_public"] = req.is_public
101    if req.difficulty is not None:
102        deck["difficulty"] = req.difficulty
103    return deck

1fastapi
2uvicorn