Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8members = {}
9workouts = {}
10next_user_id = 1
11next_member_id = 1
12next_workout_id = 1
13tokens = {}
14 
15@app.post("/signup")
16def signup(username: str, password: str):
17    global next_user_id
18    user_id = next_user_id
19    next_user_id += 1
20    token = secrets.token_hex(16)
21    is_owner = len(users) == 0
22    users[user_id] = {"id": user_id, "username": username, "password": password, "role": "owner" if is_owner else "member"}
23    tokens[token] = user_id
24    return {"user_id": user_id, "token": token, "role": users[user_id]["role"]}
25 
26@app.post("/login")
27def login(username: str, password: str):
28    for uid, u in users.items():
29        if u["username"] == username and u["password"] == password:
30            token = secrets.token_hex(16)
31            tokens[token] = uid
32            return {"token": token, "user_id": uid}
33    raise HTTPException(status_code=401, detail="Invalid credentials")
34 
35def get_current_user(authorization: Optional[str] = Header(None)):
36    if not authorization or not authorization.startswith("Bearer "):
37        raise HTTPException(status_code=401, detail="Invalid token")
38    token = authorization.split(" ")[1]
39    user_id = tokens.get(token)
40    if user_id is None:
41        raise HTTPException(status_code=401, detail="Invalid token")
42    return users[user_id]
43 
44@app.post("/members")
45def create_member(name: str, authorization: Optional[str] = Header(None)):
46    current_user = get_current_user(authorization)
47    global next_member_id
48    member_id = next_member_id
49    next_member_id += 1
50    members[member_id] = {"id": member_id, "name": name, "user_id": current_user["id"], "role": "member"}
51    return members[member_id]
52 
53@app.get("/members/{member_id}")
54def get_member(member_id: int, authorization: Optional[str] = Header(None)):
55    get_current_user(authorization)
56    if member_id not in members:
57        raise HTTPException(status_code=404, detail="Member not found")
58    return members[member_id]
59 
60@app.post("/workouts")
61def log_workout(member_id: int, duration: int, calories: int, authorization: Optional[str] = Header(None)):
62    current_user = get_current_user(authorization)
63    if member_id not in members:
64        raise HTTPException(status_code=404, detail="Member not found")
65    if members[member_id]["user_id"] != current_user["id"]:
66        raise HTTPException(status_code=403, detail="Not your member")
67    global next_workout_id
68    workout_id = next_workout_id
69    next_workout_id += 1
70    workouts[workout_id] = {"id": workout_id, "member_id": member_id, "duration": duration, "calories": calories}
71    return workouts[workout_id]
72 
73@app.get("/workouts/{workout_id}")
74def get_workout(workout_id: int, authorization: Optional[str] = Header(None)):
75    get_current_user(authorization)
76    if workout_id not in workouts:
77        raise HTTPException(status_code=404, detail="Workout not found")
78    return workouts[workout_id]
79 
80@app.post("/members/{member_id}/promote-to-coach")
81def promote_to_coach(member_id: int, authorization: Optional[str] = Header(None)):
82    current_user = get_current_user(authorization)
83    if current_user["role"] != "owner":
84        raise HTTPException(status_code=403, detail="Only owner can promote")
85    if member_id not in members:
86        raise HTTPException(status_code=404, detail="Member not found")
87    members[member_id]["role"] = "coach"
88    return members[member_id]

1fastapi
2uvicorn