Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import uuid
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9workout_sessions = {}
10workout_session_id_counter = 1
11exercises = {}
12exercise_id_counter = 1
13 
14def get_user_from_token(authorization: Optional[str] = Header(None)):
15    if not authorization:
16        raise HTTPException(status_code=401, detail="No auth header")
17    token = authorization.replace("Bearer ", "")
18    for user_id, t in tokens.items():
19        if t == token:
20            return user_id
21    raise HTTPException(status_code=401, detail="Invalid token")
22 
23@app.post("/signup")
24def signup(username: str, password: str):
25    if username in users:
26        raise HTTPException(status_code=400, detail="User exists")
27    users[username] = password
28    token = str(uuid.uuid4())
29    tokens[username] = token
30    return {"username": username, "token": token}
31 
32@app.post("/login")
33def login(username: str, password: str):
34    if username not in users or users[username] != password:
35        raise HTTPException(status_code=401, detail="Bad creds")
36    token = str(uuid.uuid4())
37    tokens[username] = token
38    return {"username": username, "token": token}
39 
40@app.post("/workout-sessions")
41def create_workout_session(workout_type: str, authorization: Optional[str] = Header(None)):
42    global workout_session_id_counter
43    user_id = get_user_from_token(authorization)
44    if workout_type not in ("strength", "cardio"):
45        raise HTTPException(status_code=400, detail="Invalid workout type")
46    sess = {
47        "id": workout_session_id_counter,
48        "user_id": user_id,
49        "workout_type": workout_type,
50        "exercises": []
51    }
52    workout_sessions[workout_session_id_counter] = sess
53    workout_session_id_counter += 1
54    return sess
55 
56@app.get("/workout-sessions/{session_id}")
57def get_workout_session(session_id: int, authorization: Optional[str] = Header(None)):
58    user_id = get_user_from_token(authorization)
59    if session_id not in workout_sessions:
60        raise HTTPException(status_code=404, detail="Not found")
61    sess = workout_sessions[session_id]
62    if sess["user_id"] != user_id:
63        raise HTTPException(status_code=403, detail="Forbidden")
64    return sess
65 
66@app.post("/workout-sessions/{session_id}/exercises")
67def add_exercise(session_id: int, name: str, sets: int, reps: int, weight: float, authorization: Optional[str] = Header(None)):
68    global exercise_id_counter
69    user_id = get_user_from_token(authorization)
70    if session_id not in workout_sessions:
71        raise HTTPException(status_code=404, detail="Session not found")
72    sess = workout_sessions[session_id]
73    if sess["user_id"] != user_id:
74        raise HTTPException(status_code=403, detail="Forbidden")
75    ex = {
76        "id": exercise_id_counter,
77        "name": name,
78        "sets": sets,
79        "reps": reps,
80        "weight": weight
81    }
82    exercises[exercise_id_counter] = ex
83    sess["exercises"].append(exercise_id_counter)
84    exercise_id_counter += 1
85    return ex
86 
87@app.get("/exercises/{exercise_id}")
88def get_exercise(exercise_id: int, authorization: Optional[str] = Header(None)):
89    user_id = get_user_from_token(authorization)
90    if exercise_id not in exercises:
91        raise HTTPException(status_code=404, detail="Not found")
92    return exercises[exercise_id]

1fastapi
2uvicorn