Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4from datetime import datetime
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10vehicles = {}
11vehicle_id_counter = 1
12maintenance_logs = []
13maintenance_id_counter = 1
14 
15def get_current_user(authorization: Optional[str] = Header(None)):
16    if not authorization:
17        raise HTTPException(status_code=401, detail="Missing auth token")
18    token = authorization.replace("Bearer ", "")
19    user_id = tokens.get(token)
20    if not user_id:
21        raise HTTPException(status_code=401, detail="Invalid token")
22    return user_id
23 
24@app.post("/signup")
25def signup(username: str, password: str):
26    if username in users:
27        raise HTTPException(status_code=400, detail="User exists")
28    users[username] = {"username": username, "password": password}
29    return {"message": "User created"}
30 
31@app.post("/login")
32def login(username: str, password: str):
33    user = users.get(username)
34    if not user or user["password"] != password:
35        raise HTTPException(status_code=401, detail="Invalid credentials")
36    token = secrets.token_hex(16)
37    tokens[token] = username
38    return {"token": token}
39 
40@app.get("/vehicles/{vehicle_id}")
41def get_vehicle(vehicle_id: int, authorization: Optional[str] = Header(None)):
42    get_current_user(authorization)
43    vehicle = vehicles.get(vehicle_id)
44    if not vehicle:
45        raise HTTPException(status_code=404, detail="Vehicle not found")
46    return vehicle
47 
48@app.post("/vehicles")
49def create_vehicle(model: str, daily_rate: float, insurance_tier: str, authorization: Optional[str] = Header(None)):
50    get_current_user(authorization)
51    global vehicle_id_counter
52    vehicle = {
53        "id": vehicle_id_counter,
54        "model": model,
55        "availability_status": "available",
56        "daily_rate": daily_rate,
57        "insurance_tier": insurance_tier
58    }
59    vehicles[vehicle_id_counter] = vehicle
60    vehicle_id_counter += 1
61    return vehicle
62 
63@app.patch("/vehicles/{vehicle_id}")
64def update_vehicle(vehicle_id: int, model: Optional[str] = None, availability_status: Optional[str] = None, daily_rate: Optional[float] = None, insurance_tier: Optional[str] = None, authorization: Optional[str] = Header(None)):
65    get_current_user(authorization)
66    vehicle = vehicles.get(vehicle_id)
67    if not vehicle:
68        raise HTTPException(status_code=404, detail="Vehicle not found")
69    if model is not None:
70        vehicle["model"] = model
71    if availability_status is not None:
72        vehicle["availability_status"] = availability_status
73    if daily_rate is not None:
74        vehicle["daily_rate"] = daily_rate
75    if insurance_tier is not None:
76        vehicle["insurance_tier"] = insurance_tier
77    global maintenance_id_counter
78    maintenance_logs.append({
79        "id": maintenance_id_counter,
80        "vehicle_id": vehicle_id,
81        "timestamp": datetime.utcnow().isoformat(),
82        "action": "updated",
83        "details": f"model={model}, availability_status={availability_status}, daily_rate={daily_rate}, insurance_tier={insurance_tier}"
84    })
85    maintenance_id_counter += 1
86    return vehicle
87 
88@app.get("/maintenance")
89def get_maintenance_logs(authorization: Optional[str] = Header(None)):
90    get_current_user(authorization)
91    return maintenance_logs

1fastapi
2uvicorn