Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3from datetime import datetime
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9workouts = {}
10progress = {}
11tokens = {}
12user_id_counter = 1
13workout_id_counter = 1
14 
15@app.post("/signup")
16def signup(username: str, password: str):
17    global user_id_counter
18    if username in users:
19        raise HTTPException(400, "User exists")
20    user_id = user_id_counter
21    user_id_counter += 1
22    users[username] = {"id": user_id, "password": password}
23    return {"user_id": user_id}
24 
25@app.post("/login")
26def login(username: str, password: str):
27    if username not in users or users[username]["password"] != password:
28        raise HTTPException(401, "Invalid credentials")
29    token = secrets.token_hex(16)
30    tokens[token] = users[username]["id"]
31    return {"token": token}
32 
33def get_current_user(authorization: Optional[str] = Header(None)):
34    if not authorization:
35        raise HTTPException(401, "No auth header")
36    token = authorization.replace("Bearer ", "")
37    if token not in tokens:
38        raise HTTPException(401, "Invalid token")
39    return tokens[token]
40 
41@app.get("/workouts")
42def get_workouts(authorization: Optional[str] = Header(None)):
43    user_id = get_current_user(authorization)
44    user_workouts = [w for w in workouts.values() if w["user_id"] == user_id]
45    result = {}
46    for w in user_workouts:
47        exercise = w["exercise"]
48        if exercise not in result:
49            result[exercise] = []
50        result[exercise].append({
51            "sets": w["sets"],
52            "reps": w["reps"],
53            "weight": w["weight"],
54            "timestamp": w["timestamp"]
55        })
56    return result
57 
58@app.post("/workouts/log")
59def log_workout(exercise: str, sets: int, reps: int, weight: float, authorization: Optional[str] = Header(None)):
60    user_id = get_current_user(authorization)
61    global workout_id_counter
62    wid = workout_id_counter
63    workout_id_counter += 1
64    workouts[wid] = {
65        "id": wid,
66        "user_id": user_id,
67        "exercise": exercise,
68        "sets": sets,
69        "reps": reps,
70        "weight": weight,
71        "timestamp": datetime.now().isoformat()
72    }
73    if exercise not in progress:
74        progress[exercise] = []
75    progress[exercise].append({
76        "weight": weight,
77        "timestamp": workouts[wid]["timestamp"]
78    })
79    return {"workout_id": wid}
80 
81@app.get("/progress/{exercise}")
82def get_progress(exercise: str, authorization: Optional[str] = Header(None)):
83    user_id = get_current_user(authorization)
84    if exercise not in progress:
85        raise HTTPException(404, "No data for exercise")
86    return progress[exercise]
87 
88@app.get("/workouts/{workout_id}")
89def get_workout(workout_id: int, authorization: Optional[str] = Header(None)):
90    user_id = get_current_user(authorization)
91    if workout_id not in workouts:
92        raise HTTPException(404, "Workout not found")
93    w = workouts[workout_id]
94    if w["user_id"] != user_id:
95        raise HTTPException(403, "Not your workout")
96    return w
97 
98@app.post("/workouts")
99def create_workout(exercise: str, sets: int, reps: int, weight: float, authorization: Optional[str] = Header(None)):
100    return log_workout(exercise, sets, reps, weight, authorization)

1fastapi
2uvicorn