Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict, List
4import secrets
5from datetime import date
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11itineraries = {}
12itinerary_id_counter = 1
13 
14class UserSignup(BaseModel):
15    username: str
16    password: str
17 
18class UserLogin(BaseModel):
19    username: str
20    password: str
21 
22class Destination(BaseModel):
23    location: str
24    date: date
25 
26class ItineraryCreate(BaseModel):
27    destinations: List[Destination]
28 
29class DayNote(BaseModel):
30    day: int
31    note: str
32 
33class DayPhoto(BaseModel):
34    day: int
35    photo_url: str
36 
37def get_current_user(authorization: str = Header(...)):
38    if not authorization.startswith("Bearer "):
39        raise HTTPException(status_code=401, detail="Invalid auth header")
40    token = authorization.split(" ")[1]
41    if token not in tokens:
42        raise HTTPException(status_code=401, detail="Invalid token")
43    return tokens[token]
44 
45@app.post("/signup")
46def signup(user: UserSignup):
47    if user.username in users:
48        raise HTTPException(status_code=400, detail="User already exists")
49    users[user.username] = {"password": user.password}
50    token = secrets.token_hex(16)
51    tokens[token] = user.username
52    return {"token": token}
53 
54@app.post("/login")
55def login(user: UserLogin):
56    if user.username not in users or users[user.username]["password"] != user.password:
57        raise HTTPException(status_code=401, detail="Invalid credentials")
58    token = secrets.token_hex(16)
59    tokens[token] = user.username
60    return {"token": token}
61 
62@app.post("/itinerary")
63def create_itinerary(itinerary: ItineraryCreate, authorization: str = Header(...)):
64    global itinerary_id_counter
65    user = get_current_user(authorization)
66    new_id = itinerary_id_counter
67    itinerary_id_counter += 1
68    itineraries[new_id] = {
69        "id": new_id,
70        "user": user,
71        "destinations": [d.dict() for d in itinerary.destinations],
72        "notes": {},
73        "photos": {}
74    }
75    return {"id": new_id}
76 
77@app.get("/itinerary/{itinerary_id}")
78def get_itinerary(itinerary_id: int, authorization: str = Header(...)):
79    user = get_current_user(authorization)
80    if itinerary_id not in itineraries:
81        raise HTTPException(status_code=404, detail="Itinerary not found")
82    return itineraries[itinerary_id]
83 
84@app.post("/itinerary/{itinerary_id}/note")
85def add_note(itinerary_id: int, day_note: DayNote, authorization: str = Header(...)):
86    user = get_current_user(authorization)
87    if itinerary_id not in itineraries:
88        raise HTTPException(status_code=404, detail="Itinerary not found")
89    day_key = f"day_{day_note.day}"
90    if day_key not in itineraries[itinerary_id]["notes"]:
91        itineraries[itinerary_id]["notes"][day_key] = []
92    itineraries[itinerary_id]["notes"][day_key].append(day_note.note)
93    return {"status": "ok"}
94 
95@app.post("/itinerary/{itinerary_id}/photo")
96def add_photo(itinerary_id: int, day_photo: DayPhoto, authorization: str = Header(...)):
97    user = get_current_user(authorization)
98    if itinerary_id not in itineraries:
99        raise HTTPException(status_code=404, detail="Itinerary not found")
100    day_key = f"day_{day_photo.day}"
101    if day_key not in itineraries[itinerary_id]["photos"]:
102        itineraries[itinerary_id]["photos"][day_key] = []
103    itineraries[itinerary_id]["photos"][day_key].append(day_photo.photo_url)
104    return {"status": "ok"}

1fastapi
2uvicorn