Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from datetime import date, datetime
3from typing import Optional
4import hashlib
5import secrets
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11habits = {}
12habit_logs = {}
13next_user_id = 1
14next_habit_id = 1
15 
16def get_current_user(authorization: Optional[str] = Header(None)):
17    if not authorization:
18        raise HTTPException(status_code=401, detail="Missing auth token")
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 next_user_id
27    for u in users.values():
28        if u["username"] == username:
29            raise HTTPException(status_code=400, detail="User exists")
30    user_id = next_user_id
31    next_user_id += 1
32    users[user_id] = {
33        "id": user_id,
34        "username": username,
35        "password_hash": hashlib.sha256(password.encode()).hexdigest()
36    }
37    return {"id": user_id, "username": username}
38 
39@app.post("/login")
40def login(username: str, password: str):
41    for u in users.values():
42        if u["username"] == username and u["password_hash"] == hashlib.sha256(password.encode()).hexdigest():
43            token = secrets.token_hex(32)
44            tokens[token] = u["id"]
45            return {"token": token}
46    raise HTTPException(status_code=401, detail="Invalid credentials")
47 
48@app.get("/users/{user_id}")
49def get_user(user_id: int, authorization: Optional[str] = Header(None)):
50    get_current_user(authorization)
51    if user_id not in users:
52        raise HTTPException(status_code=404, detail="User not found")
53    return users[user_id]
54 
55@app.post("/habits")
56def create_habit(name: str, frequency: str, goal: int, authorization: Optional[str] = Header(None)):
57    global next_habit_id
58    user_id = get_current_user(authorization)
59    habit_id = next_habit_id
60    next_habit_id += 1
61    habits[habit_id] = {
62        "id": habit_id,
63        "user_id": user_id,
64        "name": name,
65        "frequency": frequency,
66        "goal": goal,
67        "created_at": datetime.utcnow().isoformat()
68    }
69    habit_logs[habit_id] = []
70    return habits[habit_id]
71 
72@app.post("/habits/{habit_id}/check")
73def check_habit(habit_id: int, authorization: Optional[str] = Header(None)):
74    user_id = get_current_user(authorization)
75    if habit_id not in habits:
76        raise HTTPException(status_code=404, detail="Habit not found")
77    if habits[habit_id]["user_id"] != user_id:
78        raise HTTPException(status_code=403, detail="Not your habit")
79    today = date.today().isoformat()
80    logs = habit_logs[habit_id]
81    for log in logs:
82        if log["date"] == today:
83            return {"message": "Already checked today", "habit_id": habit_id}
84    logs.append({"date": today, "timestamp": datetime.utcnow().isoformat()})
85    return {"message": "Logged", "habit_id": habit_id}
86 
87@app.get("/habits/{habit_id}")
88def get_habit(habit_id: int, authorization: Optional[str] = Header(None)):
89    user_id = get_current_user(authorization)
90    if habit_id not in habits:
91        raise HTTPException(status_code=404, detail="Habit not found")
92    if habits[habit_id]["user_id"] != user_id:
93        raise HTTPException(status_code=403, detail="Not your habit")
94    
95    habit = habits[habit_id]
96    logs = habit_logs[habit_id]
97    dates = sorted([log["date"] for log in logs], reverse=True)
98    
99    # Calculate current streak
100    streak = 0
101    check_date = date.today()
102    if habit["frequency"] == "daily":
103        while True:
104            if check_date.isoformat() in dates:
105                streak += 1
106                check_date -= __import__('datetime').timedelta(days=1)
107            else:
108                break
109    elif habit["frequency"] == "weekly":
110        # Check current week
111        today = date.today()
112        start_of_week = today - __import__('datetime').timedelta(days=today.weekday())
113        while True:
114            week_start = start_of_week - __import__('datetime').timedelta(weeks=streak)
115            week_end = week_start + __import__('datetime').timedelta(days=6)
116            checked_this_week = any(
117                week_start.isoformat() <= d <= week_end.isoformat() for d in dates
118            )
119            if checked_this_week:
120                streak += 1
121            else:
122                break
123    
124    # Calculate consistency percentage
125    total_days = (date.today() - __import__('datetime').datetime.fromisoformat(habit["created_at"]).date()).days + 1
126    if habit["frequency"] == "daily":
127        consistency = (len(logs) / total_days) * 100 if total_days > 0 else 0
128    else:
129        total_weeks = max(1, total_days // 7)
130        consistency = (len(logs) / total_weeks) * 100 if total_weeks > 0 else 0
131    
132    return {
133        **habit,
134        "current_streak": streak,
135        "consistency_percentage": round(consistency, 2)
136    }
137 
138@app.get("/habits")
139def get_habits(authorization: Optional[str] = Header(None)):
140    user_id = get_current_user(authorization)
141    user_habits = []
142    for h in habits.values():
143        if h["user_id"] == user_id:
144            logs = habit_logs[h["id"]]
145            dates = sorted([log["date"] for log in logs], reverse=True)
146            
147            streak = 0
148            check_date = date.today()
149            if h["frequency"] == "daily":
150                while True:
151                    if check_date.isoformat() in dates:
152                        streak += 1
153                        check_date -= __import__('datetime').timedelta(days=1)
154                    else:
155                        break
156            elif h["frequency"] == "weekly":
157                today = date.today()
158                start_of_week = today - __import__('datetime').timedelta(days=today.weekday())
159                while True:
160                    week_start = start_of_week - __import__('datetime').timedelta(weeks=streak)
161                    week_end = week_start + __import__('datetime').timedelta(days=6)
162                    checked_this_week = any(
163                        week_start.isoformat() <= d <= week_end.isoformat() for d in dates
164                    )
165                    if checked_this_week:
166                        streak += 1
167                    else:
168                        break
169            
170            total_days = (date.today() - __import__('datetime').datetime.fromisoformat(h["created_at"]).date()).days + 1
171            if h["frequency"] == "daily":
172                consistency = (len(logs) / total_days) * 100 if total_days > 0 else 0
173            else:
174                total_weeks = max(1, total_days // 7)
175                consistency = (len(logs) / total_weeks) * 100 if total_weeks > 0 else 0
176            
177            user_habits.append({
178                **h,
179                "current_streak": streak,
180                "consistency_percentage": round(consistency, 2)
181            })
182    return user_habits

1fastapi
2uvicorn