Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9next_user_id = 1
10 
11workouts = {}
12next_workout_id = 1
13 
14fitness_summaries = {}
15next_summary_id = 1
16 
17class SignupRequest(BaseModel):
18    username: str
19    password: str
20 
21class LoginRequest(BaseModel):
22    username: str
23    password: str
24 
25class WorkoutCreate(BaseModel):
26    type: str
27    duration_minutes: int
28    calories_burned: int
29 
30class FitnessSummaryCreate(BaseModel):
31    user_id: int
32    step_count: int
33    calories_burned: int
34    weekly_streaks: int
35 
36def get_current_user(authorization: str = Header(...)):
37    if not authorization.startswith("Bearer "):
38        raise HTTPException(status_code=401, detail="Invalid auth header")
39    token = authorization[7:]
40    user_id = tokens.get(token)
41    if user_id is None:
42        raise HTTPException(status_code=401, detail="Invalid token")
43    return user_id
44 
45@app.post("/signup")
46def signup(req: SignupRequest):
47    global next_user_id
48    for u in users.values():
49        if u["username"] == req.username:
50            raise HTTPException(status_code=400, detail="Username taken")
51    user_id = next_user_id
52    next_user_id += 1
53    users[user_id] = {"username": req.username, "password": req.password}
54    return {"user_id": user_id}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    for uid, u in users.items():
59        if u["username"] == req.username and u["password"] == req.password:
60            token = secrets.token_hex(16)
61            tokens[token] = uid
62            return {"token": token, "user_id": uid}
63    raise HTTPException(status_code=401, detail="Invalid credentials")
64 
65@app.get("/workouts/{workout_id}")
66def get_workout(workout_id: int, authorization: str = Header(...)):
67    get_current_user(authorization)
68    w = workouts.get(workout_id)
69    if w is None:
70        raise HTTPException(status_code=404, detail="Workout not found")
71    return w
72 
73@app.post("/workouts")
74def create_workout(workout: WorkoutCreate, authorization: str = Header(...)):
75    user_id = get_current_user(authorization)
76    global next_workout_id
77    wid = next_workout_id
78    next_workout_id += 1
79    workouts[wid] = {"id": wid, "user_id": user_id, "type": workout.type, "duration_minutes": workout.duration_minutes, "calories_burned": workout.calories_burned}
80    return workouts[wid]
81 
82@app.get("/fitness/summary")
83def get_fitness_summary(authorization: str = Header(...)):
84    user_id = get_current_user(authorization)
85    total_steps = 0
86    total_calories = 0
87    weekly_streaks = 0
88    for sid, s in fitness_summaries.items():
89        if s["user_id"] == user_id:
90            total_steps += s["step_count"]
91            total_calories += s["calories_burned"]
92            weekly_streaks = max(weekly_streaks, s["weekly_streaks"])
93    return {"step_count": total_steps, "calories_burned": total_calories, "weekly_streaks": weekly_streaks}
94 
95@app.post("/fitness/summary")
96def create_fitness_summary(summary: FitnessSummaryCreate, authorization: str = Header(...)):
97    user_id = get_current_user(authorization)
98    global next_summary_id
99    sid = next_summary_id
100    next_summary_id += 1
101    fitness_summaries[sid] = {"id": sid, "user_id": user_id, "step_count": summary.step_count, "calories_burned": summary.calories_burned, "weekly_streaks": summary.weekly_streaks}
102    return fitness_summaries[sid]

1fastapi
2uvicorn