Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import uuid
4import datetime
5 
6app = FastAPI()
7 
8users = {}
9teams = {}
10workouts = {}
11tokens = {}
12next_user_id = 1
13next_team_id = 1
14next_workout_id = 1
15 
16def get_user_from_token(authorization: Optional[str] = Header(None)):
17    if not authorization:
18        raise HTTPException(status_code=401, detail="Missing auth token")
19    token = authorization.replace("Bearer ", "")
20    if token not in tokens:
21        raise HTTPException(status_code=401, detail="Invalid token")
22    return tokens[token]
23 
24@app.post("/signup")
25def signup(username: str, password: str):
26    global next_user_id
27    if any(u["username"] == username for u in users.values()):
28        raise HTTPException(status_code=400, detail="Username taken")
29    user_id = next_user_id
30    users[user_id] = {"id": user_id, "username": username, "password": password, "role": "member"}
31    next_user_id += 1
32    return {"user_id": user_id, "username": username}
33 
34@app.post("/login")
35def login(username: str, password: str):
36    for u in users.values():
37        if u["username"] == username and u["password"] == password:
38            token = str(uuid.uuid4())
39            tokens[token] = u["id"]
40            return {"token": token}
41    raise HTTPException(status_code=401, detail="Invalid credentials")
42 
43@app.post("/team")
44def create_team(name: str, authorization: Optional[str] = Header(None)):
45    global next_team_id
46    user_id = get_user_from_token(authorization)
47    team_id = next_team_id
48    teams[team_id] = {
49        "id": team_id,
50        "name": name,
51        "captain_id": user_id,
52        "co_captains": [],
53        "members": [user_id],
54        "workout_ids": [],
55        "weekly_scores": {}
56    }
57    next_team_id += 1
58    users[user_id]["role"] = "captain"
59    return {"team_id": team_id, "name": name}
60 
61@app.post("/team/{team_id}/join")
62def join_team(team_id: int, authorization: Optional[str] = Header(None)):
63    user_id = get_user_from_token(authorization)
64    if team_id not in teams:
65        raise HTTPException(status_code=404, detail="Team not found")
66    if user_id in teams[team_id]["members"]:
67        raise HTTPException(status_code=400, detail="Already in team")
68    teams[team_id]["members"].append(user_id)
69    return {"message": "Joined team"}
70 
71@app.post("/team/{team_id}/promote")
72def promote_to_co_captain(team_id: int, user_id: int, authorization: Optional[str] = Header(None)):
73    current_user = get_user_from_token(authorization)
74    if team_id not in teams:
75        raise HTTPException(status_code=404, detail="Team not found")
76    team = teams[team_id]
77    if team["captain_id"] != current_user:
78        raise HTTPException(status_code=403, detail="Only captain can promote")
79    if user_id not in team["members"]:
80        raise HTTPException(status_code=400, detail="User not in team")
81    if user_id in team["co_captains"]:
82        raise HTTPException(status_code=400, detail="Already co-captain")
83    team["co_captains"].append(user_id)
84    users[user_id]["role"] = "co_captain"
85    return {"message": f"User {user_id} promoted to co-captain"}
86 
87@app.post("/workout")
88def log_workout(team_id: int, exercise: str, duration_minutes: int, calories: int, authorization: Optional[str] = Header(None)):
89    global next_workout_id
90    user_id = get_user_from_token(authorization)
91    if team_id not in teams:
92        raise HTTPException(status_code=404, detail="Team not found")
93    if user_id not in teams[team_id]["members"]:
94        raise HTTPException(status_code=403, detail="Not a member of this team")
95    workout_id = next_workout_id
96    now = datetime.datetime.now()
97    iso_date = now.date().isoformat()
98    week_start = (now - datetime.timedelta(days=now.weekday())).date().isoformat()
99    workouts[workout_id] = {
100        "id": workout_id,
101        "team_id": team_id,
102        "user_id": user_id,
103        "exercise": exercise,
104        "duration_minutes": duration_minutes,
105        "calories": calories,
106        "date": iso_date,
107        "week_start": week_start
108    }
109    teams[team_id]["workout_ids"].append(workout_id)
110    # Update weekly scores
111    if week_start not in teams[team_id]["weekly_scores"]:
112        teams[team_id]["weekly_scores"][week_start] = {}
113    week = teams[team_id]["weekly_scores"][week_start]
114    if user_id not in week:
115        week[user_id] = {"total_calories": 0, "total_duration": 0, "workout_count": 0}
116    week[user_id]["total_calories"] += calories
117    week[user_id]["total_duration"] += duration_minutes
118    week[user_id]["workout_count"] += 1
119    next_workout_id += 1
120    return {"workout_id": workout_id}
121 
122@app.get("/leaderboard/weekly/{team_id}")
123def weekly_leaderboard(team_id: int, week_start: Optional[str] = None, authorization: Optional[str] = Header(None)):
124    get_user_from_token(authorization)
125    if team_id not in teams:
126        raise HTTPException(status_code=404, detail="Team not found")
127    if not week_start:
128        now = datetime.datetime.now()
129        week_start = (now - datetime.timedelta(days=now.weekday())).date().isoformat()
130    if week_start not in teams[team_id]["weekly_scores"]:
131        return {"team_id": team_id, "week_start": week_start, "leaderboard": []}
132    scores = teams[team_id]["weekly_scores"][week_start]
133    sorted_users = sorted(scores.items(), key=lambda x: x[1]["total_calories"], reverse=True)
134    leaderboard = []
135    for rank, (uid, data) in enumerate(sorted_users, 1):
136        leaderboard.append({
137            "rank": rank,
138            "user_id": uid,
139            "username": users[uid]["username"],
140            "total_calories": data["total_calories"],
141            "total_duration": data["total_duration"],
142            "workout_count": data["workout_count"]
143        })
144    return {"team_id": team_id, "week_start": week_start, "leaderboard": leaderboard}
145 
146@app.get("/user/{user_id}")
147def get_user(user_id: int, authorization: Optional[str] = Header(None)):
148    get_user_from_token(authorization)
149    if user_id not in users:
150        raise HTTPException(status_code=404, detail="User not found")
151    return users[user_id]
152 
153@app.get("/team/{team_id}")
154def get_team(team_id: int, authorization: Optional[str] = Header(None)):
155    get_user_from_token(authorization)
156    if team_id not in teams:
157        raise HTTPException(status_code=404, detail="Team not found")
158    return teams[team_id]
159 
160@app.get("/workout/{workout_id}")
161def get_workout(workout_id: int, authorization: Optional[str] = Header(None)):
162    get_user_from_token(authorization)
163    if workout_id not in workouts:
164        raise HTTPException(status_code=404, detail="Workout not found")
165    return workouts[workout_id]

1fastapi
2uvicorn