Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import uuid
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10trips = {}
11trip_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 TripCreate(BaseModel):
22    destination: str
23    start_date: str
24    end_date: str
25    preferences: Optional[str] = None
26 
27class TripUpdate(BaseModel):
28    destination: Optional[str] = None
29    start_date: Optional[str] = None
30    end_date: Optional[str] = None
31    preferences: Optional[str] = None
32    tier: Optional[str] = None
33    status: Optional[str] = None
34 
35def get_user_id_from_token(authorization: str = Header(None)):
36    if not authorization:
37        raise HTTPException(status_code=401, detail="Missing auth token")
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="User already exists")
47    users[req.username] = {"password": req.password}
48    token = str(uuid.uuid4())
49    tokens[token] = req.username
50    return {"token": token}
51 
52@app.post("/login")
53def login(req: LoginRequest):
54    if req.username not in users or users[req.username]["password"] != req.password:
55        raise HTTPException(status_code=401, detail="Invalid credentials")
56    token = str(uuid.uuid4())
57    tokens[token] = req.username
58    return {"token": token}
59 
60@app.get("/trips/{trip_id}")
61def get_trip(trip_id: int, authorization: str = Header(None)):
62    user_id = get_user_id_from_token(authorization)
63    if trip_id not in trips:
64        raise HTTPException(status_code=404, detail="Trip not found")
65    return trips[trip_id]
66 
67@app.post("/trips")
68def create_trip(trip: TripCreate, authorization: str = Header(None)):
69    global trip_id_counter
70    user_id = get_user_id_from_token(authorization)
71    new_trip = {
72        "id": trip_id_counter,
73        "destination": trip.destination,
74        "start_date": trip.start_date,
75        "end_date": trip.end_date,
76        "preferences": trip.preferences,
77        "tier": "standard",
78        "status": "active",
79        "user_id": user_id
80    }
81    trips[trip_id_counter] = new_trip
82    trip_id_counter += 1
83    return new_trip
84 
85@app.patch("/trips/{trip_id}")
86def update_trip(trip_id: int, update: TripUpdate, authorization: str = Header(None)):
87    user_id = get_user_id_from_token(authorization)
88    if trip_id not in trips:
89        raise HTTPException(status_code=404, detail="Trip not found")
90    trip = trips[trip_id]
91    if update.destination is not None:
92        trip["destination"] = update.destination
93    if update.start_date is not None:
94        trip["start_date"] = update.start_date
95    if update.end_date is not None:
96        trip["end_date"] = update.end_date
97    if update.preferences is not None:
98        trip["preferences"] = update.preferences
99    if update.tier is not None:
100        trip["tier"] = update.tier
101    if update.status is not None:
102        trip["status"] = update.status
103    return trip

1fastapi
2uvicorn