Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3from datetime import date
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10cars = {}
11rentals = {}
12car_id_counter = 1
13user_id_counter = 1
14rental_id_counter = 1
15 
16# Seed some cars
17cars[1] = {"id": 1, "make": "Toyota", "model": "Camry", "year": 2020, "daily_rate": 45.0, "mileage_limit": 150, "available": True}
18cars[2] = {"id": 2, "make": "Honda", "model": "Civic", "year": 2021, "daily_rate": 40.0, "mileage_limit": 200, "available": True}
19cars[3] = {"id": 3, "make": "Ford", "model": "Mustang", "year": 2022, "daily_rate": 80.0, "mileage_limit": 100, "available": True}
20car_id_counter = 4
21 
22def get_current_user(authorization: Optional[str] = Header(None)):
23    if not authorization:
24        raise HTTPException(status_code=401, detail="Missing auth header")
25    token = authorization.replace("Bearer ", "")
26    if token not in tokens:
27        raise HTTPException(status_code=401, detail="Invalid token")
28    return tokens[token]
29 
30@app.post("/signup")
31def signup(username: str, password: str):
32    global user_id_counter
33    if any(u["username"] == username for u in users.values()):
34        raise HTTPException(status_code=400, detail="Username taken")
35    user_id = user_id_counter
36    users[user_id] = {"id": user_id, "username": username, "password": password}
37    user_id_counter += 1
38    return {"id": user_id, "username": username}
39 
40@app.post("/login")
41def login(username: str, password: str):
42    for u in users.values():
43        if u["username"] == username and u["password"] == password:
44            token = secrets.token_hex(16)
45            tokens[token] = u["id"]
46            return {"token": token}
47    raise HTTPException(status_code=401, detail="Invalid credentials")
48 
49@app.get("/cars")
50def get_cars():
51    return [c for c in cars.values() if c["available"]]
52 
53@app.get("/cars/{car_id}")
54def get_car(car_id: int):
55    if car_id not in cars:
56        raise HTTPException(status_code=404, detail="Car not found")
57    return cars[car_id]
58 
59@app.post("/cars")
60def create_car(make: str, model: str, year: int, daily_rate: float, mileage_limit: int, authorization: Optional[str] = Header(None)):
61    get_current_user(authorization)
62    global car_id_counter
63    car_id = car_id_counter
64    cars[car_id] = {"id": car_id, "make": make, "model": model, "year": year, "daily_rate": daily_rate, "mileage_limit": mileage_limit, "available": True}
65    car_id_counter += 1
66    return cars[car_id]
67 
68@app.post("/cars/rent")
69def rent_car(car_id: int, start_date: date, end_date: date, authorization: Optional[str] = Header(None)):
70    user_id = get_current_user(authorization)
71    if car_id not in cars or not cars[car_id]["available"]:
72        raise HTTPException(status_code=400, detail="Car not available")
73    if start_date >= end_date:
74        raise HTTPException(status_code=400, detail="Invalid date range")
75    global rental_id_counter
76    rental_id = rental_id_counter
77    rentals[rental_id] = {
78        "id": rental_id,
79        "car_id": car_id,
80        "user_id": user_id,
81        "start_date": start_date.isoformat(),
82        "end_date": end_date.isoformat()
83    }
84    cars[car_id]["available"] = False
85    rental_id_counter += 1
86    return {"rental_agreement_id": rental_id, "car_id": car_id, "start_date": start_date.isoformat(), "end_date": end_date.isoformat()}
87 
88@app.get("/rentals/{rental_id}")
89def get_rental(rental_id: int, authorization: Optional[str] = Header(None)):
90    get_current_user(authorization)
91    if rental_id not in rentals:
92        raise HTTPException(status_code=404, detail="Rental not found")
93    return rentals[rental_id]
94 
95@app.post("/rentals")
96def create_rental(car_id: int, start_date: date, end_date: date, authorization: Optional[str] = Header(None)):
97    return rent_car(car_id, start_date, end_date, authorization)

1fastapi
2uvicorn