Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import hashlib
5import random
6import string
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12trips = {}
13waypoints = {}
14trip_counter = 0
15waypoint_counter = 0
16 
17def generate_token():
18    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
19 
20def hash_password(password: str):
21    return hashlib.sha256(password.encode()).hexdigest()
22 
23def get_user_from_token(authorization: str = Header(None)):
24    if not authorization:
25        raise HTTPException(status_code=401, detail="Missing auth token")
26    token = authorization.replace("Bearer ", "")
27    user_id = tokens.get(token)
28    if not user_id:
29        raise HTTPException(status_code=401, detail="Invalid token")
30    return user_id
31 
32class UserSignup(BaseModel):
33    username: str
34    password: str
35 
36class UserLogin(BaseModel):
37    username: str
38    password: str
39 
40class TripCreate(BaseModel):
41    start: str
42    end: str
43 
44class WaypointCreate(BaseModel):
45    trip_id: int
46    name: str
47    lat: float
48    lon: float
49    distance_from_prev: Optional[float] = 0
50    gas_cost: Optional[float] = 0
51    suggested_stop: Optional[str] = None
52 
53@app.post("/signup")
54def signup(user: UserSignup):
55    if user.username in users:
56        raise HTTPException(status_code=400, detail="User exists")
57    users[user.username] = hash_password(user.password)
58    return {"message": "User created"}
59 
60@app.post("/login")
61def login(user: UserLogin):
62    stored = users.get(user.username)
63    if not stored or stored != hash_password(user.password):
64        raise HTTPException(status_code=401, detail="Invalid credentials")
65    token = generate_token()
66    tokens[token] = user.username
67    return {"token": token}
68 
69@app.get("/trip")
70def get_trip(start: str, end: str, authorization: str = Header(None)):
71    user = get_user_from_token(authorization)
72    global trip_counter
73    trip_counter += 1
74    trip_id = trip_counter
75    
76    # Mock waypoints for LA to SF
77    waypoints_list = [
78        {"name": "Los Angeles", "lat": 34.0522, "lon": -118.2437, "distance_from_prev": 0, "gas_cost": 0, "suggested_stop": "Start"},
79        {"name": "Bakersfield", "lat": 35.3733, "lon": -119.0187, "distance_from_prev": 112, "gas_cost": 14.56, "suggested_stop": "Buck Owens' Crystal Palace"},
80        {"name": "Fresno", "lat": 36.7378, "lon": -119.7871, "distance_from_prev": 107, "gas_cost": 13.91, "suggested_stop": "Forestiere Underground Gardens"},
81        {"name": "San Jose", "lat": 37.3382, "lon": -121.8863, "distance_from_prev": 150, "gas_cost": 19.50, "suggested_stop": "Winchester Mystery House"},
82        {"name": "San Francisco", "lat": 37.7749, "lon": -122.4194, "distance_from_prev": 48, "gas_cost": 6.24, "suggested_stop": "Golden Gate Bridge"}
83    ]
84    
85    total_distance = sum(w["distance_from_prev"] for w in waypoints_list)
86    total_gas = sum(w["gas_cost"] for w in waypoints_list)
87    stops = [w["suggested_stop"] for w in waypoints_list if w["suggested_stop"] and w["suggested_stop"] != "Start"]
88    
89    trip_data = {
90        "id": trip_id,
91        "start": start,
92        "end": end,
93        "waypoints": waypoints_list,
94        "total_distance_miles": total_distance,
95        "total_gas_cost_usd": round(total_gas, 2),
96        "suggested_stops": stops
97    }
98    
99    trips[trip_id] = trip_data
100    return trip_data
101 
102@app.post("/optimize")
103def optimize_trip(trip_id: int, authorization: str = Header(None)):
104    user = get_user_from_token(authorization)
105    trip = trips.get(trip_id)
106    if not trip:
107        raise HTTPException(status_code=404, detail="Trip not found")
108    
109    # Simple optimization: reorder waypoints by lat (northward for LA->SF)
110    waypoints = trip["waypoints"]
111    # Keep start and end fixed, sort middle ones by latitude
112    if len(waypoints) > 2:
113        start = waypoints[0]
114        end = waypoints[-1]
115        middle = waypoints[1:-1]
116        middle.sort(key=lambda w: w["lat"])
117        waypoints = [start] + middle + [end]
118        
119        # Recalculate distances (simplified - just reorder)
120        for i in range(1, len(waypoints)):
121            waypoints[i]["distance_from_prev"] = round(abs(waypoints[i]["lat"] - waypoints[i-1]["lat"]) * 69, 2)
122            waypoints[i]["gas_cost"] = round(waypoints[i]["distance_from_prev"] * 0.13, 2)
123    
124    trip["waypoints"] = waypoints
125    trip["total_distance_miles"] = sum(w["distance_from_prev"] for w in waypoints)
126    trip["total_gas_cost_usd"] = round(sum(w["gas_cost"] for w in waypoints), 2)
127    
128    return {"optimized": True, "trip": trip}
129 
130@app.get("/trip/{trip_id}")
131def get_trip_by_id(trip_id: int, authorization: str = Header(None)):
132    user = get_user_from_token(authorization)
133    trip = trips.get(trip_id)
134    if not trip:
135        raise HTTPException(status_code=404, detail="Trip not found")
136    return trip
137 
138@app.post("/waypoint")
139def create_waypoint(waypoint: WaypointCreate, authorization: str = Header(None)):
140    user = get_user_from_token(authorization)
141    global waypoint_counter
142    waypoint_counter += 1
143    wp = waypoint.dict()
144    wp["id"] = waypoint_counter
145    waypoints[waypoint_counter] = wp
146    return wp
147 
148@app.get("/waypoint/{waypoint_id}")
149def get_waypoint(waypoint_id: int, authorization: str = Header(None)):
150    user = get_user_from_token(authorization)
151    wp = waypoints.get(waypoint_id)
152    if not wp:
153        raise HTTPException(status_code=404, detail="Waypoint not found")
154    return wp

1fastapi
2uvicorn