Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List, Dict
4import random
5import string
6 
7app = FastAPI()
8 
9# In-memory stores
10users = {}
11tokens = {}
12tournaments = {}
13matches = {}
14next_user_id = 1
15next_tournament_id = 1
16next_match_id = 1
17 
18def generate_token():
19    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
20 
21def get_current_user(authorization: str = Header(...)):
22    if not authorization.startswith("Bearer "):
23        raise HTTPException(status_code=401, detail="Invalid auth header")
24    token = authorization[7:]
25    for user_id, user_token in tokens.items():
26        if user_token == token:
27            return user_id
28    raise HTTPException(status_code=401, detail="Invalid token")
29 
30class UserCreate(BaseModel):
31    username: str
32    password: str
33 
34class UserLogin(BaseModel):
35    username: str
36    password: str
37 
38class TournamentCreate(BaseModel):
39    name: str
40    participants: List[str]
41    bracket_style: str = "single"  # single or double
42 
43class MatchScore(BaseModel):
44    score1: int
45    score2: int
46 
47@app.post("/signup")
48def signup(user: UserCreate):
49    global next_user_id
50    for u in users.values():
51        if u["username"] == user.username:
52            raise HTTPException(status_code=400, detail="Username already exists")
53    user_id = next_user_id
54    next_user_id += 1
55    users[user_id] = {"username": user.username, "password": user.password}
56    token = generate_token()
57    tokens[user_id] = token
58    return {"user_id": user_id, "token": token}
59 
60@app.post("/login")
61def login(user: UserLogin):
62    for uid, u in users.items():
63        if u["username"] == user.username and u["password"] == user.password:
64            token = generate_token()
65            tokens[uid] = token
66            return {"user_id": uid, "token": token}
67    raise HTTPException(status_code=401, detail="Invalid credentials")
68 
69@app.get("/users/{user_id}")
70def get_user(user_id: int, authorization: str = Header(...)):
71    get_current_user(authorization)
72    if user_id not in users:
73        raise HTTPException(status_code=404, detail="User not found")
74    return {"user_id": user_id, "username": users[user_id]["username"]}
75 
76@app.post("/tournaments")
77def create_tournament(tournament: TournamentCreate, authorization: str = Header(...)):
78    global next_tournament_id, next_match_id
79    user_id = get_current_user(authorization)
80    
81    participants = tournament.participants
82    if len(participants) < 2:
83        raise HTTPException(status_code=400, detail="Need at least 2 participants")
84    if (len(participants) & (len(participants) - 1)) != 0:
85        raise HTTPException(status_code=400, detail="Number of participants must be power of 2")
86    
87    tournament_id = next_tournament_id
88    next_tournament_id += 1
89    
90    # Generate bracket
91    random.shuffle(participants)
92    bracket = []
93    round_matches = []
94    for i in range(0, len(participants), 2):
95        match_id = next_match_id
96        next_match_id += 1
97        round_matches.append({
98            "match_id": match_id,
99            "participant1": participants[i],
100            "participant2": participants[i+1],
101            "score1": None,
102            "score2": None,
103            "winner": None
104        })
105        matches[match_id] = round_matches[-1]
106    bracket.append(round_matches)
107    
108    # Generate subsequent rounds
109    while len(round_matches) > 1:
110        next_round = []
111        for _ in range(0, len(round_matches), 2):
112            match_id = next_match_id
113            next_match_id += 1
114            next_round.append({
115                "match_id": match_id,
116                "participant1": None,
117                "participant2": None,
118                "score1": None,
119                "score2": None,
120                "winner": None
121            })
122            matches[match_id] = next_round[-1]
123        bracket.append(next_round)
124        round_matches = next_round
125    
126    tournaments[tournament_id] = {
127        "name": tournament.name,
128        "bracket_style": tournament.bracket_style,
129        "participants": participants,
130        "bracket": bracket,
131        "created_by": user_id
132    }
133    
134    return {"tournament_id": tournament_id, "bracket": bracket}
135 
136@app.get("/tournaments/{tournament_id}/bracket")
137def get_bracket(tournament_id: int, authorization: str = Header(...)):
138    get_current_user(authorization)
139    if tournament_id not in tournaments:
140        raise HTTPException(status_code=404, detail="Tournament not found")
141    return tournaments[tournament_id]["bracket"]
142 
143@app.post("/matches/{match_id}/score")
144def update_score(match_id: int, score: MatchScore, authorization: str = Header(...)):
145    get_current_user(authorization)
146    if match_id not in matches:
147        raise HTTPException(status_code=404, detail="Match not found")
148    
149    match = matches[match_id]
150    if match["score1"] is not None:
151        raise HTTPException(status_code=400, detail="Match already scored")
152    
153    match["score1"] = score.score1
154    match["score2"] = score.score2
155    if score.score1 > score.score2:
156        match["winner"] = match["participant1"]
157    elif score.score2 > score.score1:
158        match["winner"] = match["participant2"]
159    else:
160        raise HTTPException(status_code=400, detail="Cannot have tie")
161    
162    # Propagate winner to next round
163    for tid, tournament in tournaments.items():
164        for round_idx, round_matches in enumerate(tournament["bracket"]):
165            for m_idx, m in enumerate(round_matches):
166                if m["match_id"] == match_id:
167                    if round_idx + 1 < len(tournament["bracket"]):
168                        next_round = tournament["bracket"][round_idx + 1]
169                        next_match_idx = m_idx // 2
170                        if m_idx % 2 == 0:
171                            next_round[next_match_idx]["participant1"] = match["winner"]
172                        else:
173                            next_round[next_match_idx]["participant2"] = match["winner"]
174                    return {"match_id": match_id, "winner": match["winner"]}
175    
176    return {"match_id": match_id, "winner": match["winner"]}

1fastapi
2uvicorn