Browse the data

1import random
2from fastapi import FastAPI, HTTPException, Header
3from pydantic import BaseModel
4from typing import Optional, List
5 
6app = FastAPI()
7 
8users = {}
9next_user_id = 1
10tokens = {}
11matches = {}
12next_match_id = 1
13leaderboard = {}
14player_history = {}
15 
16def generate_token():
17    return ''.join(random.choices('abcdefghijklmnopqrstuvwxyz0123456789', k=32))
18 
19def get_user_from_token(authorization: str = Header(None)):
20    if not authorization or not authorization.startswith('Bearer '):
21        raise HTTPException(status_code=401, detail="Invalid auth header")
22    token = authorization[7:]
23    user_id = tokens.get(token)
24    if not user_id:
25        raise HTTPException(status_code=401, detail="Invalid token")
26    return user_id
27 
28class SignupRequest(BaseModel):
29    username: str
30 
31class LoginRequest(BaseModel):
32    username: str
33 
34class MatchResultRequest(BaseModel):
35    team1: List[str]
36    team2: List[str]
37    winner: str
38 
39class UserCreate(BaseModel):
40    username: str
41 
42@app.post("/signup")
43def signup(req: SignupRequest):
44    global next_user_id
45    for u in users.values():
46        if u['username'] == req.username:
47            raise HTTPException(status_code=400, detail="Username already exists")
48    user_id = next_user_id
49    next_user_id += 1
50    users[user_id] = {
51        'id': user_id,
52        'username': req.username,
53        'elo': 1000,
54        'wins': 0,
55        'losses': 0
56    }
57    token = generate_token()
58    tokens[token] = user_id
59    return {'token': token, 'user_id': user_id}
60 
61@app.post("/login")
62def login(req: LoginRequest):
63    for uid, u in users.items():
64        if u['username'] == req.username:
65            token = generate_token()
66            tokens[token] = uid
67            return {'token': token, 'user_id': uid}
68    raise HTTPException(status_code=404, detail="User not found")
69 
70@app.get("/leaderboard")
71def get_leaderboard(authorization: str = Header(None)):
72    get_user_from_token(authorization)
73    sorted_users = sorted(users.values(), key=lambda u: u['elo'], reverse=True)[:100]
74    result = []
75    for rank, u in enumerate(sorted_users, 1):
76        total = u['wins'] + u['losses']
77        winrate = u['wins'] / total if total > 0 else 0.0
78        result.append({
79            'rank': rank,
80            'username': u['username'],
81            'winrate': round(winrate, 3)
82        })
83    return result
84 
85@app.post("/match/result")
86def post_match_result(req: MatchResultRequest, authorization: str = Header(None)):
87    get_user_from_token(authorization)
88    global next_match_id
89 
90    all_players = req.team1 + req.team2
91    player_ids = []
92    for p in all_players:
93        found = None
94        for uid, u in users.items():
95            if u['username'] == p:
96                found = uid
97                break
98        if not found:
99            raise HTTPException(status_code=404, detail=f"Player {p} not found")
100        player_ids.append(found)
101 
102    team1_ids = player_ids[:len(req.team1)]
103    team2_ids = player_ids[len(req.team1):]
104 
105    winner_team = req.winner
106    if winner_team not in ['team1', 'team2']:
107        raise HTTPException(status_code=400, detail="winner must be team1 or team2")
108 
109    k_factor = 32
110    avg_elo1 = sum(users[uid]['elo'] for uid in team1_ids) / len(team1_ids)
111    avg_elo2 = sum(users[uid]['elo'] for uid in team2_ids) / len(team2_ids)
112    expected1 = 1 / (1 + 10 ** ((avg_elo2 - avg_elo1) / 400))
113    expected2 = 1 - expected1
114 
115    if winner_team == 'team1':
116        score1, score2 = 1, 0
117    else:
118        score1, score2 = 0, 1
119 
120    for uid in team1_ids:
121        users[uid]['elo'] += int(k_factor * (score1 - expected1))
122        if score1 == 1:
123            users[uid]['wins'] += 1
124        else:
125            users[uid]['losses'] += 1
126    for uid in team2_ids:
127        users[uid]['elo'] += int(k_factor * (score2 - expected2))
128        if score2 == 1:
129            users[uid]['wins'] += 1
130        else:
131            users[uid]['losses'] += 1
132 
133    match_id = next_match_id
134    next_match_id += 1
135    matches[match_id] = {
136        'id': match_id,
137        'team1': req.team1,
138        'team2': req.team2,
139        'winner': winner_team
140    }
141 
142    for p in all_players:
143        if p not in player_history:
144            player_history[p] = []
145        player_history[p].append(match_id)
146 
147    return {'match_id': match_id}
148 
149@app.get("/stats/{player}")
150def get_player_stats(player: str, authorization: str = Header(None)):
151    get_user_from_token(authorization)
152    found_user = None
153    for uid, u in users.items():
154        if u['username'] == player:
155            found_user = u
156            break
157    if not found_user:
158        raise HTTPException(status_code=404, detail="Player not found")
159 
160    history = player_history.get(player, [])
161    match_details = []
162    for mid in history:
163        m = matches[mid]
164        match_details.append({
165            'match_id': mid,
166            'team1': m['team1'],
167            'team2': m['team2'],
168            'winner': m['winner']
169        })
170 
171    return {
172        'username': player,
173        'elo': found_user['elo'],
174        'wins': found_user['wins'],
175        'losses': found_user['losses'],
176        'match_history': match_details
177    }
178 
179@app.get("/user/{user_id}")
180def get_user(user_id: int, authorization: str = Header(None)):
181    get_user_from_token(authorization)
182    u = users.get(user_id)
183    if not u:
184        raise HTTPException(status_code=404, detail="User not found")
185    return u
186 
187@app.post("/user")
188def create_user(req: UserCreate, authorization: str = Header(None)):
189    get_user_from_token(authorization)
190    global next_user_id
191    for u in users.values():
192        if u['username'] == req.username:
193            raise HTTPException(status_code=400, detail="Username already exists")
194    user_id = next_user_id
195    next_user_id += 1
196    users[user_id] = {
197        'id': user_id,
198        'username': req.username,
199        'elo': 1000,
200        'wins': 0,
201        'losses': 0
202    }
203    return users[user_id]

1fastapi
2uvicorn