Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import uuid
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10itineraries = {}
11itinerary_id_counter = 1
12 
13class SignupRequest(BaseModel):
14    username: str
15    password: str
16 
17class LoginRequest(BaseModel):
18    username: str
19    password: str
20 
21class ItineraryCreate(BaseModel):
22    destination: str
23    dates: str
24    activities: list[str] = []
25    tier: Optional[str] = None
26    is_shared: Optional[bool] = False
27 
28class ItineraryUpdate(BaseModel):
29    destination: Optional[str] = None
30    dates: Optional[str] = None
31    activities: Optional[list[str]] = None
32    tier: Optional[str] = None
33    is_shared: Optional[bool] = None
34 
35def get_current_user(authorization: str = Header(None)):
36    if not authorization:
37        raise HTTPException(status_code=401, detail="Missing Authorization header")
38    token = authorization.replace("Bearer ", "")
39    if token not in tokens:
40        raise HTTPException(status_code=401, detail="Invalid token")
41    return tokens[token]
42 
43@app.post("/signup")
44def signup(req: SignupRequest):
45    if req.username in users:
46        raise HTTPException(status_code=400, detail="Username already exists")
47    users[req.username] = {"password": req.password}
48    return {"message": "User created"}
49 
50@app.post("/login")
51def login(req: LoginRequest):
52    user = users.get(req.username)
53    if not user or user["password"] != req.password:
54        raise HTTPException(status_code=401, detail="Invalid credentials")
55    token = str(uuid.uuid4())
56    tokens[token] = req.username
57    return {"token": token}
58 
59@app.get("/itineraries/{itinerary_id}")
60def get_itinerary(itinerary_id: int, authorization: str = Header(None)):
61    get_current_user(authorization)
62    itinerary = itineraries.get(itinerary_id)
63    if not itinerary:
64        raise HTTPException(status_code=404, detail="Itinerary not found")
65    return itinerary
66 
67@app.post("/itineraries")
68def create_itinerary(req: ItineraryCreate, authorization: str = Header(None)):
69    get_current_user(authorization)
70    global itinerary_id_counter
71    itinerary = {
72        "id": itinerary_id_counter,
73        "destination": req.destination,
74        "dates": req.dates,
75        "activities": req.activities,
76        "tier": req.tier,
77        "is_shared": req.is_shared
78    }
79    itineraries[itinerary_id_counter] = itinerary
80    itinerary_id_counter += 1
81    return itinerary
82 
83@app.put("/itineraries/{itinerary_id}")
84def update_itinerary(itinerary_id: int, req: ItineraryUpdate, authorization: str = Header(None)):
85    get_current_user(authorization)
86    itinerary = itineraries.get(itinerary_id)
87    if not itinerary:
88        raise HTTPException(status_code=404, detail="Itinerary not found")
89    if req.destination is not None:
90        itinerary["destination"] = req.destination
91    if req.dates is not None:
92        itinerary["dates"] = req.dates
93    if req.activities is not None:
94        itinerary["activities"] = req.activities
95    if req.tier is not None:
96        itinerary["tier"] = req.tier
97    if req.is_shared is not None:
98        itinerary["is_shared"] = req.is_shared
99    return itinerary

1fastapi
2uvicorn