Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10flights = {}
11hotels = {}
12activities = {}
13trips = {}
14trip_items = {}
15 
16user_id_counter = 1
17flight_id_counter = 1
18hotel_id_counter = 1
19activity_id_counter = 1
20trip_id_counter = 1
21trip_item_id_counter = 1
22 
23class SignupRequest(BaseModel):
24    username: str
25    password: str
26 
27class LoginRequest(BaseModel):
28    username: str
29    password: str
30 
31class FlightCreate(BaseModel):
32    airline: str
33    flight_number: str
34    departure: str
35    arrival: str
36    cost: float
37 
38class HotelCreate(BaseModel):
39    name: str
40    location: str
41    check_in: str
42    check_out: str
43    cost: float
44 
45class ActivityCreate(BaseModel):
46    name: str
47    location: str
48    date: str
49    cost: float
50 
51class TripCreate(BaseModel):
52    name: str
53 
54class TripItemCreate(BaseModel):
55    trip_id: int
56    resource_type: str
57    resource_id: int
58 
59def get_current_user(authorization: str = Header(...)):
60    if not authorization.startswith("Bearer "):
61        raise HTTPException(status_code=401, detail="Invalid token")
62    token = authorization[7:]
63    if token not in tokens:
64        raise HTTPException(status_code=401, detail="Invalid token")
65    return tokens[token]
66 
67@app.post("/signup")
68def signup(req: SignupRequest):
69    global user_id_counter
70    if req.username in users:
71        raise HTTPException(status_code=400, detail="Username already exists")
72    user_id = user_id_counter
73    user_id_counter += 1
74    users[req.username] = {"id": user_id, "username": req.username, "password": req.password}
75    return {"id": user_id, "username": req.username}
76 
77@app.post("/login")
78def login(req: LoginRequest):
79    user = users.get(req.username)
80    if not user or user["password"] != req.password:
81        raise HTTPException(status_code=401, detail="Invalid credentials")
82    token = secrets.token_hex(16)
83    tokens[token] = req.username
84    return {"token": token}
85 
86@app.post("/flights")
87def create_flight(flight: FlightCreate, authorization: str = Header(...)):
88    get_current_user(authorization)
89    global flight_id_counter
90    flight_id = flight_id_counter
91    flight_id_counter += 1
92    flights[flight_id] = {"id": flight_id, **flight.dict()}
93    return flights[flight_id]
94 
95@app.get("/flights/{flight_id}")
96def get_flight(flight_id: int, authorization: str = Header(...)):
97    get_current_user(authorization)
98    if flight_id not in flights:
99        raise HTTPException(status_code=404, detail="Flight not found")
100    return flights[flight_id]
101 
102@app.post("/hotels")
103def create_hotel(hotel: HotelCreate, authorization: str = Header(...)):
104    get_current_user(authorization)
105    global hotel_id_counter
106    hotel_id = hotel_id_counter
107    hotel_id_counter += 1
108    hotels[hotel_id] = {"id": hotel_id, **hotel.dict()}
109    return hotels[hotel_id]
110 
111@app.get("/hotels/{hotel_id}")
112def get_hotel(hotel_id: int, authorization: str = Header(...)):
113    get_current_user(authorization)
114    if hotel_id not in hotels:
115        raise HTTPException(status_code=404, detail="Hotel not found")
116    return hotels[hotel_id]
117 
118@app.post("/activities")
119def create_activity(activity: ActivityCreate, authorization: str = Header(...)):
120    get_current_user(authorization)
121    global activity_id_counter
122    activity_id = activity_id_counter
123    activity_id_counter += 1
124    activities[activity_id] = {"id": activity_id, **activity.dict()}
125    return activities[activity_id]
126 
127@app.get("/activities/{activity_id}")
128def get_activity(activity_id: int, authorization: str = Header(...)):
129    get_current_user(authorization)
130    if activity_id not in activities:
131        raise HTTPException(status_code=404, detail="Activity not found")
132    return activities[activity_id]
133 
134@app.post("/trips")
135def create_trip(trip: TripCreate, authorization: str = Header(...)):
136    get_current_user(authorization)
137    global trip_id_counter
138    trip_id = trip_id_counter
139    trip_id_counter += 1
140    trips[trip_id] = {"id": trip_id, "name": trip.name, "items": []}
141    return trips[trip_id]
142 
143@app.get("/trips/{trip_id}")
144def get_trip(trip_id: int, authorization: str = Header(...)):
145    get_current_user(authorization)
146    if trip_id not in trips:
147        raise HTTPException(status_code=404, detail="Trip not found")
148    return trips[trip_id]
149 
150@app.post("/trip_items")
151def add_trip_item(item: TripItemCreate, authorization: str = Header(...)):
152    get_current_user(authorization)
153    if item.trip_id not in trips:
154        raise HTTPException(status_code=404, detail="Trip not found")
155    resource_map = {"flight": flights, "hotel": hotels, "activity": activities}
156    if item.resource_type not in resource_map:
157        raise HTTPException(status_code=400, detail="Invalid resource type")
158    if item.resource_id not in resource_map[item.resource_type]:
159        raise HTTPException(status_code=404, detail="Resource not found")
160    global trip_item_id_counter
161    trip_item_id = trip_item_id_counter
162    trip_item_id_counter += 1
163    trip_item = {"id": trip_item_id, "trip_id": item.trip_id, "resource_type": item.resource_type, "resource_id": item.resource_id}
164    trip_items[trip_item_id] = trip_item
165    trips[item.trip_id]["items"].append(trip_item)
166    return trip_item
167 
168@app.get("/trips/{trip_id}/cost")
169def get_trip_cost(trip_id: int, authorization: str = Header(...)):
170    get_current_user(authorization)
171    if trip_id not in trips:
172        raise HTTPException(status_code=404, detail="Trip not found")
173    total = 0.0
174    breakdown = {"flights": 0.0, "hotels": 0.0, "activities": 0.0}
175    for item in trips[trip_id]["items"]:
176        rtype = item["resource_type"]
177        rid = item["resource_id"]
178        if rtype == "flight":
179            cost = flights[rid]["cost"]
180            breakdown["flights"] += cost
181        elif rtype == "hotel":
182            cost = hotels[rid]["cost"]
183            breakdown["hotels"] += cost
184        elif rtype == "activity":
185            cost = activities[rid]["cost"]
186            breakdown["activities"] += cost
187        total += cost
188    return {"total": total, "breakdown": breakdown}

1fastapi
2uvicorn