Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import uvicorn
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11parking_spots = {}
12bookings = {}
13next_user_id = 1
14next_spot_id = 1
15next_booking_id = 1
16 
17class SignupRequest(BaseModel):
18    username: str
19    password: str
20 
21class LoginRequest(BaseModel):
22    username: str
23    password: str
24 
25class ParkingSpot(BaseModel):
26    location: str
27    price_per_hour: float
28    availability: bool = True
29 
30class BookingRequest(BaseModel):
31    spot_id: int
32    start_time: str
33    end_time: str
34 
35def get_current_user(authorization: str = Header(None)):
36    if not authorization:
37        raise HTTPException(status_code=401, detail="Missing auth token")
38    token = authorization.replace("Bearer ", "")
39    if token not in tokens:
40        raise HTTPException(status_code=401, detail="Invalid token")
41    return tokens[token]
42 
43@app.post("/signup")
44def signup(req: SignupRequest):
45    global next_user_id
46    user_id = next_user_id
47    next_user_id += 1
48    users[user_id] = {"username": req.username, "password": req.password}
49    token = secrets.token_hex(16)
50    tokens[token] = user_id
51    return {"user_id": user_id, "token": token}
52 
53@app.post("/login")
54def login(req: LoginRequest):
55    for uid, u in users.items():
56        if u["username"] == req.username and u["password"] == req.password:
57            token = secrets.token_hex(16)
58            tokens[token] = uid
59            return {"token": token}
60    raise HTTPException(status_code=401, detail="Invalid credentials")
61 
62@app.post("/parking-spots")
63def create_spot(spot: ParkingSpot, authorization: str = Header(None)):
64    get_current_user(authorization)
65    global next_spot_id
66    spot_id = next_spot_id
67    next_spot_id += 1
68    parking_spots[spot_id] = {
69        "id": spot_id,
70        "location": spot.location,
71        "price_per_hour": spot.price_per_hour,
72        "availability": spot.availability
73    }
74    return parking_spots[spot_id]
75 
76@app.get("/parking-spots/{spot_id}")
77def get_spot(spot_id: int, authorization: str = Header(None)):
78    get_current_user(authorization)
79    if spot_id not in parking_spots:
80        raise HTTPException(status_code=404, detail="Spot not found")
81    return parking_spots[spot_id]
82 
83@app.get("/parking-spots")
84def list_spots(authorization: str = Header(None)):
85    get_current_user(authorization)
86    return list(parking_spots.values())
87 
88@app.post("/bookings")
89def create_booking(booking: BookingRequest, authorization: str = Header(None)):
90    user_id = get_current_user(authorization)
91    if booking.spot_id not in parking_spots:
92        raise HTTPException(status_code=404, detail="Spot not found")
93    if not parking_spots[booking.spot_id]["availability"]:
94        raise HTTPException(status_code=400, detail="Spot not available")
95    global next_booking_id
96    booking_id = next_booking_id
97    next_booking_id += 1
98    bookings[booking_id] = {
99        "id": booking_id,
100        "user_id": user_id,
101        "spot_id": booking.spot_id,
102        "start_time": booking.start_time,
103        "end_time": booking.end_time
104    }
105    parking_spots[booking.spot_id]["availability"] = False
106    return bookings[booking_id]
107 
108@app.get("/bookings/{booking_id}")
109def get_booking(booking_id: int, authorization: str = Header(None)):
110    get_current_user(authorization)
111    if booking_id not in bookings:
112        raise HTTPException(status_code=404, detail="Booking not found")
113    return bookings[booking_id]
114 
115@app.get("/bookings")
116def list_bookings(authorization: str = Header(None)):
117    get_current_user(authorization)
118    return list(bookings.values())

1fastapi
2uvicorn