Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8workouts = {}
9trainers = {}
10logs = {}
11tokens = {}
12user_counter = 0
13workout_counter = 0
14trainer_counter = 0
15log_counter = 0
16 
17def get_current_user(authorization: Optional[str] = Header(None)):
18    if not authorization:
19        raise HTTPException(status_code=401, detail="Missing auth header")
20    token = authorization.replace("Bearer ", "")
21    user_id = tokens.get(token)
22    if not user_id:
23        raise HTTPException(status_code=401, detail="Invalid token")
24    return user_id
25 
26@app.post("/signup")
27def signup(username: str, password: str, role: str = "user"):
28    global user_counter
29    user_counter += 1
30    users[user_counter] = {"id": user_counter, "username": username, "password": password, "role": role}
31    return {"id": user_counter, "username": username, "role": role}
32 
33@app.post("/login")
34def login(username: str, password: str):
35    for uid, u in users.items():
36        if u["username"] == username and u["password"] == password:
37            token = secrets.token_hex(16)
38            tokens[token] = uid
39            return {"token": token}
40    raise HTTPException(status_code=401, detail="Invalid credentials")
41 
42@app.post("/workouts")
43def create_workout(name: str, description: str = "", authorization: Optional[str] = Header(None)):
44    current_user = get_current_user(authorization)
45    global workout_counter
46    workout_counter += 1
47    workouts[workout_counter] = {"id": workout_counter, "name": name, "description": description, "user_id": current_user}
48    return workouts[workout_counter]
49 
50@app.get("/workouts/{workout_id}")
51def get_workout(workout_id: int, authorization: Optional[str] = Header(None)):
52    get_current_user(authorization)
53    w = workouts.get(workout_id)
54    if not w:
55        raise HTTPException(status_code=404, detail="Workout not found")
56    return w
57 
58@app.post("/logs")
59def create_log(workout_id: int, reps: int = 0, weight: float = 0.0, notes: str = "", authorization: Optional[str] = Header(None)):
60    current_user = get_current_user(authorization)
61    global log_counter
62    log_counter += 1
63    logs[log_counter] = {"id": log_counter, "workout_id": workout_id, "reps": reps, "weight": weight, "notes": notes, "user_id": current_user}
64    return logs[log_counter]
65 
66@app.get("/logs/{log_id}")
67def get_log(log_id: int, authorization: Optional[str] = Header(None)):
68    get_current_user(authorization)
69    l = logs.get(log_id)
70    if not l:
71        raise HTTPException(status_code=404, detail="Log not found")
72    return l
73 
74@app.post("/trainers")
75def create_trainer(name: str, specialization: str = "", authorization: Optional[str] = Header(None)):
76    current_user = get_current_user(authorization)
77    global trainer_counter
78    trainer_counter += 1
79    trainers[trainer_counter] = {"id": trainer_counter, "name": name, "specialization": specialization, "user_id": current_user}
80    return trainers[trainer_counter]
81 
82@app.get("/trainers/{trainer_id}")
83def get_trainer(trainer_id: int, authorization: Optional[str] = Header(None)):
84    get_current_user(authorization)
85    t = trainers.get(trainer_id)
86    if not t:
87        raise HTTPException(status_code=404, detail="Trainer not found")
88    return t
89 
90@app.get("/logs/by-workout/{workout_id}")
91def get_logs_by_workout(workout_id: int, authorization: Optional[str] = Header(None)):
92    get_current_user(authorization)
93    result = [l for l in logs.values() if l["workout_id"] == workout_id]
94    return result

1fastapi
2uvicorn