Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3from pydantic import BaseModel
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10habits = {}
11habit_entries = {}
12next_user_id = 1
13next_habit_id = 1
14next_entry_id = 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] = ""
27 
28class HabitEntryCreate(BaseModel):
29    habit_id: int
30    note: Optional[str] = ""
31 
32def get_user_from_token(authorization: Optional[str] = Header(None)):
33    if not authorization:
34        raise HTTPException(status_code=401, detail="Missing Authorization header")
35    token = authorization.replace("Bearer ", "")
36    user_id = tokens.get(token)
37    if user_id is None:
38        raise HTTPException(status_code=401, detail="Invalid token")
39    return user_id
40 
41@app.post("/signup")
42def signup(req: SignupRequest):
43    global next_user_id
44    for u in users.values():
45        if u["username"] == req.username:
46            raise HTTPException(status_code=400, detail="Username already exists")
47    user_id = next_user_id
48    next_user_id += 1
49    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
50    return {"id": user_id, "username": req.username}
51 
52@app.post("/login")
53def login(req: LoginRequest):
54    for u in users.values():
55        if u["username"] == req.username and u["password"] == req.password:
56            token = secrets.token_hex(16)
57            tokens[token] = u["id"]
58            return {"token": token}
59    raise HTTPException(status_code=401, detail="Invalid credentials")
60 
61@app.post("/habits")
62def create_habit(habit: HabitCreate, authorization: Optional[str] = Header(None)):
63    user_id = get_user_from_token(authorization)
64    global next_habit_id
65    habit_id = next_habit_id
66    next_habit_id += 1
67    habits[habit_id] = {"id": habit_id, "name": habit.name, "description": habit.description, "user_id": user_id}
68    return habits[habit_id]
69 
70@app.get("/habits/{habit_id}")
71def get_habit(habit_id: int, authorization: Optional[str] = Header(None)):
72    get_user_from_token(authorization)
73    if habit_id not in habits:
74        raise HTTPException(status_code=404, detail="Habit not found")
75    return habits[habit_id]
76 
77@app.post("/habit-entries")
78def create_habit_entry(entry: HabitEntryCreate, authorization: Optional[str] = Header(None)):
79    user_id = get_user_from_token(authorization)
80    if entry.habit_id not in habits:
81        raise HTTPException(status_code=404, detail="Habit not found")
82    if habits[entry.habit_id]["user_id"] != user_id:
83        raise HTTPException(status_code=403, detail="This habit does not belong to you")
84    global next_entry_id
85    entry_id = next_entry_id
86    next_entry_id += 1
87    habit_entries[entry_id] = {"id": entry_id, "habit_id": entry.habit_id, "note": entry.note, "user_id": user_id}
88    return habit_entries[entry_id]
89 
90@app.get("/habit-entries/{entry_id}")
91def get_habit_entry(entry_id: int, authorization: Optional[str] = Header(None)):
92    get_user_from_token(authorization)
93    if entry_id not in habit_entries:
94        raise HTTPException(status_code=404, detail="Entry not found")
95    return habit_entries[entry_id]

1fastapi
2uvicorn