Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional, List, Dict
3from pydantic import BaseModel
4import secrets
5 
6app = FastAPI()
7 
8users_db = {}
9auth_tokens = {}
10user_counter = 0
11 
12class UserSignup(BaseModel):
13    username: str
14    password: str
15    email: str
16 
17class UserLogin(BaseModel):
18    username: str
19    password: str
20 
21class UserPatch(BaseModel):
22    weight: Optional[float] = None
23    goals: Optional[str] = None
24    workout_split: Optional[str] = None
25    subscription_tier: Optional[str] = None
26    trainer_access_level: Optional[str] = None
27    past_activity: Optional[List[Dict]] = None
28 
29class ActivityData(BaseModel):
30    date: str
31    type: str
32    duration_minutes: int
33    calories_burned: int
34 
35class BulkActivityImport(BaseModel):
36    activities: List[ActivityData]
37 
38def verify_token(authorization: str = Header(...)):
39    if not authorization.startswith("Bearer "):
40        raise HTTPException(status_code=401, detail="Invalid auth header")
41    token = authorization.split(" ")[1]
42    if token not in auth_tokens:
43        raise HTTPException(status_code=401, detail="Invalid token")
44    return auth_tokens[token]
45 
46@app.post("/signup")
47def signup(user: UserSignup):
48    global user_counter
49    user_counter += 1
50    user_id = user_counter
51    users_db[user_id] = {
52        "id": user_id,
53        "username": user.username,
54        "password": user.password,
55        "email": user.email,
56        "weight": None,
57        "goals": None,
58        "workout_split": None,
59        "subscription_tier": "free",
60        "trainer_access_level": "none",
61        "past_activity": []
62    }
63    return {"id": user_id, "username": user.username}
64 
65@app.post("/login")
66def login(credentials: UserLogin):
67    for uid, user in users_db.items():
68        if user["username"] == credentials.username and user["password"] == credentials.password:
69            token = secrets.token_hex(32)
70            auth_tokens[token] = uid
71            return {"access_token": token, "token_type": "bearer"}
72    raise HTTPException(status_code=401, detail="Invalid credentials")
73 
74@app.patch("/users/{user_id}")
75def update_user(user_id: int, updates: UserPatch, authorization: str = Header(...)):
76    current_user_id = verify_token(authorization)
77    if current_user_id != user_id:
78        raise HTTPException(status_code=403, detail="Cannot modify other users")
79    if user_id not in users_db:
80        raise HTTPException(status_code=404, detail="User not found")
81    
82    user = users_db[user_id]
83    if updates.weight is not None:
84        user["weight"] = updates.weight
85    if updates.goals is not None:
86        user["goals"] = updates.goals
87    if updates.workout_split is not None:
88        user["workout_split"] = updates.workout_split
89    if updates.subscription_tier is not None:
90        user["subscription_tier"] = updates.subscription_tier
91    if updates.trainer_access_level is not None:
92        user["trainer_access_level"] = updates.trainer_access_level
93    
94    # Bulk import past activity if provided
95    if updates.past_activity is not None:
96        if "past_activity" not in user:
97            user["past_activity"] = []
98        user["past_activity"].extend(updates.past_activity)
99    
100    return user
101 
102@app.get("/users/{user_id}")
103def get_user(user_id: int, authorization: str = Header(...)):
104    current_user_id = verify_token(authorization)
105    if current_user_id != user_id:
106        raise HTTPException(status_code=403, detail="Cannot view other users")
107    if user_id not in users_db:
108        raise HTTPException(status_code=404, detail="User not found")
109    return users_db[user_id]
110 
111@app.post("/users/{user_id}/activities/bulk")
112def bulk_import_activities(user_id: int, bulk: BulkActivityImport, authorization: str = Header(...)):
113    current_user_id = verify_token(authorization)
114    if current_user_id != user_id:
115        raise HTTPException(status_code=403, detail="Cannot modify other users")
116    if user_id not in users_db:
117        raise HTTPException(status_code=404, detail="User not found")
118    
119    user = users_db[user_id]
120    if "past_activity" not in user:
121        user["past_activity"] = []
122    
123    activities_list = [act.dict() for act in bulk.activities]
124    user["past_activity"].extend(activities_list)
125    
126    return {"imported": len(activities_list), "total": len(user["past_activity"])}

1fastapi
2uvicorn
3pydantic