Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import time
5import random
6import string
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12scores = {}
13next_user_id = 1
14next_score_id = 1
15 
16def generate_token():
17    return ''.join(random.choices(string.ascii_lowercase + string.digits, k=32))
18 
19def get_current_user(authorization: str = Header(None)):
20    if not authorization:
21        raise HTTPException(status_code=401, detail="Missing auth header")
22    token = authorization.replace("Bearer ", "")
23    if token not in tokens:
24        raise HTTPException(status_code=401, detail="Invalid token")
25    return tokens[token]
26 
27class SignupRequest(BaseModel):
28    username: str
29    password: str
30 
31class LoginRequest(BaseModel):
32    username: str
33    password: str
34 
35class ScoreCreate(BaseModel):
36    player_name: str
37    score: int
38    region: Optional[str] = None
39    game_mode: Optional[str] = None
40 
41@app.post("/signup")
42def signup(req: SignupRequest):
43    global next_user_id
44    for uid, u in users.items():
45        if u["username"] == req.username:
46            raise HTTPException(status_code=400, detail="Username taken")
47    uid = next_user_id
48    next_user_id += 1
49    users[uid] = {"id": uid, "username": req.username, "password": req.password}
50    token = generate_token()
51    tokens[token] = uid
52    return {"user_id": uid, "token": token}
53 
54@app.post("/login")
55def login(req: LoginRequest):
56    for uid, u in users.items():
57        if u["username"] == req.username and u["password"] == req.password:
58            token = generate_token()
59            tokens[token] = uid
60            return {"token": token}
61    raise HTTPException(status_code=401, detail="Invalid credentials")
62 
63@app.get("/players")
64def get_players(name: str, region: Optional[str] = None, game_mode: Optional[str] = None, authorization: str = Header(None)):
65    get_current_user(authorization)
66    results = []
67    for sid, s in scores.items():
68        if s["player_name"] != name:
69            continue
70        if region and s.get("region") != region:
71            continue
72        if game_mode and s.get("game_mode") != game_mode:
73            continue
74        results.append(s)
75    if not results:
76        raise HTTPException(status_code=404, detail="Player not found")
77    sorted_scores = sorted(results, key=lambda x: x["score"], reverse=True)
78    best = sorted_scores[0]
79    rank = 1
80    for i, s in enumerate(sorted_scores):
81        if s["score"] > best["score"]:
82            rank = i + 1
83    return {"rank": rank, "score": best["score"], "join_date": best["join_date"]}
84 
85@app.get("/scores/{score_id}")
86def get_score(score_id: int, authorization: str = Header(None)):
87    get_current_user(authorization)
88    if score_id not in scores:
89        raise HTTPException(status_code=404, detail="Score not found")
90    return scores[score_id]
91 
92@app.post("/scores")
93def create_score(score: ScoreCreate, authorization: str = Header(None)):
94    global next_score_id
95    get_current_user(authorization)
96    sid = next_score_id
97    next_score_id += 1
98    scores[sid] = {
99        "id": sid,
100        "player_name": score.player_name,
101        "score": score.score,
102        "region": score.region,
103        "game_mode": score.game_mode,
104        "join_date": int(time.time())
105    }
106    return scores[sid]

1fastapi
2uvicorn