Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import random
5import string
6from datetime import datetime
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12leaderboard = {}
13player_stats = {}
14player_history = {}
15next_user_id = 1
16next_game_id = 1
17games = {}
18 
19def generate_token():
20    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
21 
22def get_current_user(authorization: str = Header(None)):
23    if not authorization:
24        raise HTTPException(status_code=401, detail="Missing auth token")
25    token = authorization.replace("Bearer ", "")
26    user_id = tokens.get(token)
27    if not user_id:
28        raise HTTPException(status_code=401, detail="Invalid token")
29    return user_id
30 
31class SignupRequest(BaseModel):
32    username: str
33    password: str
34 
35class LoginRequest(BaseModel):
36    username: str
37    password: str
38 
39class GameResult(BaseModel):
40    score: int
41    total_questions: int
42 
43@app.post("/signup")
44def signup(req: SignupRequest):
45    global next_user_id
46    for u in users.values():
47        if u["username"] == req.username:
48            raise HTTPException(status_code=400, detail="Username taken")
49    user_id = next_user_id
50    next_user_id += 1
51    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
52    leaderboard[user_id] = {"score": 0, "games_played": 0, "wins": 0}
53    player_stats[user_id] = {"rank_history": []}
54    return {"id": user_id, "username": req.username}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    for uid, u in users.items():
59        if u["username"] == req.username and u["password"] == req.password:
60            token = generate_token()
61            tokens[token] = uid
62            return {"token": token}
63    raise HTTPException(status_code=401, detail="Invalid credentials")
64 
65@app.get("/leaderboard")
66def get_leaderboard():
67    sorted_users = sorted(leaderboard.items(), key=lambda x: (-x[1]["score"], x[1]["games_played"]))
68    top10 = []
69    for i, (uid, stats) in enumerate(sorted_users[:10]):
70        user = users[uid]
71        win_rate = (stats["wins"] / stats["games_played"] * 100) if stats["games_played"] > 0 else 0
72        top10.append({
73            "rank": i + 1,
74            "username": user["username"],
75            "score": stats["score"],
76            "games_played": stats["games_played"],
77            "win_rate": round(win_rate, 2)
78        })
79    return {"leaderboard": top10}
80 
81@app.get("/stats")
82def get_stats(authorization: str = Header(None)):
83    user_id = get_current_user(authorization)
84    user = users[user_id]
85    stats = leaderboard[user_id]
86    win_rate = (stats["wins"] / stats["games_played"] * 100) if stats["games_played"] > 0 else 0
87    return {
88        "username": user["username"],
89        "score": stats["score"],
90        "games_played": stats["games_played"],
91        "wins": stats["wins"],
92        "win_rate": round(win_rate, 2),
93        "rank_history": player_stats[user_id]["rank_history"]
94    }
95 
96@app.post("/game")
97def submit_game_result(result: GameResult, authorization: str = Header(None)):
98    global next_game_id
99    user_id = get_current_user(authorization)
100    game_id = next_game_id
101    next_game_id += 1
102    
103    games[game_id] = {
104        "id": game_id,
105        "user_id": user_id,
106        "score": result.score,
107        "total_questions": result.total_questions,
108        "timestamp": datetime.now().isoformat()
109    }
110    
111    stats = leaderboard[user_id]
112    stats["games_played"] += 1
113    stats["score"] += result.score
114    
115    if result.score >= result.total_questions * 0.7:
116        stats["wins"] += 1
117    
118    # Update rank history
119    sorted_users = sorted(leaderboard.items(), key=lambda x: (-x[1]["score"], x[1]["games_played"]))
120    current_rank = next((i+1 for i, (uid, _) in enumerate(sorted_users) if uid == user_id), 0)
121    player_stats[user_id]["rank_history"].append({
122        "rank": current_rank,
123        "timestamp": datetime.now().isoformat()
124    })
125    
126    return {"game_id": game_id, "status": "submitted"}
127 
128@app.get("/game/{game_id}")
129def get_game(game_id: int):
130    if game_id not in games:
131        raise HTTPException(status_code=404, detail="Game not found")
132    return games[game_id]
133 
134@app.get("/user/{user_id}")
135def get_user(user_id: int):
136    if user_id not in users:
137        raise HTTPException(status_code=404, detail="User not found")
138    user = users[user_id]
139    stats = leaderboard[user_id]
140    win_rate = (stats["wins"] / stats["games_played"] * 100) if stats["games_played"] > 0 else 0
141    return {
142        "id": user["id"],
143        "username": user["username"],
144        "score": stats["score"],
145        "games_played": stats["games_played"],
146        "win_rate": round(win_rate, 2)
147    }
148 
149@app.post("/user")
150def create_user(req: SignupRequest):
151    return signup(req)

1fastapi
2uvicorn
3pydantic