Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3from datetime import date
4import secrets
5import uvicorn
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11rentals = {}
12rental_id_counter = 0
13 
14def get_current_user(authorization: Optional[str] = Header(None)):
15    if not authorization or not authorization.startswith("Bearer "):
16        raise HTTPException(status_code=401, detail="Invalid or missing token")
17    token = authorization.split(" ")[1]
18    if token not in tokens:
19        raise HTTPException(status_code=401, detail="Invalid token")
20    return tokens[token]
21 
22@app.post("/signup")
23def signup(username: str, password: str):
24    if username in users:
25        raise HTTPException(status_code=400, detail="User already exists")
26    users[username] = {"username": username, "password": password}
27    return {"message": "User created"}
28 
29@app.post("/login")
30def login(username: str, password: str):
31    if username not in users or users[username]["password"] != password:
32        raise HTTPException(status_code=401, detail="Invalid credentials")
33    token = secrets.token_hex(16)
34    tokens[token] = username
35    return {"token": token}
36 
37@app.get("/rentals/{rental_id}")
38def get_rental(rental_id: int, authorization: Optional[str] = Header(None)):
39    get_current_user(authorization)
40    if rental_id not in rentals:
41        raise HTTPException(status_code=404, detail="Rental not found")
42    return rentals[rental_id]
43 
44@app.post("/rentals")
45def create_rental(
46    renter_name: str,
47    start_date: date,
48    end_date: date,
49    vehicle: str,
50    insurance_tier: str,
51    additional_driver: Optional[str] = None,
52    authorization: Optional[str] = Header(None)
53):
54    get_current_user(authorization)
55    global rental_id_counter
56    rental_id_counter += 1
57    rentals[rental_id_counter] = {
58        "id": rental_id_counter,
59        "renter_name": renter_name,
60        "start_date": start_date.isoformat(),
61        "end_date": end_date.isoformat(),
62        "vehicle": vehicle,
63        "insurance_tier": insurance_tier,
64        "additional_driver": additional_driver
65    }
66    return rentals[rental_id_counter]
67 
68@app.put("/rentals/{rental_id}")
69def update_rental(
70    rental_id: int,
71    renter_name: Optional[str] = None,
72    start_date: Optional[date] = None,
73    end_date: Optional[date] = None,
74    vehicle: Optional[str] = None,
75    insurance_tier: Optional[str] = None,
76    additional_driver: Optional[str] = None,
77    authorization: Optional[str] = Header(None)
78):
79    get_current_user(authorization)
80    if rental_id not in rentals:
81        raise HTTPException(status_code=404, detail="Rental not found")
82    rental = rentals[rental_id]
83    if renter_name is not None:
84        rental["renter_name"] = renter_name
85    if start_date is not None:
86        rental["start_date"] = start_date.isoformat()
87    if end_date is not None:
88        rental["end_date"] = end_date.isoformat()
89    if vehicle is not None:
90        rental["vehicle"] = vehicle
91    if insurance_tier is not None:
92        rental["insurance_tier"] = insurance_tier
93    if additional_driver is not None:
94        rental["additional_driver"] = additional_driver
95    return rental

1fastapi
2uvicorn