Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import uuid
4import datetime
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10habits = {}
11habit_logs = {}
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    if any(u["username"] == username for u in users.values()):
28        raise HTTPException(status_code=400, detail="Username already exists")
29    user_id = user_id_counter
30    user_id_counter += 1
31    users[user_id] = {"id": user_id, "username": username, "password": password}
32    return {"id": user_id, "username": username}
33 
34@app.post("/login")
35def login(username: str, password: str):
36    for user in users.values():
37        if user["username"] == username and user["password"] == password:
38            token = str(uuid.uuid4())
39            tokens[token] = user["id"]
40            return {"token": token}
41    raise HTTPException(status_code=401, detail="Invalid credentials")
42 
43@app.get("/users/{user_id}")
44def get_user(user_id: int, authorization: Optional[str] = Header(None)):
45    current_user = get_current_user(authorization)
46    if user_id not in users:
47        raise HTTPException(status_code=404, detail="User not found")
48    return users[user_id]
49 
50@app.post("/habits")
51def create_habit(name: str, description: str = "", authorization: Optional[str] = Header(None)):
52    global habit_id_counter
53    current_user = get_current_user(authorization)
54    habit_id = habit_id_counter
55    habit_id_counter += 1
56    habits[habit_id] = {"id": habit_id, "name": name, "description": description, "user_id": current_user}
57    return habits[habit_id]
58 
59@app.get("/habits/{habit_id}")
60def get_habit(habit_id: int, authorization: Optional[str] = Header(None)):
61    current_user = get_current_user(authorization)
62    if habit_id not in habits:
63        raise HTTPException(status_code=404, detail="Habit not found")
64    return habits[habit_id]
65 
66@app.post("/habit-logs")
67def create_habit_log(habit_id: int, date: str, note: str = "", authorization: Optional[str] = Header(None)):
68    global log_id_counter
69    current_user = get_current_user(authorization)
70    if habit_id not in habits:
71        raise HTTPException(status_code=404, detail="Habit not found")
72    if habits[habit_id]["user_id"] != current_user:
73        raise HTTPException(status_code=403, detail="Not your habit")
74    log_id = log_id_counter
75    log_id_counter += 1
76    habit_logs[log_id] = {"id": log_id, "habit_id": habit_id, "date": date, "note": note, "user_id": current_user}
77    return habit_logs[log_id]
78 
79@app.get("/habit-logs/{log_id}")
80def get_habit_log(log_id: int, authorization: Optional[str] = Header(None)):
81    current_user = get_current_user(authorization)
82    if log_id not in habit_logs:
83        raise HTTPException(status_code=404, detail="Log not found")
84    return habit_logs[log_id]
85 
86@app.get("/habits/{habit_id}/logs")
87def get_habit_logs(habit_id: int, authorization: Optional[str] = Header(None)):
88    current_user = get_current_user(authorization)
89    if habit_id not in habits:
90        raise HTTPException(status_code=404, detail="Habit not found")
91    if habits[habit_id]["user_id"] != current_user:
92        raise HTTPException(status_code=403, detail="Not your habit")
93    logs = [log for log in habit_logs.values() if log["habit_id"] == habit_id]
94    return logs

1fastapi
2uvicorn