Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9spaces = {}
10amenities = {}
11reservations = {}
12next_user_id = 1
13next_space_id = 1
14next_amenity_id = 1
15next_reservation_id = 1
16 
17def get_current_user(authorization: Optional[str] = Header(None)):
18    if not authorization:
19        raise HTTPException(status_code=401, detail="Missing auth header")
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 next_user_id
28    for u in users.values():
29        if u["username"] == username:
30            raise HTTPException(status_code=400, detail="Username taken")
31    user_id = next_user_id
32    next_user_id += 1
33    users[user_id] = {"id": user_id, "username": username, "password": password}
34    return {"user_id": user_id}
35 
36@app.post("/login")
37def login(username: str, password: str):
38    for u in users.values():
39        if u["username"] == username and u["password"] == password:
40            token = secrets.token_hex(16)
41            tokens[token] = u["id"]
42            return {"token": token}
43    raise HTTPException(status_code=401, detail="Invalid credentials")
44 
45@app.post("/spaces")
46def create_space(name: str, capacity: int, authorization: Optional[str] = Header(None)):
47    get_current_user(authorization)
48    global next_space_id
49    space_id = next_space_id
50    next_space_id += 1
51    spaces[space_id] = {"id": space_id, "name": name, "capacity": capacity}
52    return spaces[space_id]
53 
54@app.get("/spaces/{space_id}")
55def get_space(space_id: int, authorization: Optional[str] = Header(None)):
56    get_current_user(authorization)
57    if space_id not in spaces:
58        raise HTTPException(status_code=404, detail="Space not found")
59    return spaces[space_id]
60 
61@app.post("/amenities")
62def create_amenity(space_id: int, name: str, authorization: Optional[str] = Header(None)):
63    get_current_user(authorization)
64    if space_id not in spaces:
65        raise HTTPException(status_code=404, detail="Space not found")
66    global next_amenity_id
67    amenity_id = next_amenity_id
68    next_amenity_id += 1
69    amenities[amenity_id] = {"id": amenity_id, "space_id": space_id, "name": name}
70    return amenities[amenity_id]
71 
72@app.get("/amenities/{amenity_id}")
73def get_amenity(amenity_id: int, authorization: Optional[str] = Header(None)):
74    get_current_user(authorization)
75    if amenity_id not in amenities:
76        raise HTTPException(status_code=404, detail="Amenity not found")
77    return amenities[amenity_id]
78 
79@app.post("/reservations")
80def create_reservation(space_id: int, user_id: int, authorization: Optional[str] = Header(None)):
81    current_user = get_current_user(authorization)
82    if space_id not in spaces:
83        raise HTTPException(status_code=404, detail="Space not found")
84    if user_id != current_user:
85        raise HTTPException(status_code=403, detail="Cannot reserve for other users")
86    space = spaces[space_id]
87    current_reservations = sum(1 for r in reservations.values() if r["space_id"] == space_id)
88    if current_reservations >= space["capacity"]:
89        raise HTTPException(status_code=400, detail="Space at capacity")
90    global next_reservation_id
91    reservation_id = next_reservation_id
92    next_reservation_id += 1
93    reservations[reservation_id] = {"id": reservation_id, "space_id": space_id, "user_id": user_id}
94    return reservations[reservation_id]
95 
96@app.get("/reservations/{reservation_id}")
97def get_reservation(reservation_id: int, authorization: Optional[str] = Header(None)):
98    get_current_user(authorization)
99    if reservation_id not in reservations:
100        raise HTTPException(status_code=404, detail="Reservation not found")
101    return reservations[reservation_id]
102 
103@app.get("/spaces/{space_id}/amenities")
104def list_amenities_by_space(space_id: int, authorization: Optional[str] = Header(None)):
105    get_current_user(authorization)
106    if space_id not in spaces:
107        raise HTTPException(status_code=404, detail="Space not found")
108    return [a for a in amenities.values() if a["space_id"] == space_id]

1fastapi
2uvicorn