Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9spaces = {}
10bookings = {}
11next_user_id = 1
12next_space_id = 1
13next_booking_id = 1
14admin_id = None
15 
16def get_current_user(authorization: Optional[str] = Header(None)):
17    if not authorization:
18        raise HTTPException(status_code=401, detail="Missing auth header")
19    token = authorization.replace("Bearer ", "")
20    if token not in tokens:
21        raise HTTPException(status_code=401, detail="Invalid token")
22    return tokens[token]
23 
24@app.post("/signup")
25def signup(username: str, password: str):
26    global next_user_id, admin_id
27    user_id = next_user_id
28    next_user_id += 1
29    users[user_id] = {"id": user_id, "username": username, "password": password, "role": "member"}
30    if admin_id is None:
31        users[user_id]["role"] = "admin"
32        admin_id = user_id
33    return {"user_id": user_id, "username": username, "role": users[user_id]["role"]}
34 
35@app.post("/login")
36def login(username: str, password: str):
37    for uid, u in users.items():
38        if u["username"] == username and u["password"] == password:
39            token = secrets.token_hex(16)
40            tokens[token] = uid
41            return {"token": token}
42    raise HTTPException(status_code=401, detail="Invalid credentials")
43 
44@app.get("/users/{user_id}")
45def get_user(user_id: int):
46    if user_id not in users:
47        raise HTTPException(status_code=404, detail="User not found")
48    return users[user_id]
49 
50@app.get("/spaces/{space_id}")
51def get_space(space_id: int):
52    if space_id not in spaces:
53        raise HTTPException(status_code=404, detail="Space not found")
54    return spaces[space_id]
55 
56@app.get("/bookings/{booking_id}")
57def get_booking(booking_id: int):
58    if booking_id not in bookings:
59        raise HTTPException(status_code=404, detail="Booking not found")
60    return bookings[booking_id]
61 
62@app.post("/spaces")
63def create_space(name: str, authorization: Optional[str] = Header(None)):
64    user_id = get_current_user(authorization)
65    global next_space_id
66    space_id = next_space_id
67    next_space_id += 1
68    spaces[space_id] = {"id": space_id, "name": name, "created_by": user_id}
69    return spaces[space_id]
70 
71@app.post("/bookings")
72def create_booking(room: str, date: str, amenities: str, space_id: int, authorization: Optional[str] = Header(None)):
73    user_id = get_current_user(authorization)
74    if space_id not in spaces:
75        raise HTTPException(status_code=404, detail="Space not found")
76    global next_booking_id
77    booking_id = next_booking_id
78    next_booking_id += 1
79    bookings[booking_id] = {
80        "id": booking_id,
81        "room": room,
82        "date": date,
83        "amenities": amenities,
84        "space_id": space_id,
85        "user_id": user_id
86    }
87    return bookings[booking_id]
88 
89@app.post("/space/{space_id}/promote")
90def promote_to_host(space_id: int, user_id: int, authorization: Optional[str] = Header(None)):
91    requester_id = get_current_user(authorization)
92    if users[requester_id]["role"] != "admin":
93        raise HTTPException(status_code=403, detail="Only admin can promote")
94    if user_id not in users:
95        raise HTTPException(status_code=404, detail="User not found")
96    users[user_id]["role"] = "host"
97    return {"message": f"User {user_id} promoted to host"}

1fastapi
2uvicorn