Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import secrets
5import uvicorn
6from datetime import date
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12cars = {}
13car_id_counter = 1
14user_id_counter = 1
15 
16class UserSignup(BaseModel):
17    username: str
18    password: str
19 
20class UserLogin(BaseModel):
21    username: str
22    password: str
23 
24class CarCreate(BaseModel):
25    type: str
26    location: str
27    daily_rate: float
28    license_plate: str
29    mileage_allowance: Optional[int] = None
30    insurance_included: Optional[bool] = False
31 
32@app.post("/signup")
33def signup(user: UserSignup):
34    global user_id_counter
35    if any(u["username"] == user.username for u in users.values()):
36        raise HTTPException(400, "Username taken")
37    uid = user_id_counter
38    users[uid] = {"id": uid, "username": user.username, "password": user.password}
39    user_id_counter += 1
40    token = secrets.token_hex(16)
41    tokens[token] = uid
42    return {"user_id": uid, "token": token}
43 
44@app.post("/login")
45def login(user: UserLogin):
46    for uid, u in users.items():
47        if u["username"] == user.username and u["password"] == user.password:
48            token = secrets.token_hex(16)
49            tokens[token] = uid
50            return {"user_id": uid, "token": token}
51    raise HTTPException(401, "Invalid credentials")
52 
53def get_current_user(authorization: str = Header(...)):
54    token = authorization.replace("Bearer ", "")
55    if token not in tokens:
56        raise HTTPException(401, "Invalid token")
57    return tokens[token]
58 
59@app.get("/cars/{car_id}")
60def get_car(car_id: int, authorization: str = Header(...)):
61    get_current_user(authorization)
62    if car_id not in cars:
63        raise HTTPException(404, "Car not found")
64    return cars[car_id]
65 
66@app.post("/cars")
67def create_car(car: CarCreate, authorization: str = Header(...)):
68    global car_id_counter
69    get_current_user(authorization)
70    cid = car_id_counter
71    cars[cid] = {
72        "id": cid,
73        "type": car.type,
74        "location": car.location,
75        "daily_rate": car.daily_rate,
76        "license_plate": car.license_plate,
77        "mileage_allowance": car.mileage_allowance,
78        "insurance_included": car.insurance_included
79    }
80    car_id_counter += 1
81    return cars[cid]
82 
83@app.get("/cars")
84def list_cars(
85    type: Optional[str] = None,
86    location: Optional[str] = None,
87    rental_start: Optional[date] = None,
88    locations: Optional[str] = None,
89    authorization: str = Header(...)
90):
91    get_current_user(authorization)
92    result = []
93    locs = []
94    if locations:
95        locs = [l.strip() for l in locations.split(",")]
96    for c in cars.values():
97        if type and c["type"] != type:
98            continue
99        if location and c["location"] != location:
100            continue
101        if rental_start:
102            pass
103        if locs and c["location"] not in locs:
104            continue
105        result.append({
106            "daily_rate": c["daily_rate"],
107            "license_plate": c["license_plate"],
108            "mileage_allowance": c["mileage_allowance"],
109            "insurance_included": c["insurance_included"]
110        })
111    return result

1fastapi
2uvicorn