Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from datetime import datetime
4from typing import Optional
5import secrets
6import uvicorn
7 
8app = FastAPI()
9 
10# In-memory stores
11users = {}
12tokens = {}
13workouts = {}
14workout_id_counter = 1
15user_id_counter = 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    user_id: int
27    type: str
28    duration_minutes: float
29    calories_burned: float
30    date: str
31 
32@app.post("/signup")
33def signup(req: SignupRequest):
34    global user_id_counter
35    if any(u["username"] == req.username for u in users.values()):
36        raise HTTPException(status_code=400, detail="Username already exists")
37    user_id = user_id_counter
38    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
39    user_id_counter += 1
40    return {"id": user_id, "username": req.username}
41 
42@app.post("/login")
43def login(req: LoginRequest):
44    for u in users.values():
45        if u["username"] == req.username and u["password"] == req.password:
46            token = secrets.token_hex(16)
47            tokens[token] = u["id"]
48            return {"token": token}
49    raise HTTPException(status_code=401, detail="Invalid credentials")
50 
51@app.get("/users/{user_id}")
52def get_user(user_id: int, authorization: Optional[str] = Header(None)):
53    if not authorization or authorization not in tokens:
54        raise HTTPException(status_code=401, detail="Invalid or missing token")
55    if user_id not in users:
56        raise HTTPException(status_code=404, detail="User not found")
57    return users[user_id]
58 
59@app.post("/workouts")
60def create_workout(workout: WorkoutCreate, authorization: Optional[str] = Header(None)):
61    global workout_id_counter
62    if not authorization or authorization not in tokens:
63        raise HTTPException(status_code=401, detail="Invalid or missing token")
64    if workout.user_id not in users:
65        raise HTTPException(status_code=400, detail="User not found")
66    workout_id = workout_id_counter
67    workouts[workout_id] = {
68        "id": workout_id,
69        "user_id": workout.user_id,
70        "type": workout.type,
71        "duration_minutes": workout.duration_minutes,
72        "calories_burned": workout.calories_burned,
73        "date": workout.date
74    }
75    workout_id_counter += 1
76    return workouts[workout_id]
77 
78@app.get("/workouts/{workout_id}")
79def get_workout(workout_id: int, authorization: Optional[str] = Header(None)):
80    if not authorization or authorization not in tokens:
81        raise HTTPException(status_code=401, detail="Invalid or missing token")
82    if workout_id not in workouts:
83        raise HTTPException(status_code=404, detail="Workout not found")
84    return workouts[workout_id]
85 
86@app.get("/workouts")
87def get_workouts(
88    user_id: Optional[int] = None,
89    type: Optional[str] = None,
90    after_date: Optional[str] = None,
91    authorization: Optional[str] = Header(None)
92):
93    if not authorization or authorization not in tokens:
94        raise HTTPException(status_code=401, detail="Invalid or missing token")
95    results = []
96    for w in workouts.values():
97        if user_id is not None and w["user_id"] != user_id:
98            continue
99        if type is not None and w["type"] != type:
100            continue
101        if after_date is not None and w["date"] < after_date:
102            continue
103        results.append({
104            "id": w["id"],
105            "total_duration_minutes": w["duration_minutes"],
106            "total_calories_burned": w["calories_burned"]
107        })
108    return results

1fastapi
2uvicorn