Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10habits = {}
11habit_logs = {}
12user_id_counter = 1
13habit_id_counter = 1
14log_id_counter = 1
15 
16class SignupRequest(BaseModel):
17    username: str
18    password: str
19    name: str
20    goals: str = ""
21    daily_reminders: bool = False
22    subscription_tier: str = "free"
23    role: str = "user"
24 
25class LoginRequest(BaseModel):
26    username: str
27    password: str
28 
29class UserUpdate(BaseModel):
30    name: Optional[str] = None
31    goals: Optional[str] = None
32    daily_reminders: Optional[bool] = None
33    subscription_tier: Optional[str] = None
34    role: Optional[str] = None
35 
36class HabitCreate(BaseModel):
37    name: str
38    description: str = ""
39 
40class HabitLogCreate(BaseModel):
41    habit_id: int
42    date: str
43    completed: bool = False
44 
45def get_current_user(authorization: str = Header(...)):
46    if not authorization.startswith("Bearer "):
47        raise HTTPException(status_code=401, detail="Invalid auth header")
48    token = authorization.split(" ")[1]
49    if token not in tokens:
50        raise HTTPException(status_code=401, detail="Invalid token")
51    return tokens[token]
52 
53@app.post("/signup")
54def signup(req: SignupRequest):
55    global user_id_counter
56    user_id = user_id_counter
57    user_id_counter += 1
58    users[user_id] = {
59        "id": user_id,
60        "username": req.username,
61        "password": req.password,
62        "name": req.name,
63        "goals": req.goals,
64        "daily_reminders": req.daily_reminders,
65        "subscription_tier": req.subscription_tier,
66        "role": req.role
67    }
68    return {"id": user_id, "message": "User created"}
69 
70@app.post("/login")
71def login(req: LoginRequest):
72    for uid, user in users.items():
73        if user["username"] == req.username and user["password"] == req.password:
74            token = secrets.token_hex(16)
75            tokens[token] = uid
76            return {"token": token, "user_id": uid}
77    raise HTTPException(status_code=401, detail="Invalid credentials")
78 
79@app.get("/users/{user_id}")
80def get_user(user_id: int, authorization: str = Header(...)):
81    current_user = get_current_user(authorization)
82    if user_id not in users:
83        raise HTTPException(status_code=404, detail="User not found")
84    return users[user_id]
85 
86@app.put("/users/{user_id}")
87def update_user(user_id: int, update: UserUpdate, authorization: str = Header(...)):
88    current_user = get_current_user(authorization)
89    if user_id not in users:
90        raise HTTPException(status_code=404, detail="User not found")
91    user = users[user_id]
92    if update.name is not None:
93        user["name"] = update.name
94    if update.goals is not None:
95        user["goals"] = update.goals
96    if update.daily_reminders is not None:
97        user["daily_reminders"] = update.daily_reminders
98    if update.subscription_tier is not None:
99        user["subscription_tier"] = update.subscription_tier
100    if update.role is not None:
101        user["role"] = update.role
102    return user
103 
104@app.post("/habits")
105def create_habit(habit: HabitCreate, authorization: str = Header(...)):
106    global habit_id_counter
107    current_user = get_current_user(authorization)
108    hid = habit_id_counter
109    habit_id_counter += 1
110    habits[hid] = {
111        "id": hid,
112        "user_id": current_user,
113        "name": habit.name,
114        "description": habit.description
115    }
116    return habits[hid]
117 
118@app.get("/habits/{habit_id}")
119def get_habit(habit_id: int, authorization: str = Header(...)):
120    current_user = get_current_user(authorization)
121    if habit_id not in habits:
122        raise HTTPException(status_code=404, detail="Habit not found")
123    return habits[habit_id]
124 
125@app.post("/habit-logs")
126def create_habit_log(log: HabitLogCreate, authorization: str = Header(...)):
127    global log_id_counter
128    current_user = get_current_user(authorization)
129    if log.habit_id not in habits:
130        raise HTTPException(status_code=404, detail="Habit not found")
131    lid = log_id_counter
132    log_id_counter += 1
133    habit_logs[lid] = {
134        "id": lid,
135        "user_id": current_user,
136        "habit_id": log.habit_id,
137        "date": log.date,
138        "completed": log.completed
139    }
140    return habit_logs[lid]
141 
142@app.get("/habit-logs/{log_id}")
143def get_habit_log(log_id: int, authorization: str = Header(...)):
144    current_user = get_current_user(authorization)
145    if log_id not in habit_logs:
146        raise HTTPException(status_code=404, detail="Log not found")
147    return habit_logs[log_id]

1fastapi
2uvicorn