Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4from datetime import datetime
5import secrets
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11sessions = {}
12session_history = {}
13next_user_id = 1
14next_session_id = 1
15 
16class SignupRequest(BaseModel):
17    username: str
18    password: str
19 
20class LoginRequest(BaseModel):
21    username: str
22    password: str
23 
24class SessionCreate(BaseModel):
25    workout_type: str
26    duration: int
27    intensity_level: str
28 
29class SessionUpdate(BaseModel):
30    workout_type: Optional[str] = None
31    duration: Optional[int] = None
32    intensity_level: Optional[str] = None
33    notes: Optional[str] = None
34 
35def get_user_from_token(authorization: str = Header(None)):
36    if not authorization:
37        raise HTTPException(status_code=401, detail="Missing auth header")
38    token = authorization.replace("Bearer ", "")
39    if token not in tokens:
40        raise HTTPException(status_code=401, detail="Invalid token")
41    return tokens[token]
42 
43@app.post("/signup")
44def signup(req: SignupRequest):
45    global next_user_id
46    for user in users.values():
47        if user["username"] == req.username:
48            raise HTTPException(status_code=400, detail="Username taken")
49    user_id = next_user_id
50    next_user_id += 1
51    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
52    return {"id": user_id, "username": req.username}
53 
54@app.post("/login")
55def login(req: LoginRequest):
56    for user in users.values():
57        if user["username"] == req.username and user["password"] == req.password:
58            token = secrets.token_hex(16)
59            tokens[token] = user["id"]
60            return {"token": token}
61    raise HTTPException(status_code=401, detail="Invalid credentials")
62 
63@app.post("/sessions")
64def create_session(req: SessionCreate, authorization: str = Header(None)):
65    user_id = get_user_from_token(authorization)
66    global next_session_id
67    session_id = next_session_id
68    next_session_id += 1
69    now = datetime.utcnow().isoformat()
70    sessions[session_id] = {
71        "id": session_id,
72        "user_id": user_id,
73        "workout_type": req.workout_type,
74        "duration": req.duration,
75        "intensity_level": req.intensity_level,
76        "created_at": now,
77        "updated_at": now,
78        "notes": None
79    }
80    session_history[session_id] = []
81    return sessions[session_id]
82 
83@app.get("/sessions/{session_id}")
84def get_session(session_id: int, authorization: str = Header(None)):
85    user_id = get_user_from_token(authorization)
86    if session_id not in sessions:
87        raise HTTPException(status_code=404, detail="Session not found")
88    return sessions[session_id]
89 
90@app.patch("/sessions/{session_id}")
91def update_session(session_id: int, req: SessionUpdate, authorization: str = Header(None)):
92    user_id = get_user_from_token(authorization)
93    if session_id not in sessions:
94        raise HTTPException(status_code=404, detail="Session not found")
95    session = sessions[session_id]
96    now = datetime.utcnow().isoformat()
97    changes = {}
98    if req.workout_type is not None:
99        changes["workout_type"] = {"old": session["workout_type"], "new": req.workout_type}
100        session["workout_type"] = req.workout_type
101    if req.duration is not None:
102        changes["duration"] = {"old": session["duration"], "new": req.duration}
103        session["duration"] = req.duration
104    if req.intensity_level is not None:
105        changes["intensity_level"] = {"old": session["intensity_level"], "new": req.intensity_level}
106        session["intensity_level"] = req.intensity_level
107    if req.notes is not None:
108        changes["notes"] = {"old": session["notes"], "new": req.notes}
109        session["notes"] = req.notes
110    session["updated_at"] = now
111    if changes:
112        session_history[session_id].append({
113            "timestamp": now,
114            "changes": changes
115        })
116    return session
117 
118@app.get("/sessions/{session_id}/history")
119def get_session_history(session_id: int, authorization: str = Header(None)):
120    user_id = get_user_from_token(authorization)
121    if session_id not in sessions:
122        raise HTTPException(status_code=404, detail="Session not found")
123    return session_history.get(session_id, [])

1fastapi
2uvicorn