Browse the data

1import random
2import time
3from fastapi import FastAPI, HTTPException, Header
4from pydantic import BaseModel
5from typing import Optional, List
6 
7app = FastAPI()
8 
9users = {}
10user_id_counter = 1
11tokens = {}
12rides = {}
13ride_id_counter = 1
14drivers = {}
15driver_id_counter = 1
16 
17for i in range(20):
18    drivers[driver_id_counter] = {
19        "id": driver_id_counter,
20        "lat": 37.7749 + random.uniform(-0.05, 0.05),
21        "lng": -122.4194 + random.uniform(-0.05, 0.05),
22        "status": "available"
23    }
24    driver_id_counter += 1
25 
26class SignupRequest(BaseModel):
27    username: str
28    password: str
29 
30class LoginRequest(BaseModel):
31    username: str
32    password: str
33 
34class RideEstimateRequest(BaseModel):
35    origin_lat: float
36    origin_lng: float
37    dest_lat: float
38    dest_lng: float
39 
40class RideCreateRequest(BaseModel):
41    origin_lat: float
42    origin_lng: float
43    dest_lat: float
44    dest_lng: float
45 
46def auth(token: str = Header(...)):
47    if token not in tokens:
48        raise HTTPException(status_code=401, detail="Invalid token")
49    return tokens[token]
50 
51@app.post("/signup")
52def signup(req: SignupRequest):
53    global user_id_counter
54    for u in users.values():
55        if u["username"] == req.username:
56            raise HTTPException(status_code=400, detail="Username taken")
57    uid = user_id_counter
58    user_id_counter += 1
59    users[uid] = {"id": uid, "username": req.username, "password": req.password}
60    token = f"tok-{uid}-{random.randint(1000,9999)}"
61    tokens[token] = uid
62    return {"user_id": uid, "token": token}
63 
64@app.post("/login")
65def login(req: LoginRequest):
66    for u in users.values():
67        if u["username"] == req.username and u["password"] == req.password:
68            token = f"tok-{u['id']}-{random.randint(1000,9999)}"
69            tokens[token] = u["id"]
70            return {"user_id": u["id"], "token": token}
71    raise HTTPException(status_code=401, detail="Invalid credentials")
72 
73@app.get("/users/{user_id}")
74def get_user(user_id: int, authorization: str = Header(...)):
75    auth(authorization)
76    if user_id not in users:
77        raise HTTPException(status_code=404, detail="User not found")
78    return users[user_id]
79 
80@app.post("/users")
81def create_user(req: SignupRequest):
82    return signup(req)
83 
84@app.get("/rides/estimate")
85def estimate_ride(origin_lat: float, origin_lng: float, dest_lat: float, dest_lng: float, authorization: str = Header(...)):
86    auth(authorization)
87    base_fare = 8.50
88    dist = abs(origin_lat - dest_lat) + abs(origin_lng - dest_lng)
89    surge = 1.0 + random.random() * 0.5
90    wait = random.randint(2, 15)
91    fare = round(base_fare * dist * 80 * surge, 2)
92    return {
93        "fare": fare,
94        "surge_multiplier": round(surge, 2),
95        "wait_time_minutes": wait,
96        "distance_km": round(dist * 111, 2)
97    }
98 
99@app.get("/drivers/nearby")
100def nearby_drivers(lat: float, lng: float, authorization: str = Header(...)):
101    auth(authorization)
102    nearby = []
103    for d in drivers.values():
104        if d["status"] == "available":
105            ddist = abs(d["lat"] - lat) + abs(d["lng"] - lng)
106            if ddist < 0.1:
107                eta = random.randint(2, 20)
108                nearby.append({"driver_id": d["id"], "eta_minutes": eta})
109    count = len(nearby)
110    if count > 0:
111        avg_eta = round(sum(n["eta_minutes"] for n in nearby) / count, 1)
112    else:
113        avg_eta = 0
114    return {
115        "driver_count": count,
116        "eta_distribution": {
117            "min": min([n["eta_minutes"] for n in nearby]) if nearby else 0,
118            "max": max([n["eta_minutes"] for n in nearby]) if nearby else 0,
119            "average": avg_eta
120        }
121    }
122 
123@app.post("/rides")
124def create_ride(req: RideCreateRequest, authorization: str = Header(...)):
125    global ride_id_counter
126    uid = auth(authorization)
127    rid = ride_id_counter
128    ride_id_counter += 1
129    rides[rid] = {
130        "id": rid,
131        "user_id": uid,
132        "origin_lat": req.origin_lat,
133        "origin_lng": req.origin_lng,
134        "dest_lat": req.dest_lat,
135        "dest_lng": req.dest_lng,
136        "status": "requested",
137        "created_at": time.time()
138    }
139    return rides[rid]
140 
141@app.get("/rides/{ride_id}")
142def get_ride(ride_id: int, authorization: str = Header(...)):
143    auth(authorization)
144    if ride_id not in rides:
145        raise HTTPException(status_code=404, detail="Ride not found")
146    return rides[ride_id]

1fastapi
2uvicorn