Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3from datetime import datetime, timedelta
4import hashlib
5import secrets
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11workouts = {}
12workout_id_counter = 1
13user_workout_ids = {}
14 
15def generate_token():
16    return secrets.token_hex(16)
17 
18def hash_password(password: str):
19    return hashlib.sha256(password.encode()).hexdigest()
20 
21def get_user_from_token(authorization: Optional[str] = Header(None)):
22    if not authorization:
23        raise HTTPException(status_code=401, detail="No auth token")
24    token = authorization.replace("Bearer ", "")
25    if token not in tokens:
26        raise HTTPException(status_code=401, detail="Invalid token")
27    return tokens[token]
28 
29@app.post("/signup")
30def signup(username: str, password: str):
31    if username in users:
32        raise HTTPException(status_code=400, detail="User exists")
33    users[username] = {"password": hash_password(password), "created_at": datetime.now()}
34    token = generate_token()
35    tokens[token] = username
36    return {"token": token, "username": username}
37 
38@app.post("/login")
39def login(username: str, password: str):
40    if username not in users:
41        raise HTTPException(status_code=401, detail="Invalid credentials")
42    if users[username]["password"] != hash_password(password):
43        raise HTTPException(status_code=401, detail="Invalid credentials")
44    token = generate_token()
45    tokens[token] = username
46    return {"token": token, "username": username}
47 
48@app.post("/workouts")
49def create_workout(authorization: Optional[str] = Header(None)):
50    global workout_id_counter
51    username = get_user_from_token(authorization)
52    workout_id = workout_id_counter
53    workout_id_counter += 1
54    workouts[workout_id] = {
55        "id": workout_id,
56        "username": username,
57        "exercises": [],
58        "created_at": datetime.now().isoformat()
59    }
60    if username not in user_workout_ids:
61        user_workout_ids[username] = []
62    user_workout_ids[username].append(workout_id)
63    return workouts[workout_id]
64 
65@app.get("/workouts/{workout_id}")
66def get_workout(workout_id: int, authorization: Optional[str] = Header(None)):
67    username = get_user_from_token(authorization)
68    if workout_id not in workouts:
69        raise HTTPException(status_code=404, detail="Workout not found")
70    return workouts[workout_id]
71 
72@app.post("/workouts/{workout_id}/exercises")
73def add_exercise(workout_id: int, exercise_name: str, sets: int, reps: int, weight: float, authorization: Optional[str] = Header(None)):
74    username = get_user_from_token(authorization)
75    if workout_id not in workouts:
76        raise HTTPException(status_code=404, detail="Workout not found")
77    workout = workouts[workout_id]
78    exercise = {
79        "name": exercise_name,
80        "sets": sets,
81        "reps": reps,
82        "weight": weight
83    }
84    workout["exercises"].append(exercise)
85    return workout
86 
87@app.get("/weekly-summary")
88def weekly_summary(authorization: Optional[str] = Header(None)):
89    username = get_user_from_token(authorization)
90    now = datetime.now()
91    week_ago = now - timedelta(days=7)
92    
93    total_exercises = 0
94    total_sets = 0
95    total_reps = 0
96    total_volume = 0.0
97    workout_count = 0
98    
99    if username in user_workout_ids:
100        for wid in user_workout_ids[username]:
101            workout = workouts[wid]
102            created = datetime.fromisoformat(workout["created_at"])
103            if created >= week_ago:
104                workout_count += 1
105                for ex in workout["exercises"]:
106                    total_exercises += 1
107                    total_sets += ex["sets"]
108                    total_reps += ex["reps"] * ex["sets"]
109                    total_volume += ex["weight"] * ex["reps"] * ex["sets"]
110    
111    return {
112        "username": username,
113        "workout_count": workout_count,
114        "total_exercises": total_exercises,
115        "total_sets": total_sets,
116        "total_reps": total_reps,
117        "total_volume_kg": round(total_volume, 2),
118        "period": "last 7 days"
119    }

1fastapi
2uvicorn