Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import uuid
4from datetime import date, timedelta
5 
6app = FastAPI()
7 
8users = {}
9habits = {}
10habit_logs = {}
11tokens = {}
12user_id_counter = 1
13habit_id_counter = 1
14log_id_counter = 1
15 
16def get_current_user(authorization: Optional[str] = Header(None)):
17    if not authorization:
18        raise HTTPException(status_code=401, detail="Missing Authorization header")
19    token = authorization.replace("Bearer ", "")
20    if token not in tokens:
21        raise HTTPException(status_code=401, detail="Invalid token")
22    return tokens[token]
23 
24@app.post("/signup")
25def signup(username: str, password: str):
26    global user_id_counter
27    for u in users.values():
28        if u["username"] == username:
29            raise HTTPException(status_code=400, detail="Username already exists")
30    user_id = user_id_counter
31    user_id_counter += 1
32    users[user_id] = {"id": user_id, "username": username, "password": password}
33    return {"user_id": user_id, "username": username}
34 
35@app.post("/login")
36def login(username: str, password: str):
37    for u in users.values():
38        if u["username"] == username and u["password"] == password:
39            token = str(uuid.uuid4())
40            tokens[token] = u["id"]
41            return {"token": token}
42    raise HTTPException(status_code=401, detail="Invalid credentials")
43 
44@app.get("/users/{user_id}")
45def get_user(user_id: int, authorization: Optional[str] = Header(None)):
46    get_current_user(authorization)
47    if user_id not in users:
48        raise HTTPException(status_code=404, detail="User not found")
49    return users[user_id]
50 
51@app.post("/habits")
52def create_habit(name: str, frequency: int, authorization: Optional[str] = Header(None)):
53    user_id = get_current_user(authorization)
54    global habit_id_counter
55    habit_id = habit_id_counter
56    habit_id_counter += 1
57    habits[habit_id] = {
58        "id": habit_id,
59        "user_id": user_id,
60        "name": name,
61        "frequency": frequency,
62        "created_at": str(date.today())
63    }
64    return habits[habit_id]
65 
66@app.get("/habits/{habit_id}")
67def get_habit(habit_id: int, authorization: Optional[str] = Header(None)):
68    get_current_user(authorization)
69    if habit_id not in habits:
70        raise HTTPException(status_code=404, detail="Habit not found")
71    return habits[habit_id]
72 
73@app.post("/habit_logs")
74def log_habit(habit_id: int, completion_date: str, authorization: Optional[str] = Header(None)):
75    user_id = get_current_user(authorization)
76    if habit_id not in habits:
77        raise HTTPException(status_code=404, detail="Habit not found")
78    if habits[habit_id]["user_id"] != user_id:
79        raise HTTPException(status_code=403, detail="Not your habit")
80    global log_id_counter
81    log_id = log_id_counter
82    log_id_counter += 1
83    habit_logs[log_id] = {
84        "id": log_id,
85        "habit_id": habit_id,
86        "user_id": user_id,
87        "completion_date": completion_date
88    }
89    return habit_logs[log_id]
90 
91@app.get("/habit_logs/{log_id}")
92def get_habit_log(log_id: int, authorization: Optional[str] = Header(None)):
93    get_current_user(authorization)
94    if log_id not in habit_logs:
95        raise HTTPException(status_code=404, detail="Log not found")
96    return habit_logs[log_id]
97 
98@app.get("/habits/{habit_id}/streak")
99def get_streak(habit_id: int, authorization: Optional[str] = Header(None)):
100    user_id = get_current_user(authorization)
101    if habit_id not in habits:
102        raise HTTPException(status_code=404, detail="Habit not found")
103    if habits[habit_id]["user_id"] != user_id:
104        raise HTTPException(status_code=403, detail="Not your habit")
105    
106    logs = [l for l in habit_logs.values() if l["habit_id"] == habit_id]
107    dates = sorted(set(l["completion_date"] for l in logs), reverse=True)
108    
109    if not dates:
110        return {"streak": 0, "consecutive_days": []}
111    
112    today = str(date.today())
113    if dates[0] != today:
114        return {"streak": 0, "consecutive_days": []}
115    
116    streak = 1
117    consecutive = [dates[0]]
118    for i in range(1, len(dates)):
119        prev_date = date.fromisoformat(dates[i-1])
120        curr_date = date.fromisoformat(dates[i])
121        if prev_date - curr_date == timedelta(days=1):
122            streak += 1
123            consecutive.append(dates[i])
124        else:
125            break
126    
127    return {"streak": streak, "consecutive_days": consecutive}

1fastapi
2uvicorn