Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import secrets
5import uvicorn
6 
7app = FastAPI()
8 
9# In-memory storage
10users = {}
11teams = {}
12next_user_id = 1
13next_team_id = 1
14tokens = {}
15 
16# Auth helper
17def get_user_from_token(authorization: str = Header(None)):
18    if not authorization:
19        raise HTTPException(status_code=401, detail="Missing auth header")
20    token = authorization.replace("Bearer ", "")
21    user_id = tokens.get(token)
22    if not user_id:
23        raise HTTPException(status_code=401, detail="Invalid token")
24    return user_id
25 
26# User models
27class SignupRequest(BaseModel):
28    username: str
29    password: str
30 
31class LoginRequest(BaseModel):
32    username: str
33    password: str
34 
35# Team models
36class TeamCreate(BaseModel):
37    name: str
38    roster: List[str] = []
39    budget: float = 100.0
40    is_active: bool = True
41 
42class TeamUpdate(BaseModel):
43    name: Optional[str] = None
44    roster: Optional[List[str]] = None
45    budget: Optional[float] = None
46    is_active: Optional[bool] = None
47 
48# User endpoints
49@app.post("/signup")
50def signup(req: SignupRequest):
51    global next_user_id
52    for u in users.values():
53        if u["username"] == req.username:
54            raise HTTPException(status_code=400, detail="Username taken")
55    user_id = next_user_id
56    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
57    next_user_id += 1
58    return {"id": user_id, "username": req.username}
59 
60@app.post("/login")
61def login(req: LoginRequest):
62    for u in users.values():
63        if u["username"] == req.username and u["password"] == req.password:
64            token = secrets.token_hex(16)
65            tokens[token] = u["id"]
66            return {"token": token}
67    raise HTTPException(status_code=401, detail="Invalid credentials")
68 
69# Team endpoints
70@app.get("/teams/{team_id}")
71def get_team(team_id: int, authorization: str = Header(None)):
72    user_id = get_user_from_token(authorization)
73    team = teams.get(team_id)
74    if not team:
75        raise HTTPException(status_code=404, detail="Team not found")
76    return team
77 
78@app.post("/teams")
79def create_team(team: TeamCreate, authorization: str = Header(None)):
80    global next_team_id
81    user_id = get_user_from_token(authorization)
82    team_id = next_team_id
83    teams[team_id] = {
84        "id": team_id,
85        "owner_id": user_id,
86        "name": team.name,
87        "roster": team.roster,
88        "budget": team.budget,
89        "is_active": team.is_active
90    }
91    next_team_id += 1
92    return teams[team_id]
93 
94@app.patch("/teams/{team_id}")
95def update_team(team_id: int, update: TeamUpdate, authorization: str = Header(None)):
96    user_id = get_user_from_token(authorization)
97    team = teams.get(team_id)
98    if not team:
99        raise HTTPException(status_code=404, detail="Team not found")
100    if team["owner_id"] != user_id:
101        raise HTTPException(status_code=403, detail="Not your team")
102    if update.name is not None:
103        team["name"] = update.name
104    if update.roster is not None:
105        team["roster"] = update.roster
106    if update.budget is not None:
107        team["budget"] = update.budget
108    if update.is_active is not None:
109        team["is_active"] = update.is_active
110    return team

1fastapi
2uvicorn