Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8next_user_id = 1
9tokens = {}
10 
11leagues = {}
12next_league_id = 1
13 
14players = {}
15next_player_id = 1
16 
17drafts = {}
18next_draft_id = 1
19 
20draft_picks = {}
21next_draft_pick_id = 1
22 
23def get_current_user(authorization: Optional[str] = Header(None)):
24    if not authorization:
25        raise HTTPException(status_code=401, detail="Missing auth token")
26    token = authorization.replace("Bearer ", "")
27    user_id = tokens.get(token)
28    if not user_id:
29        raise HTTPException(status_code=401, detail="Invalid token")
30    return user_id
31 
32@app.post("/signup")
33def signup(username: str, password: str):
34    global next_user_id
35    user_id = next_user_id
36    next_user_id += 1
37    users[user_id] = {"username": username, "password": password}
38    return {"user_id": user_id, "username": username}
39 
40@app.post("/login")
41def login(username: str, password: str):
42    for uid, u in users.items():
43        if u["username"] == username and u["password"] == password:
44            token = secrets.token_hex(16)
45            tokens[token] = uid
46            return {"token": token, "user_id": uid}
47    raise HTTPException(status_code=401, detail="Invalid credentials")
48 
49@app.get("/user/{user_id}")
50def get_user(user_id: int):
51    user = users.get(user_id)
52    if not user:
53        raise HTTPException(status_code=404, detail="User not found")
54    return {"user_id": user_id, "username": user["username"]}
55 
56@app.post("/user")
57def create_user(username: str, password: str):
58    global next_user_id
59    user_id = next_user_id
60    next_user_id += 1
61    users[user_id] = {"username": username, "password": password}
62    return {"user_id": user_id, "username": username}
63 
64@app.get("/league/{league_id}")
65def get_league(league_id: int):
66    league = leagues.get(league_id)
67    if not league:
68        raise HTTPException(status_code=404, detail="League not found")
69    return league
70 
71@app.post("/league")
72def create_league(name: str, max_teams: int, roster_limit: int, authorization: Optional[str] = Header(None)):
73    user_id = get_current_user(authorization)
74    global next_league_id
75    league_id = next_league_id
76    next_league_id += 1
77    leagues[league_id] = {
78        "id": league_id,
79        "name": name,
80        "max_teams": max_teams,
81        "roster_limit": roster_limit,
82        "teams": [],
83        "draft_order": [],
84        "current_pick": 0,
85        "commissioner_id": user_id
86    }
87    return leagues[league_id]
88 
89@app.get("/player/{player_id}")
90def get_player(player_id: int):
91    player = players.get(player_id)
92    if not player:
93        raise HTTPException(status_code=404, detail="Player not found")
94    return player
95 
96@app.post("/player")
97def create_player(name: str, position: str, authorization: Optional[str] = Header(None)):
98    get_current_user(authorization)
99    global next_player_id
100    player_id = next_player_id
101    next_player_id += 1
102    players[player_id] = {
103        "id": player_id,
104        "name": name,
105        "position": position,
106        "drafted": False
107    }
108    return players[player_id]
109 
110@app.post("/league/{league_id}/join")
111def join_league(league_id: int, team_name: str, authorization: Optional[str] = Header(None)):
112    user_id = get_current_user(authorization)
113    league = leagues.get(league_id)
114    if not league:
115        raise HTTPException(status_code=404, detail="League not found")
116    if len(league["teams"]) >= league["max_teams"]:
117        raise HTTPException(status_code=400, detail="League is full")
118    for team in league["teams"]:
119        if team["owner_id"] == user_id:
120            raise HTTPException(status_code=400, detail="Already in league")
121    team = {"owner_id": user_id, "team_name": team_name, "roster": []}
122    league["teams"].append(team)
123    league["draft_order"].append(user_id)
124    return {"team_name": team_name, "owner_id": user_id}
125 
126@app.post("/league/{league_id}/start_draft")
127def start_draft(league_id: int, authorization: Optional[str] = Header(None)):
128    user_id = get_current_user(authorization)
129    league = leagues.get(league_id)
130    if not league:
131        raise HTTPException(status_code=404, detail="League not found")
132    if league["commissioner_id"] != user_id:
133        raise HTTPException(status_code=403, detail="Only commissioner can start draft")
134    if len(league["teams"]) < 2:
135        raise HTTPException(status_code=400, detail="Need at least 2 teams")
136    import random
137    random.shuffle(league["draft_order"])
138    league["current_pick"] = 0
139    return {"message": "Draft started", "draft_order": league["draft_order"]}
140 
141@app.post("/league/{league_id}/draft")
142def draft_player(league_id: int, player_id: int, authorization: Optional[str] = Header(None)):
143    user_id = get_current_user(authorization)
144    league = leagues.get(league_id)
145    if not league:
146        raise HTTPException(status_code=404, detail="League not found")
147    if league["current_pick"] >= len(league["draft_order"]):
148        raise HTTPException(status_code=400, detail="Draft is over")
149    current_owner_id = league["draft_order"][league["current_pick"]]
150    if current_owner_id != user_id:
151        raise HTTPException(status_code=400, detail="Not your turn")
152    player = players.get(player_id)
153    if not player:
154        raise HTTPException(status_code=404, detail="Player not found")
155    if player["drafted"]:
156        raise HTTPException(status_code=400, detail="Player already drafted")
157    team = None
158    for t in league["teams"]:
159        if t["owner_id"] == user_id:
160            team = t
161            break
162    if not team:
163        raise HTTPException(status_code=400, detail="Team not found")
164    if len(team["roster"]) >= league["roster_limit"]:
165        raise HTTPException(status_code=400, detail="Roster full")
166    player["drafted"] = True
167    team["roster"].append(player_id)
168    league["current_pick"] += 1
169    return {"player_id": player_id, "team": team["team_name"], "pick_number": league["current_pick"]}
170 
171@app.get("/draft/{draft_id}")
172def get_draft(draft_id: int):
173    draft = drafts.get(draft_id)
174    if not draft:
175        raise HTTPException(status_code=404, detail="Draft not found")
176    return draft
177 
178@app.post("/draft")
179def create_draft(league_id: int, authorization: Optional[str] = Header(None)):
180    user_id = get_current_user(authorization)
181    league = leagues.get(league_id)
182    if not league:
183        raise HTTPException(status_code=404, detail="League not found")
184    global next_draft_id
185    draft_id = next_draft_id
186    next_draft_id += 1
187    drafts[draft_id] = {
188        "id": draft_id,
189        "league_id": league_id,
190        "picks": []
191    }
192    return drafts[draft_id]
193 
194@app.get("/draft_pick/{draft_pick_id}")
195def get_draft_pick(draft_pick_id: int):
196    pick = draft_picks.get(draft_pick_id)
197    if not pick:
198        raise HTTPException(status_code=404, detail="Draft pick not found")
199    return pick
200 
201@app.post("/draft_pick")
202def create_draft_pick(draft_id: int, player_id: int, team_owner_id: int, authorization: Optional[str] = Header(None)):
203    get_current_user(authorization)
204    global next_draft_pick_id
205    pick_id = next_draft_pick_id
206    next_draft_pick_id += 1
207    draft_picks[pick_id] = {
208        "id": pick_id,
209        "draft_id": draft_id,
210        "player_id": player_id,
211        "team_owner_id": team_owner_id
212    }
213    return draft_picks[pick_id]

1fastapi
2uvicorn