Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import datetime
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11lots = {
12    1: {"name": "Downtown Garage", "total": 50, "available": 50},
13    2: {"name": "Airport Parking", "total": 200, "available": 200},
14    3: {"name": "Mall Lot A", "total": 100, "available": 100},
15}
16reservations = {}
17reservation_id_counter = 1
18 
19def get_current_user(authorization: Optional[str] = Header(None)):
20    if not authorization:
21        raise HTTPException(status_code=401, detail="Missing auth header")
22    token = authorization.replace("Bearer ", "")
23    user_id = tokens.get(token)
24    if not user_id:
25        raise HTTPException(status_code=401, detail="Invalid token")
26    return user_id
27 
28class SignupRequest(BaseModel):
29    username: str
30    password: str
31 
32class LoginRequest(BaseModel):
33    username: str
34    password: str
35 
36class ReservationRequest(BaseModel):
37    lot_id: int
38    start_time: str
39    end_time: str
40    license_plate: str
41 
42@app.post("/signup")
43def signup(req: SignupRequest):
44    if req.username in users:
45        raise HTTPException(status_code=400, detail="User already exists")
46    user_id = len(users) + 1
47    users[req.username] = {"id": user_id, "password": req.password}
48    return {"id": user_id, "username": req.username}
49 
50@app.post("/login")
51def login(req: LoginRequest):
52    user = users.get(req.username)
53    if not user or user["password"] != req.password:
54        raise HTTPException(status_code=401, detail="Invalid credentials")
55    token = secrets.token_hex(16)
56    tokens[token] = user["id"]
57    return {"token": token}
58 
59@app.get("/lots")
60def get_lots(authorization: Optional[str] = Header(None)):
61    get_current_user(authorization)
62    result = []
63    for lid, lot in lots.items():
64        result.append({
65            "id": lid,
66            "name": lot["name"],
67            "total": lot["total"],
68            "available": lot["available"]
69        })
70    return result
71 
72@app.get("/lots/{lot_id}")
73def get_lot(lot_id: int, authorization: Optional[str] = Header(None)):
74    get_current_user(authorization)
75    lot = lots.get(lot_id)
76    if not lot:
77        raise HTTPException(status_code=404, detail="Lot not found")
78    return {"id": lot_id, "name": lot["name"], "total": lot["total"], "available": lot["available"]}
79 
80@app.post("/reservations")
81def create_reservation(req: ReservationRequest, authorization: Optional[str] = Header(None)):
82    user_id = get_current_user(authorization)
83    global reservation_id_counter
84    lot = lots.get(req.lot_id)
85    if not lot:
86        raise HTTPException(status_code=404, detail="Lot not found")
87    if lot["available"] <= 0:
88        raise HTTPException(status_code=400, detail="No spots available")
89    rid = reservation_id_counter
90    reservation_id_counter += 1
91    reservations[rid] = {
92        "id": rid,
93        "user_id": user_id,
94        "lot_id": req.lot_id,
95        "start_time": req.start_time,
96        "end_time": req.end_time,
97        "license_plate": req.license_plate,
98        "status": "active"
99    }
100    lot["available"] -= 1
101    return reservations[rid]
102 
103@app.get("/reservations/{reservation_id}")
104def get_reservation(reservation_id: int, authorization: Optional[str] = Header(None)):
105    get_current_user(authorization)
106    res = reservations.get(reservation_id)
107    if not res:
108        raise HTTPException(status_code=404, detail="Reservation not found")
109    return res
110 
111@app.delete("/reservations/{reservation_id}")
112def cancel_reservation(reservation_id: int, authorization: Optional[str] = Header(None)):
113    user_id = get_current_user(authorization)
114    res = reservations.get(reservation_id)
115    if not res:
116        raise HTTPException(status_code=404, detail="Reservation not found")
117    if res["user_id"] != user_id:
118        raise HTTPException(status_code=403, detail="Not your reservation")
119    lot = lots[res["lot_id"]]
120    lot["available"] += 1
121    del reservations[reservation_id]
122    return {"detail": "Reservation cancelled"}

1fastapi
2uvicorn