Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import random
5import string
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11matches = {}
12leaderboard = {}
13player_history = {}
14player_elo = {}
15next_user_id = 1
16next_match_id = 1
17next_token_id = 1
18 
19def generate_token():
20    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
21 
22def get_current_user(authorization: str = Header(...)):
23    token = authorization.replace("Bearer ", "")
24    if token not in tokens:
25        raise HTTPException(status_code=401, detail="Invalid token")
26    return tokens[token]
27 
28class SignupRequest(BaseModel):
29    username: str
30 
31class LoginRequest(BaseModel):
32    username: str
33 
34class MatchResultRequest(BaseModel):
35    winning_team: List[str]
36    losing_team: List[str]
37 
38@app.post("/signup")
39def signup(req: SignupRequest):
40    global next_user_id
41    for u in users.values():
42        if u["username"] == req.username:
43            raise HTTPException(status_code=400, detail="Username already exists")
44    user_id = next_user_id
45    next_user_id += 1
46    users[user_id] = {"id": user_id, "username": req.username}
47    player_elo[req.username] = 1000
48    player_history[req.username] = []
49    return {"id": user_id, "username": req.username}
50 
51@app.post("/login")
52def login(req: LoginRequest):
53    user = None
54    for u in users.values():
55        if u["username"] == req.username:
56            user = u
57            break
58    if not user:
59        raise HTTPException(status_code=404, detail="User not found")
60    token = generate_token()
61    tokens[token] = user["id"]
62    return {"token": token}
63 
64@app.get("/leaderboard")
65def get_leaderboard(authorization: str = Header(...)):
66    current_user_id = get_current_user(authorization)
67    players = []
68    for uid, u in users.items():
69        username = u["username"]
70        wins = 0
71        total = len(player_history[username])
72        for match in player_history[username]:
73            if match["result"] == "win":
74                wins += 1
75        winrate = (wins / total * 100) if total > 0 else 0.0
76        players.append({"username": username, "winrate": round(winrate, 2), "elo": player_elo[username]})
77    players.sort(key=lambda x: (-x["elo"], -x["winrate"]))
78    top100 = players[:100]
79    result = []
80    for i, p in enumerate(top100):
81        result.append({"rank": i+1, "username": p["username"], "winrate": p["winrate"], "elo": p["elo"]})
82    return result
83 
84@app.post("/match/result")
85def post_match_result(req: MatchResultRequest, authorization: str = Header(...)):
86    global next_match_id
87    current_user_id = get_current_user(authorization)
88    winning_team = req.winning_team
89    losing_team = req.losing_team
90    for player in winning_team + losing_team:
91        if player not in player_elo:
92            raise HTTPException(status_code=400, detail=f"Player {player} not found")
93    avg_winner_elo = sum(player_elo[p] for p in winning_team) / len(winning_team)
94    avg_loser_elo = sum(player_elo[p] for p in losing_team) / len(losing_team)
95    expected_winner = 1 / (1 + 10 ** ((avg_loser_elo - avg_winner_elo) / 400))
96    expected_loser = 1 / (1 + 10 ** ((avg_winner_elo - avg_loser_elo) / 400))
97    k = 32
98    elo_change_winner = round(k * (1 - expected_winner))
99    elo_change_loser = round(k * (0 - expected_loser))
100    match_id = next_match_id
101    next_match_id += 1
102    for player in winning_team:
103        player_elo[player] += elo_change_winner
104        player_history[player].append({
105            "match_id": match_id,
106            "result": "win",
107            "elo_change": elo_change_winner,
108            "opponents": losing_team
109        })
110    for player in losing_team:
111        player_elo[player] += elo_change_loser
112        player_history[player].append({
113            "match_id": match_id,
114            "result": "loss",
115            "elo_change": elo_change_loser,
116            "opponents": winning_team
117        })
118    matches[match_id] = {
119        "id": match_id,
120        "winning_team": winning_team,
121        "losing_team": losing_team,
122        "elo_change_winner": elo_change_winner,
123        "elo_change_loser": elo_change_loser
124    }
125    return {"match_id": match_id, "elo_change_winner": elo_change_winner, "elo_change_loser": elo_change_loser}
126 
127@app.get("/stats/{player}")
128def get_player_stats(player: str, authorization: str = Header(...)):
129    current_user_id = get_current_user(authorization)
130    if player not in player_history:
131        raise HTTPException(status_code=404, detail="Player not found")
132    return {"username": player, "history": player_history[player], "current_elo": player_elo[player]}
133 
134@app.get("/users/{user_id}")
135def get_user(user_id: int, authorization: str = Header(...)):
136    current_user_id = get_current_user(authorization)
137    if user_id not in users:
138        raise HTTPException(status_code=404, detail="User not found")
139    return users[user_id]
140 
141@app.post("/users")
142def create_user(req: SignupRequest):
143    return signup(req)

1fastapi
2uvicorn