Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import time
6 
7app = FastAPI()
8 
9# In-memory stores
10users = {}
11tokens = {}
12lots = {}
13reservations = {}
14lot_id_counter = 1
15user_id_counter = 1
16 
17# Seed some parking lots
18lots[1] = {"id": 1, "total_spots": 50, "available_spots": 50, "hourly_rate": 5.0, "address": "123 Main St"}
19lots[2] = {"id": 2, "total_spots": 30, "available_spots": 30, "hourly_rate": 3.5, "address": "456 Oak Ave"}
20lots[3] = {"id": 3, "total_spots": 100, "available_spots": 100, "hourly_rate": 7.0, "address": "789 Pine Rd"}
21lot_id_counter = 4
22 
23class SignupRequest(BaseModel):
24    username: str
25    password: str
26 
27class LoginRequest(BaseModel):
28    username: str
29    password: str
30 
31class ReserveRequest(BaseModel):
32    license_plate: str
33 
34def get_current_user(authorization: Optional[str] = Header(None)):
35    if not authorization:
36        raise HTTPException(status_code=401, detail="Missing authorization header")
37    token = authorization.replace("Bearer ", "")
38    if token not in tokens:
39        raise HTTPException(status_code=401, detail="Invalid token")
40    return tokens[token]
41 
42@app.post("/signup")
43def signup(req: SignupRequest):
44    global user_id_counter
45    if req.username in users:
46        raise HTTPException(status_code=400, detail="Username already exists")
47    user_id = user_id_counter
48    user_id_counter += 1
49    users[req.username] = {"id": user_id, "username": req.username, "password": req.password}
50    return {"id": user_id, "username": req.username}
51 
52@app.post("/login")
53def login(req: LoginRequest):
54    if req.username not in users:
55        raise HTTPException(status_code=400, detail="Invalid credentials")
56    user = users[req.username]
57    if user["password"] != req.password:
58        raise HTTPException(status_code=400, detail="Invalid credentials")
59    token = secrets.token_hex(16)
60    tokens[token] = user["id"]
61    return {"token": token}
62 
63@app.get("/lots")
64def get_lots():
65    return list(lots.values())
66 
67@app.get("/lots/{lot_id}")
68def get_lot(lot_id: int):
69    if lot_id not in lots:
70        raise HTTPException(status_code=404, detail="Lot not found")
71    return lots[lot_id]
72 
73@app.post("/lots")
74def create_lot(total_spots: int, hourly_rate: float, address: str):
75    global lot_id_counter
76    lot_id = lot_id_counter
77    lot_id_counter += 1
78    lots[lot_id] = {
79        "id": lot_id,
80        "total_spots": total_spots,
81        "available_spots": total_spots,
82        "hourly_rate": hourly_rate,
83        "address": address
84    }
85    return lots[lot_id]
86 
87@app.post("/reserve")
88def reserve_spot(lot_id: int, req: ReserveRequest, authorization: Optional[str] = Header(None)):
89    user_id = get_current_user(authorization)
90    if lot_id not in lots:
91        raise HTTPException(status_code=404, detail="Lot not found")
92    lot = lots[lot_id]
93    if lot["available_spots"] <= 0:
94        raise HTTPException(status_code=400, detail="No spots available")
95    lot["available_spots"] -= 1
96    reservation_id = len(reservations) + 1
97    reservations[reservation_id] = {
98        "id": reservation_id,
99        "user_id": user_id,
100        "lot_id": lot_id,
101        "license_plate": req.license_plate,
102        "expires_at": time.time() + 1800
103    }
104    return reservations[reservation_id]

1fastapi
2uvicorn