Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import hashlib
4import secrets
5from datetime import datetime
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11trips = {}
12items = {}
13trip_counter = 0
14item_counter = 0
15user_counter = 0
16 
17CATEGORIES = {
18    "clothing": 1,
19    "toiletries": 2,
20    "electronics": 3,
21    "documents": 4,
22    "health": 5,
23    "misc": 6
24}
25 
26@app.post("/signup")
27def signup(username: str, password: str):
28    global user_counter
29    if username in users:
30        raise HTTPException(400, "User exists")
31    user_counter += 1
32    users[username] = {"id": user_counter, "password": hashlib.sha256(password.encode()).hexdigest()}
33    token = secrets.token_hex(16)
34    tokens[token] = username
35    return {"user_id": user_counter, "token": token}
36 
37@app.post("/login")
38def login(username: str, password: str):
39    if username not in users:
40        raise HTTPException(400, "Invalid credentials")
41    if users[username]["password"] != hashlib.sha256(password.encode()).hexdigest():
42        raise HTTPException(400, "Invalid credentials")
43    token = secrets.token_hex(16)
44    tokens[token] = username
45    return {"token": token}
46 
47def get_user(token: str = Header(...)):
48    if token not in tokens:
49        raise HTTPException(401, "Invalid token")
50    return tokens[token]
51 
52@app.post("/trips")
53def create_trip(destination: str, start_date: str, end_date: str, token: str = Header(...)):
54    global trip_counter
55    user = get_user(token)
56    trip_counter += 1
57    trips[trip_counter] = {
58        "id": trip_counter,
59        "user": user,
60        "destination": destination,
61        "start_date": start_date,
62        "end_date": end_date,
63        "items": []
64    }
65    return trips[trip_counter]
66 
67@app.get("/trips/{trip_id}")
68def get_trip(trip_id: int, token: str = Header(...)):
69    user = get_user(token)
70    if trip_id not in trips:
71        raise HTTPException(404, "Trip not found")
72    return trips[trip_id]
73 
74@app.post("/items")
75def create_item(trip_id: int, name: str, quantity: int = 1, category: str = "misc", token: str = Header(...)):
76    global item_counter
77    user = get_user(token)
78    if trip_id not in trips:
79        raise HTTPException(404, "Trip not found")
80    if trips[trip_id]["user"] != user:
81        raise HTTPException(403, "Not your trip")
82    if category not in CATEGORIES:
83        raise HTTPException(400, "Invalid category")
84    item_counter += 1
85    items[item_counter] = {
86        "id": item_counter,
87        "trip_id": trip_id,
88        "name": name,
89        "quantity": quantity,
90        "category": category
91    }
92    trips[trip_id]["items"].append(item_counter)
93    return items[item_counter]
94 
95@app.get("/items/{item_id}")
96def get_item(item_id: int, token: str = Header(...)):
97    user = get_user(token)
98    if item_id not in items:
99        raise HTTPException(404, "Item not found")
100    return items[item_id]
101 
102@app.get("/trips/{trip_id}/checklist")
103def get_checklist(trip_id: int, token: str = Header(...)):
104    user = get_user(token)
105    if trip_id not in trips:
106        raise HTTPException(404, "Trip not found")
107    if trips[trip_id]["user"] != user:
108        raise HTTPException(403, "Not your trip")
109    trip_items = [items[i] for i in trips[trip_id]["items"]]
110    trip_items.sort(key=lambda x: CATEGORIES.get(x["category"], 99))
111    return trip_items

1fastapi
2uvicorn