Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import uuid
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11cards = {}
12progress = {}
13next_user_id = 1
14next_card_id = 1
15 
16class SignupRequest(BaseModel):
17    username: str
18    password: str
19 
20class LoginRequest(BaseModel):
21    username: str
22    password: str
23 
24class CardCreate(BaseModel):
25    word_lang1: str
26    word_lang2: str
27 
28class ProgressUpdate(BaseModel):
29    status: str  # "learned" or "review"
30 
31def get_user_id_from_token(authorization: str = Header(None)):
32    if not authorization:
33        raise HTTPException(status_code=401, detail="Missing auth header")
34    token = authorization.replace("Bearer ", "")
35    if token not in tokens:
36        raise HTTPException(status_code=401, detail="Invalid token")
37    return tokens[token]
38 
39@app.post("/signup")
40def signup(req: SignupRequest):
41    global next_user_id
42    for u in users.values():
43        if u["username"] == req.username:
44            raise HTTPException(status_code=400, detail="User exists")
45    user_id = next_user_id
46    next_user_id += 1
47    users[user_id] = {"username": req.username, "password": req.password}
48    return {"user_id": user_id, "message": "User created"}
49 
50@app.post("/login")
51def login(req: LoginRequest):
52    for uid, u in users.items():
53        if u["username"] == req.username and u["password"] == req.password:
54            token = secrets.token_hex(16)
55            tokens[token] = uid
56            return {"token": token}
57    raise HTTPException(status_code=401, detail="Invalid credentials")
58 
59@app.post("/cards")
60def create_card(card: CardCreate, authorization: str = Header(None)):
61    user_id = get_user_id_from_token(authorization)
62    global next_card_id
63    card_id = next_card_id
64    next_card_id += 1
65    cards[card_id] = {"id": card_id, "word_lang1": card.word_lang1, "word_lang2": card.word_lang2}
66    progress[card_id] = {"user_id": user_id, "status": "review", "history": []}
67    return cards[card_id]
68 
69@app.get("/cards/{card_id}")
70def get_card(card_id: int, authorization: str = Header(None)):
71    user_id = get_user_id_from_token(authorization)
72    if card_id not in cards:
73        raise HTTPException(status_code=404, detail="Card not found")
74    return cards[card_id]
75 
76@app.post("/cards/{card_id}/progress")
77def update_progress(card_id: int, update: ProgressUpdate, authorization: str = Header(None)):
78    user_id = get_user_id_from_token(authorization)
79    if card_id not in cards:
80        raise HTTPException(status_code=404, detail="Card not found")
81    if update.status not in ("learned", "review"):
82        raise HTTPException(status_code=400, detail="Status must be 'learned' or 'review'")
83    if card_id not in progress:
84        progress[card_id] = {"user_id": user_id, "status": "review", "history": []}
85    progress[card_id]["status"] = update.status
86    progress[card_id]["history"].append({"status": update.status, "timestamp": None})
87    return {"card_id": card_id, "status": update.status}
88 
89@app.get("/cards/{card_id}/progress")
90def get_progress(card_id: int, authorization: str = Header(None)):
91    user_id = get_user_id_from_token(authorization)
92    if card_id not in cards:
93        raise HTTPException(status_code=404, detail="Card not found")
94    if card_id not in progress:
95        return {"card_id": card_id, "status": "review", "history": []}
96    return progress[card_id]

1fastapi
2uvicorn