Browse the data

1import secrets
2from datetime import date, datetime
3from fastapi import FastAPI, HTTPException, Header
4from pydantic import BaseModel
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10cars = {}
11bookings = {}
12car_id_counter = 1
13booking_id_counter = 1
14user_id_counter = 1
15 
16class SignupRequest(BaseModel):
17    username: str
18    password: str
19 
20class LoginRequest(BaseModel):
21    username: str
22    password: str
23 
24class CarCreate(BaseModel):
25    model: str
26    mileage: int
27    daily_rate: float
28 
29class BookingCreate(BaseModel):
30    car_id: int
31    start_date: str
32    end_date: str
33 
34def get_user_id_from_token(authorization: str):
35    if not authorization or not authorization.startswith("Bearer "):
36        raise HTTPException(status_code=401, detail="Invalid auth header")
37    token = authorization.split(" ")[1]
38    for uid, t in tokens.items():
39        if t == token:
40            return uid
41    raise HTTPException(status_code=401, detail="Invalid token")
42 
43@app.post("/signup")
44def signup(req: SignupRequest):
45    global user_id_counter
46    for u in users.values():
47        if u["username"] == req.username:
48            raise HTTPException(status_code=400, detail="Username taken")
49    uid = user_id_counter
50    user_id_counter += 1
51    users[uid] = {"id": uid, "username": req.username, "password": req.password}
52    token = secrets.token_hex(32)
53    tokens[uid] = token
54    return {"user_id": uid, "token": token}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    for uid, u in users.items():
59        if u["username"] == req.username and u["password"] == req.password:
60            token = secrets.token_hex(32)
61            tokens[uid] = token
62            return {"user_id": uid, "token": token}
63    raise HTTPException(status_code=401, detail="Invalid credentials")
64 
65@app.get("/cars/{car_id}")
66def get_car(car_id: int, authorization: str = Header(None)):
67    get_user_id_from_token(authorization)
68    if car_id not in cars:
69        raise HTTPException(status_code=404, detail="Car not found")
70    return cars[car_id]
71 
72@app.get("/cars")
73def get_cars(authorization: str = Header(None)):
74    get_user_id_from_token(authorization)
75    return [c for c in cars.values()]
76 
77@app.post("/cars")
78def create_car(car: CarCreate, authorization: str = Header(None)):
79    global car_id_counter
80    get_user_id_from_token(authorization)
81    cid = car_id_counter
82    car_id_counter += 1
83    cars[cid] = {"id": cid, "model": car.model, "mileage": car.mileage, "daily_rate": car.daily_rate}
84    return cars[cid]
85 
86@app.get("/bookings/{booking_id}")
87def get_booking(booking_id: int, authorization: str = Header(None)):
88    get_user_id_from_token(authorization)
89    if booking_id not in bookings:
90        raise HTTPException(status_code=404, detail="Booking not found")
91    return bookings[booking_id]
92 
93@app.post("/bookings")
94def create_booking(booking: BookingCreate, authorization: str = Header(None)):
95    global booking_id_counter
96    user_id = get_user_id_from_token(authorization)
97    if booking.car_id not in cars:
98        raise HTTPException(status_code=404, detail="Car not found")
99    start = datetime.strptime(booking.start_date, "%Y-%m-%d").date()
100    end = datetime.strptime(booking.end_date, "%Y-%m-%d").date()
101    if start >= end:
102        raise HTTPException(status_code=400, detail="start_date must be before end_date")
103    for b in bookings.values():
104        if b["car_id"] == booking.car_id:
105            b_start = datetime.strptime(b["start_date"], "%Y-%m-%d").date()
106            b_end = datetime.strptime(b["end_date"], "%Y-%m-%d").date()
107            if not (end <= b_start or start >= b_end):
108                raise HTTPException(status_code=400, detail="Car not available for those dates")
109    bid = booking_id_counter
110    booking_id_counter += 1
111    bookings[bid] = {
112        "id": bid,
113        "car_id": booking.car_id,
114        "user_id": user_id,
115        "start_date": booking.start_date,
116        "end_date": booking.end_date
117    }
118    return bookings[bid]

1fastapi
2uvicorn