Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import uuid
4import hashlib
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10workouts = {}
11workout_logs = {}
12next_user_id = 1
13next_workout_id = 1
14next_log_id = 1
15 
16def hash_password(password: str) -> str:
17    return hashlib.sha256(password.encode()).hexdigest()
18 
19def get_current_user(authorization: Optional[str] = Header(None)):
20    if not authorization:
21        raise HTTPException(status_code=401, detail="Missing auth token")
22    token = authorization.replace("Bearer ", "")
23    if token not in tokens:
24        raise HTTPException(status_code=401, detail="Invalid token")
25    return tokens[token]
26 
27@app.post("/signup")
28def signup(username: str, password: str):
29    global next_user_id
30    if any(u["username"] == username for u in users.values()):
31        raise HTTPException(status_code=400, detail="Username already exists")
32    user_id = next_user_id
33    users[user_id] = {"id": user_id, "username": username, "password": hash_password(password)}
34    next_user_id += 1
35    return {"user_id": user_id, "username": username}
36 
37@app.post("/login")
38def login(username: str, password: str):
39    for user in users.values():
40        if user["username"] == username and user["password"] == hash_password(password):
41            token = str(uuid.uuid4())
42            tokens[token] = user["id"]
43            return {"token": token}
44    raise HTTPException(status_code=401, detail="Invalid credentials")
45 
46@app.post("/workouts")
47def create_workout(name: str, description: str = "", authorization: Optional[str] = Header(None)):
48    user_id = get_current_user(authorization)
49    global next_workout_id
50    workout_id = next_workout_id
51    workouts[workout_id] = {"id": workout_id, "name": name, "description": description, "trainer_id": user_id}
52    next_workout_id += 1
53    return workouts[workout_id]
54 
55@app.get("/workouts/{workout_id}")
56def get_workout(workout_id: int):
57    if workout_id not in workouts:
58        raise HTTPException(status_code=404, detail="Workout not found")
59    return workouts[workout_id]
60 
61@app.post("/logs")
62def create_log(workout_id: int, reps: int, sets: int, weight: float = 0, authorization: Optional[str] = Header(None)):
63    user_id = get_current_user(authorization)
64    if workout_id not in workouts:
65        raise HTTPException(status_code=404, detail="Workout not found")
66    global next_log_id
67    log_id = next_log_id
68    workout_logs[log_id] = {"id": log_id, "workout_id": workout_id, "user_id": user_id, "reps": reps, "sets": sets, "weight": weight}
69    next_log_id += 1
70    return workout_logs[log_id]
71 
72@app.get("/logs/{log_id}")
73def get_log(log_id: int, authorization: Optional[str] = Header(None)):
74    user_id = get_current_user(authorization)
75    if log_id not in workout_logs:
76        raise HTTPException(status_code=404, detail="Log not found")
77    log = workout_logs[log_id]
78    if log["user_id"] != user_id:
79        raise HTTPException(status_code=403, detail="Not your log")
80    return log
81 
82@app.get("/trainer/logs/{workout_id}")
83def get_logs_by_workout(workout_id: int, authorization: Optional[str] = Header(None)):
84    trainer_id = get_current_user(authorization)
85    if workout_id not in workouts:
86        raise HTTPException(status_code=404, detail="Workout not found")
87    workout = workouts[workout_id]
88    if workout["trainer_id"] != trainer_id:
89        raise HTTPException(status_code=403, detail="You are not the trainer for this workout")
90    logs = [log for log in workout_logs.values() if log["workout_id"] == workout_id]
91    return logs

1fastapi
2uvicorn