Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import datetime
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11goals = {}
12progress_logs = {}
13goal_id_counter = 1
14progress_id_counter = 1
15 
16class SignupRequest(BaseModel):
17    username: str
18    password: str
19 
20class LoginRequest(BaseModel):
21    username: str
22    password: str
23 
24class GoalCreate(BaseModel):
25    target_value: float
26    deadline: str
27    title: str = ""
28 
29class ProgressLog(BaseModel):
30    goal_id: int
31    measurement: float
32 
33def get_user_id_from_token(authorization: str = Header(...)):
34    token = authorization.replace("Bearer ", "")
35    user_id = tokens.get(token)
36    if not user_id:
37        raise HTTPException(status_code=401, detail="Invalid token")
38    return user_id
39 
40@app.post("/signup")
41def signup(req: SignupRequest):
42    if req.username in users:
43        raise HTTPException(status_code=400, detail="Username already exists")
44    user_id = len(users) + 1
45    users[req.username] = {"id": user_id, "password": req.password, "username": req.username}
46    token = secrets.token_hex(16)
47    tokens[token] = user_id
48    return {"user_id": user_id, "token": token}
49 
50@app.post("/login")
51def login(req: LoginRequest):
52    user = users.get(req.username)
53    if not user or user["password"] != req.password:
54        raise HTTPException(status_code=401, detail="Invalid credentials")
55    token = secrets.token_hex(16)
56    tokens[token] = user["id"]
57    return {"user_id": user["id"], "token": token}
58 
59@app.get("/goals/{goal_id}")
60def get_goal(goal_id: int, authorization: str = Header(...)):
61    user_id = get_user_id_from_token(authorization)
62    goal = goals.get(goal_id)
63    if not goal:
64        raise HTTPException(status_code=404, detail="Goal not found")
65    return goal
66 
67@app.get("/goals")
68def list_goals(authorization: str = Header(...)):
69    user_id = get_user_id_from_token(authorization)
70    user_goals = [g for g in goals.values() if g["user_id"] == user_id]
71    return user_goals
72 
73@app.post("/goals")
74def create_goal(goal: GoalCreate, authorization: str = Header(...)):
75    global goal_id_counter
76    user_id = get_user_id_from_token(authorization)
77    new_goal = {
78        "id": goal_id_counter,
79        "user_id": user_id,
80        "target_value": goal.target_value,
81        "deadline": goal.deadline,
82        "title": goal.title,
83        "current_progress": 0.0,
84        "completion_percentage": 0.0,
85        "streak_days": 0,
86        "last_logged_date": None
87    }
88    goals[goal_id_counter] = new_goal
89    goal_id_counter += 1
90    return new_goal
91 
92@app.post("/goals/progress")
93def log_progress(progress: ProgressLog, authorization: str = Header(...)):
94    global progress_id_counter
95    user_id = get_user_id_from_token(authorization)
96    goal = goals.get(progress.goal_id)
97    if not goal:
98        raise HTTPException(status_code=404, detail="Goal not found")
99    
100    today = datetime.date.today().isoformat()
101    
102    log_entry = {
103        "id": progress_id_counter,
104        "goal_id": progress.goal_id,
105        "measurement": progress.measurement,
106        "logged_at": today
107    }
108    progress_logs[progress_id_counter] = log_entry
109    progress_id_counter += 1
110    
111    goal["current_progress"] += progress.measurement
112    if goal["target_value"] > 0:
113        goal["completion_percentage"] = min(100.0, (goal["current_progress"] / goal["target_value"]) * 100)
114    else:
115        goal["completion_percentage"] = 0.0
116    
117    if goal["last_logged_date"] == today:
118        pass
119    elif goal["last_logged_date"] == (datetime.date.today() - datetime.timedelta(days=1)).isoformat():
120        goal["streak_days"] += 1
121    else:
122        goal["streak_days"] = 1
123    
124    goal["last_logged_date"] = today
125    
126    return {"message": "Progress logged", "goal": goal}

1fastapi
2uvicorn