Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import hashlib
4import secrets
5import datetime
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11spots = {}
12bookings = {}
13spot_id_counter = 1
14booking_id_counter = 1
15user_id_counter = 1
16 
17def get_current_user(authorization: Optional[str] = Header(None)):
18    if not authorization:
19        raise HTTPException(status_code=401, detail="Missing auth token")
20    token = authorization.replace("Bearer ", "")
21    if token not in tokens:
22        raise HTTPException(status_code=401, detail="Invalid token")
23    return tokens[token]
24 
25@app.post("/signup")
26def signup(username: str, password: str):
27    global user_id_counter
28    if any(u["username"] == username for u in users.values()):
29        raise HTTPException(status_code=400, detail="Username already exists")
30    user_id = user_id_counter
31    user_id_counter += 1
32    users[user_id] = {"id": user_id, "username": username, "password": hashlib.sha256(password.encode()).hexdigest()}
33    return {"user_id": user_id, "username": username}
34 
35@app.post("/login")
36def login(username: str, password: str):
37    for u in users.values():
38        if u["username"] == username and u["password"] == hashlib.sha256(password.encode()).hexdigest():
39            token = secrets.token_hex(32)
40            tokens[token] = u["id"]
41            return {"token": token}
42    raise HTTPException(status_code=401, detail="Invalid credentials")
43 
44@app.post("/spots")
45def create_spot(location: str, price_per_hour: float, availability: bool = True, authorization: Optional[str] = Header(None)):
46    get_current_user(authorization)
47    global spot_id_counter
48    spot_id = spot_id_counter
49    spot_id_counter += 1
50    spots[spot_id] = {"id": spot_id, "location": location, "price_per_hour": price_per_hour, "availability": availability}
51    return spots[spot_id]
52 
53@app.get("/spots/{spot_id}")
54def get_spot(spot_id: int, authorization: Optional[str] = Header(None)):
55    get_current_user(authorization)
56    if spot_id not in spots:
57        raise HTTPException(status_code=404, detail="Spot not found")
58    return spots[spot_id]
59 
60@app.get("/spots")
61def list_spots(authorization: Optional[str] = Header(None)):
62    get_current_user(authorization)
63    return list(spots.values())
64 
65@app.post("/bookings")
66def create_booking(spot_id: int, start_time: str, end_time: str, authorization: Optional[str] = Header(None)):
67    user_id = get_current_user(authorization)
68    if spot_id not in spots:
69        raise HTTPException(status_code=404, detail="Spot not found")
70    if not spots[spot_id]["availability"]:
71        raise HTTPException(status_code=400, detail="Spot not available")
72    global booking_id_counter
73    booking_id = booking_id_counter
74    booking_id_counter += 1
75    bookings[booking_id] = {
76        "id": booking_id,
77        "spot_id": spot_id,
78        "user_id": user_id,
79        "start_time": start_time,
80        "end_time": end_time
81    }
82    spots[spot_id]["availability"] = False
83    return bookings[booking_id]
84 
85@app.get("/bookings/{booking_id}")
86def get_booking(booking_id: int, authorization: Optional[str] = Header(None)):
87    get_current_user(authorization)
88    if booking_id not in bookings:
89        raise HTTPException(status_code=404, detail="Booking not found")
90    return bookings[booking_id]

1fastapi
2uvicorn