Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4from datetime import datetime
5 
6app = FastAPI()
7 
8users = {}
9user_id_counter = 1
10tokens = {}
11habits = {}
12habit_id_counter = 1
13entries = {}
14entry_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 HabitCreate(BaseModel):
25    name: str
26    description: Optional[str] = None
27 
28class EntryCreate(BaseModel):
29    habit_id: int
30    note: Optional[str] = None
31 
32def get_user_id_from_token(authorization: str = Header(...)):
33    token = authorization.replace("Bearer ", "")
34    user_id = tokens.get(token)
35    if user_id is None:
36        raise HTTPException(status_code=401, detail="Invalid token")
37    return user_id
38 
39@app.post("/signup")
40def signup(req: SignupRequest):
41    global user_id_counter
42    user_id = user_id_counter
43    user_id_counter += 1
44    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
45    return {"user_id": user_id, "username": req.username}
46 
47@app.post("/login")
48def login(req: LoginRequest):
49    for uid, u in users.items():
50        if u["username"] == req.username and u["password"] == req.password:
51            token = f"tok_{uid}_{datetime.now().timestamp()}"
52            tokens[token] = uid
53            return {"token": token}
54    raise HTTPException(status_code=401, detail="Invalid credentials")
55 
56@app.post("/habits")
57def create_habit(habit: HabitCreate, authorization: str = Header(...)):
58    get_user_id_from_token(authorization)
59    global habit_id_counter
60    hid = habit_id_counter
61    habit_id_counter += 1
62    habits[hid] = {"id": hid, "name": habit.name, "description": habit.description}
63    return habits[hid]
64 
65@app.get("/habits/{habit_id}")
66def get_habit(habit_id: int, authorization: str = Header(...)):
67    get_user_id_from_token(authorization)
68    habit = habits.get(habit_id)
69    if not habit:
70        raise HTTPException(status_code=404, detail="Habit not found")
71    return habit
72 
73@app.post("/entries")
74def create_entry(entry: EntryCreate, authorization: str = Header(...)):
75    get_user_id_from_token(authorization)
76    if entry.habit_id not in habits:
77        raise HTTPException(status_code=404, detail="Habit not found")
78    global entry_id_counter
79    eid = entry_id_counter
80    entry_id_counter += 1
81    entries[eid] = {"id": eid, "habit_id": entry.habit_id, "note": entry.note, "created_at": datetime.now().isoformat()}
82    return entries[eid]
83 
84@app.get("/entries/{entry_id}")
85def get_entry(entry_id: int, authorization: str = Header(...)):
86    get_user_id_from_token(authorization)
87    entry = entries.get(entry_id)
88    if not entry:
89        raise HTTPException(status_code=404, detail="Entry not found")
90    return entry

1fastapi
2uvicorn