Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4from datetime import datetime, date, timedelta
5import uuid
6import math
7 
8app = FastAPI()
9 
10users = {}
11habits = {}
12habit_logs = {}
13tokens = {}
14user_id_counter = 1
15habit_id_counter = 1
16log_id_counter = 1
17 
18def get_current_user(authorization: str = Header(None)):
19    if not authorization:
20        raise HTTPException(status_code=401, detail="Missing auth header")
21    token = authorization.replace("Bearer ", "")
22    user_id = tokens.get(token)
23    if not user_id:
24        raise HTTPException(status_code=401, detail="Invalid token")
25    return user_id
26 
27class SignupRequest(BaseModel):
28    username: str
29    password: str
30 
31class LoginRequest(BaseModel):
32    username: str
33    password: str
34 
35class CreateHabitRequest(BaseModel):
36    name: str
37    description: Optional[str] = ""
38 
39class LogHabitRequest(BaseModel):
40    habit_id: int
41    notes: Optional[str] = ""
42    mood_rating: Optional[int] = 3
43 
44@app.post("/signup")
45def signup(req: SignupRequest):
46    global user_id_counter
47    for u in users.values():
48        if u["username"] == req.username:
49            raise HTTPException(status_code=400, detail="Username taken")
50    uid = user_id_counter
51    user_id_counter += 1
52    users[uid] = {"id": uid, "username": req.username, "password": req.password}
53    return {"id": uid, "username": req.username}
54 
55@app.post("/login")
56def login(req: LoginRequest):
57    for u in users.values():
58        if u["username"] == req.username and u["password"] == req.password:
59            token = str(uuid.uuid4())
60            tokens[token] = u["id"]
61            return {"token": token}
62    raise HTTPException(status_code=401, detail="Invalid credentials")
63 
64@app.get("/habits/{habit_id}")
65def get_habit(habit_id: int, authorization: str = Header(None)):
66    user_id = get_current_user(authorization)
67    habit = habits.get(habit_id)
68    if not habit:
69        raise HTTPException(status_code=404, detail="Habit not found")
70    return habit
71 
72@app.post("/habits")
73def create_habit(req: CreateHabitRequest, authorization: str = Header(None)):
74    global habit_id_counter
75    user_id = get_current_user(authorization)
76    hid = habit_id_counter
77    habit_id_counter += 1
78    habits[hid] = {
79        "id": hid,
80        "user_id": user_id,
81        "name": req.name,
82        "description": req.description,
83        "created_at": datetime.utcnow().isoformat()
84    }
85    return habits[hid]
86 
87@app.get("/habits/{habit_id}/streak")
88def get_streak(habit_id: int, authorization: str = Header(None)):
89    user_id = get_current_user(authorization)
90    habit = habits.get(habit_id)
91    if not habit:
92        raise HTTPException(status_code=404, detail="Habit not found")
93    if habit["user_id"] != user_id:
94        raise HTTPException(status_code=403, detail="Not your habit")
95 
96    logs = [l for l in habit_logs.values() if l["habit_id"] == habit_id]
97    logs.sort(key=lambda x: x["logged_date"], reverse=True)
98 
99    if not logs:
100        return {
101            "current_streak": 0,
102            "history_graph": "",
103            "next_milestone": 1
104        }
105 
106    today = date.today()
107    current_streak = 0
108    check_date = today
109    log_dates = set()
110    for l in logs:
111        log_dates.add(l["logged_date"])
112 
113    while check_date in log_dates:
114        current_streak += 1
115        check_date -= timedelta(days=1)
116 
117    history = []
118    for i in range(30):
119        d = today - timedelta(days=i)
120        history.append("X" if d in log_dates else ".")
121 
122    history_graph = "".join(reversed(history))
123 
124    milestones = [1, 3, 7, 14, 21, 30, 60, 90, 180, 365]
125    next_milestone = 1
126    for m in milestones:
127        if current_streak < m:
128            next_milestone = m
129            break
130 
131    return {
132        "current_streak": current_streak,
133        "history_graph": history_graph,
134        "next_milestone": next_milestone
135    }
136 
137@app.post("/habits/log")
138def log_habit(req: LogHabitRequest, authorization: str = Header(None)):
139    global log_id_counter
140    user_id = get_current_user(authorization)
141    habit = habits.get(req.habit_id)
142    if not habit:
143        raise HTTPException(status_code=404, detail="Habit not found")
144    if habit["user_id"] != user_id:
145        raise HTTPException(status_code=403, detail="Not your habit")
146 
147    lid = log_id_counter
148    log_id_counter += 1
149    habit_logs[lid] = {
150        "id": lid,
151        "habit_id": req.habit_id,
152        "user_id": user_id,
153        "notes": req.notes,
154        "mood_rating": req.mood_rating,
155        "logged_date": date.today().isoformat(),
156        "logged_at": datetime.utcnow().isoformat()
157    }
158    return habit_logs[lid]

1fastapi
2uvicorn