Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import time
5import secrets
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11leaderboard = {}
12matches = {}
13user_counter = 1
14match_counter = 1
15 
16class SignupRequest(BaseModel):
17    username: str
18 
19class LoginRequest(BaseModel):
20    username: str
21 
22class ScoreUpdate(BaseModel):
23    score: int
24 
25class MatchResult(BaseModel):
26    opponent: str
27    result: str
28    score: int
29 
30def get_current_user(authorization: Optional[str] = Header(None)):
31    if not authorization:
32        raise HTTPException(status_code=401, detail="Missing auth header")
33    token = authorization.replace("Bearer ", "")
34    if token not in tokens:
35        raise HTTPException(status_code=401, detail="Invalid token")
36    return tokens[token]
37 
38@app.post("/signup")
39def signup(req: SignupRequest):
40    global user_counter
41    if req.username in [u["username"] for u in users.values()]:
42        raise HTTPException(status_code=400, detail="Username taken")
43    user_id = user_counter
44    user_counter += 1
45    users[user_id] = {"id": user_id, "username": req.username, "score": 0}
46    leaderboard[user_id] = {"username": req.username, "score": 0, "matches": []}
47    return {"id": user_id, "username": req.username}
48 
49@app.post("/login")
50def login(req: LoginRequest):
51    for uid, u in users.items():
52        if u["username"] == req.username:
53            token = secrets.token_hex(16)
54            tokens[token] = uid
55            return {"token": token}
56    raise HTTPException(status_code=404, detail="User not found")
57 
58@app.get("/leaderboard")
59def get_leaderboard(authorization: Optional[str] = Header(None)):
60    current_user_id = get_current_user(authorization)
61    sorted_players = sorted(leaderboard.values(), key=lambda x: x["score"], reverse=True)
62    result = []
63    for p in sorted_players:
64        recent = p["matches"][-5:] if p["matches"] else []
65        result.append({
66            "username": p["username"],
67            "score": p["score"],
68            "recent_matches": recent
69        })
70    return result
71 
72@app.post("/leaderboard/score")
73def update_score(req: ScoreUpdate, authorization: Optional[str] = Header(None)):
74    user_id = get_current_user(authorization)
75    leaderboard[user_id]["score"] = req.score
76    users[user_id]["score"] = req.score
77    return {"status": "ok"}
78 
79@app.post("/leaderboard/match")
80def add_match(req: MatchResult, authorization: Optional[str] = Header(None)):
81    global match_counter
82    user_id = get_current_user(authorization)
83    match_id = match_counter
84    match_counter += 1
85    match_entry = {
86        "id": match_id,
87        "opponent": req.opponent,
88        "result": req.result,
89        "score": req.score,
90        "timestamp": time.time()
91    }
92    matches[match_id] = match_entry
93    leaderboard[user_id]["matches"].append(match_entry)
94    return match_entry
95 
96@app.get("/leaderboard/{user_id}")
97def get_player(user_id: int, authorization: Optional[str] = Header(None)):
98    current_user_id = get_current_user(authorization)
99    if user_id not in leaderboard:
100        raise HTTPException(status_code=404, detail="Player not found")
101    p = leaderboard[user_id]
102    return {
103        "username": p["username"],
104        "score": p["score"],
105        "recent_matches": p["matches"][-5:]
106    }

1fastapi
2uvicorn