Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8rides = {}
9ride_requests = {}
10tokens = {}
11user_id_counter = 1
12ride_id_counter = 1
13request_id_counter = 1
14 
15def get_current_user(authorization: Optional[str] = Header(None)):
16    if not authorization:
17        raise HTTPException(status_code=401, detail="Missing auth header")
18    token = authorization.replace("Bearer ", "")
19    user_id = tokens.get(token)
20    if not user_id:
21        raise HTTPException(status_code=401, detail="Invalid token")
22    return user_id
23 
24@app.post("/signup")
25def signup(username: str, password: str):
26    global user_id_counter
27    user_id = user_id_counter
28    user_id_counter += 1
29    users[user_id] = {"id": user_id, "username": username, "password": password}
30    return {"id": user_id, "username": username}
31 
32@app.post("/login")
33def login(username: str, password: str):
34    for uid, u in users.items():
35        if u["username"] == username and u["password"] == password:
36            token = secrets.token_hex(16)
37            tokens[token] = uid
38            return {"token": token}
39    raise HTTPException(status_code=401, detail="Invalid credentials")
40 
41@app.get("/users/{user_id}")
42def get_user(user_id: int):
43    user = users.get(user_id)
44    if not user:
45        raise HTTPException(status_code=404, detail="User not found")
46    return user
47 
48@app.post("/rides")
49def create_ride(route: str, seats: int, authorization: Optional[str] = Header(None)):
50    driver_id = get_current_user(authorization)
51    global ride_id_counter
52    ride_id = ride_id_counter
53    ride_id_counter += 1
54    rides[ride_id] = {
55        "id": ride_id,
56        "driver_id": driver_id,
57        "route": route,
58        "seats": seats,
59        "co_drivers": [],
60        "riders": []
61    }
62    return rides[ride_id]
63 
64@app.get("/rides/{ride_id}")
65def get_ride(ride_id: int):
66    ride = rides.get(ride_id)
67    if not ride:
68        raise HTTPException(status_code=404, detail="Ride not found")
69    return ride
70 
71@app.post("/ride-requests")
72def create_ride_request(ride_id: int, authorization: Optional[str] = Header(None)):
73    rider_id = get_current_user(authorization)
74    ride = rides.get(ride_id)
75    if not ride:
76        raise HTTPException(status_code=404, detail="Ride not found")
77    if ride["seats"] <= len(ride["riders"]):
78        raise HTTPException(status_code=400, detail="No seats available")
79    global request_id_counter
80    request_id = request_id_counter
81    request_id_counter += 1
82    ride_requests[request_id] = {
83        "id": request_id,
84        "ride_id": ride_id,
85        "rider_id": rider_id,
86        "status": "pending"
87    }
88    return ride_requests[request_id]
89 
90@app.get("/ride-requests/{request_id}")
91def get_ride_request(request_id: int):
92    req = ride_requests.get(request_id)
93    if not req:
94        raise HTTPException(status_code=404, detail="Request not found")
95    return req
96 
97@app.post("/rides/{ride_id}/make-co-driver")
98def make_co_driver(ride_id: int, rider_id: int, authorization: Optional[str] = Header(None)):
99    driver_id = get_current_user(authorization)
100    ride = rides.get(ride_id)
101    if not ride:
102        raise HTTPException(status_code=404, detail="Ride not found")
103    if ride["driver_id"] != driver_id:
104        raise HTTPException(status_code=403, detail="Only the driver can promote co-drivers")
105    if rider_id not in ride["riders"]:
106        # maybe they're not yet a rider — we'll add them as co-driver directly
107        pass
108    if rider_id not in ride["co_drivers"]:
109        ride["co_drivers"].append(rider_id)
110    return {"message": "Co-driver added", "ride_id": ride_id, "co_drivers": ride["co_drivers"]}

1fastapi
2uvicorn