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 = {}
10spots = {}
11bookings = {}
12next_user_id = 1
13next_spot_id = 1
14next_booking_id = 1
15 
16def get_current_user(authorization: Optional[str] = Header(None)):
17    if not authorization:
18        raise HTTPException(status_code=401, detail="Missing auth header")
19    token = authorization.replace("Bearer ", "")
20    user_id = tokens.get(token)
21    if not user_id:
22        raise HTTPException(status_code=401, detail="Invalid token")
23    return user_id
24 
25class SignupRequest(BaseModel):
26    username: str
27    password: str
28 
29class LoginRequest(BaseModel):
30    username: str
31    password: str
32 
33class SpotCreate(BaseModel):
34    name: str
35    address: str
36    price_per_hour: float
37 
38class BookingCreate(BaseModel):
39    spot_id: int
40    start_time: str
41    end_time: str
42 
43@app.post("/signup")
44def signup(req: SignupRequest):
45    global next_user_id
46    for u in users.values():
47        if u["username"] == req.username:
48            raise HTTPException(status_code=400, detail="Username taken")
49    user_id = next_user_id
50    next_user_id += 1
51    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
52    return {"user_id": user_id}
53 
54@app.post("/login")
55def login(req: LoginRequest):
56    for u in users.values():
57        if u["username"] == req.username and u["password"] == req.password:
58            token = secrets.token_hex(16)
59            tokens[token] = u["id"]
60            return {"token": token}
61    raise HTTPException(status_code=401, detail="Invalid credentials")
62 
63@app.get("/users/{user_id}")
64def get_user(user_id: int):
65    user = users.get(user_id)
66    if not user:
67        raise HTTPException(status_code=404, detail="User not found")
68    return user
69 
70@app.get("/spots/{spot_id}")
71def get_spot(spot_id: int):
72    spot = spots.get(spot_id)
73    if not spot:
74        raise HTTPException(status_code=404, detail="Spot not found")
75    return spot
76 
77@app.get("/bookings/{booking_id}")
78def get_booking(booking_id: int, authorization: Optional[str] = Header(None)):
79    get_current_user(authorization)
80    booking = bookings.get(booking_id)
81    if not booking:
82        raise HTTPException(status_code=404, detail="Booking not found")
83    return booking
84 
85@app.post("/spots")
86def create_spot(spot: SpotCreate, authorization: Optional[str] = Header(None)):
87    owner_id = get_current_user(authorization)
88    global next_spot_id
89    spot_id = next_spot_id
90    next_spot_id += 1
91    spots[spot_id] = {
92        "id": spot_id,
93        "name": spot.name,
94        "address": spot.address,
95        "price_per_hour": spot.price_per_hour,
96        "owner_id": owner_id
97    }
98    return spots[spot_id]
99 
100@app.post("/bookings")
101def create_booking(booking: BookingCreate, authorization: Optional[str] = Header(None)):
102    driver_id = get_current_user(authorization)
103    if booking.spot_id not in spots:
104        raise HTTPException(status_code=404, detail="Spot not found")
105    global next_booking_id
106    booking_id = next_booking_id
107    next_booking_id += 1
108    bookings[booking_id] = {
109        "id": booking_id,
110        "spot_id": booking.spot_id,
111        "driver_id": driver_id,
112        "start_time": booking.start_time,
113        "end_time": booking.end_time
114    }
115    return bookings[booking_id]

1fastapi
2uvicorn