Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import random
5import hashlib
6import secrets
7from datetime import datetime, timedelta
8 
9app = FastAPI()
10 
11# In-memory stores
12users = {}
13tokens = {}
14itineraries = {}
15itinerary_id_counter = 1
16 
17# Simple token auth middleware
18def get_user_from_token(authorization: str = Header(None)):
19    if not authorization:
20        raise HTTPException(status_code=401, detail="Missing Authorization header")
21    token = authorization.replace("Bearer ", "")
22    if token not in tokens:
23        raise HTTPException(status_code=401, detail="Invalid token")
24    return tokens[token]
25 
26# Models
27class SignupRequest(BaseModel):
28    username: str
29    password: str
30 
31class LoginRequest(BaseModel):
32    username: str
33    password: str
34 
35class ItineraryRequest(BaseModel):
36    destination: str
37    start_date: str
38    end_date: str
39 
40class Attraction(BaseModel):
41    name: str
42    cost: float
43 
44class DayPlan(BaseModel):
45    day: int
46    date: str
47    attractions: List[Attraction]
48    total_cost: float
49 
50class Itinerary(BaseModel):
51    id: int
52    destination: str
53    start_date: str
54    end_date: str
55    days: List[DayPlan]
56 
57class WeatherInfo(BaseModel):
58    date: str
59    condition: str
60    temp_high: float
61    temp_low: float
62 
63# Attractions database per destination (hardcoded for speed)
64ATTRACTIONS_DB = {
65    "paris": [("Eiffel Tower", 25.0), ("Louvre Museum", 17.0), ("Notre-Dame", 0), ("Montmartre", 0), ("Seine River Cruise", 15.0)],
66    "tokyo": [("Senso-ji Temple", 0), ("Shibuya Crossing", 0), ("Tokyo Tower", 12.0), ("Tsukiji Market", 0), ("Meiji Shrine", 0)],
67    "new york": [("Statue of Liberty", 24.0), ("Central Park", 0), ("Times Square", 0), ("Broadway Show", 100.0), ("MET Museum", 25.0)],
68    "london": [("Big Ben", 0), ("Tower of London", 30.0), ("British Museum", 0), ("London Eye", 28.0), ("Buckingham Palace", 0)],
69    "rome": [("Colosseum", 16.0), ("Vatican Museums", 17.0), ("Trevi Fountain", 0), ("Pantheon", 5.0), ("Spanish Steps", 0)],
70    "bali": [("Ubud Monkey Forest", 8.0), ("Tanah Lot Temple", 6.0), ("Tegallalang Rice Terrace", 5.0), ("Uluwatu Temple", 5.0), ("Seminyak Beach", 0)],
71    "bangkok": [("Grand Palace", 15.0), ("Wat Pho", 5.0), ("Chatuchak Market", 0), ("Khao San Road", 0), ("Floating Market", 10.0)],
72    "sydney": [("Sydney Opera House", 40.0), ("Bondi Beach", 0), ("Harbour Bridge", 43.0), ("Taronga Zoo", 35.0), ("The Rocks", 0)],
73    "dubai": [("Burj Khalifa", 40.0), ("Dubai Mall", 0), ("Palm Jumeirah", 0), ("Desert Safari", 50.0), ("Gold Souk", 0)],
74    "barcelona": [("Sagrada Familia", 26.0), ("Park Guell", 10.0), ("La Rambla", 0), ("Barceloneta Beach", 0), ("Gothic Quarter", 0)],
75}
76 
77WEATHER_CONDITIONS = ["Sunny", "Partly Cloudy", "Cloudy", "Light Rain", "Clear", "Windy"]
78 
79# Auth endpoints
80@app.post("/signup")
81def signup(req: SignupRequest):
82    if req.username in users:
83        raise HTTPException(status_code=400, detail="Username already exists")
84    users[req.username] = hashlib.sha256(req.password.encode()).hexdigest()
85    return {"message": "User created", "username": req.username}
86 
87@app.post("/login")
88def login(req: LoginRequest):
89    if req.username not in users:
90        raise HTTPException(status_code=401, detail="Invalid credentials")
91    if users[req.username] != hashlib.sha256(req.password.encode()).hexdigest():
92        raise HTTPException(status_code=401, detail="Invalid credentials")
93    token = secrets.token_hex(32)
94    tokens[token] = req.username
95    return {"token": token, "username": req.username}
96 
97# Itinerary endpoints
98@app.post("/itineraries")
99def create_itinerary(req: ItineraryRequest, authorization: str = Header(None)):
100    get_user_from_token(authorization)
101    global itinerary_id_counter
102    
103    dest_lower = req.destination.lower()
104    if dest_lower not in ATTRACTIONS_DB:
105        raise HTTPException(status_code=400, detail=f"Destination '{req.destination}' not supported. Try: {', '.join(ATTRACTIONS_DB.keys())}")
106    
107    start = datetime.strptime(req.start_date, "%Y-%m-%d")
108    end = datetime.strptime(req.end_date, "%Y-%m-%d")
109    num_days = (end - start).days + 1
110    
111    if num_days < 1:
112        raise HTTPException(status_code=400, detail="End date must be after or equal to start date")
113    
114    attractions = ATTRACTIONS_DB[dest_lower]
115    days = []
116    
117    for i in range(num_days):
118        current_date = start + timedelta(days=i)
119        # Pick 2-3 random attractions per day
120        num_attractions = random.randint(2, min(3, len(attractions)))
121        selected = random.sample(attractions, num_attractions)
122        day_attractions = []
123        day_total = 0.0
124        for name, cost in selected:
125            day_attractions.append(Attraction(name=name, cost=cost))
126            day_total += cost
127        
128        days.append(DayPlan(
129            day=i+1,
130            date=current_date.strftime("%Y-%m-%d"),
131            attractions=day_attractions,
132            total_cost=round(day_total, 2)
133        ))
134    
135    itinerary = Itinerary(
136        id=itinerary_id_counter,
137        destination=req.destination,
138        start_date=req.start_date,
139        end_date=req.end_date,
140        days=days
141    )
142    
143    itineraries[itinerary_id_counter] = itinerary
144    itinerary_id_counter += 1
145    
146    return itinerary
147 
148@app.get("/itineraries/{itinerary_id}")
149def get_itinerary(itinerary_id: int, authorization: str = Header(None)):
150    get_user_from_token(authorization)
151    if itinerary_id not in itineraries:
152        raise HTTPException(status_code=404, detail="Itinerary not found")
153    return itineraries[itinerary_id]
154 
155@app.get("/itineraries/{itinerary_id}/weather")
156def get_itinerary_weather(itinerary_id: int, authorization: str = Header(None)):
157    get_user_from_token(authorization)
158    if itinerary_id not in itineraries:
159        raise HTTPException(status_code=404, detail="Itinerary not found")
160    
161    itinerary = itineraries[itinerary_id]
162    weather_data = []
163    
164    for day in itinerary.days:
165        weather_data.append(WeatherInfo(
166            date=day.date,
167            condition=random.choice(WEATHER_CONDITIONS),
168            temp_high=round(random.uniform(15.0, 35.0), 1),
169            temp_low=round(random.uniform(5.0, 20.0), 1)
170        ))
171    
172    return weather_data

1fastapi==0.104.1
2uvicorn==0.24.0
3pydantic==2.5.2