Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8# In-memory stores
9users = {}
10bookings = {}
11tokens = {}
12user_id_counter = 1
13booking_id_counter = 1
14 
15# Auth helper
16def get_user_id(authorization: str = Header(None)):
17    if not authorization or not authorization.startswith("Bearer "):
18        raise HTTPException(status_code=401, detail="Invalid auth")
19    token = authorization.split(" ")[1]
20    if token not in tokens:
21        raise HTTPException(status_code=401, detail="Invalid token")
22    return tokens[token]
23 
24# Models
25class SignupRequest(BaseModel):
26    username: str
27    password: str
28 
29class LoginRequest(BaseModel):
30    username: str
31    password: str
32 
33class BookingCreate(BaseModel):
34    member_name: str
35    desk_number: int
36    time_slot: str
37    membership_tier: str
38    access_badge: str
39 
40class BookingUpdate(BaseModel):
41    member_name: Optional[str] = None
42    desk_number: Optional[int] = None
43    time_slot: Optional[str] = None
44    membership_tier: Optional[str] = None
45    access_badge: Optional[str] = None
46 
47# Auth endpoints
48@app.post("/signup")
49def signup(req: SignupRequest):
50    global user_id_counter
51    for u in users.values():
52        if u["username"] == req.username:
53            raise HTTPException(status_code=400, detail="User exists")
54    user_id = user_id_counter
55    user_id_counter += 1
56    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
57    token = secrets.token_hex(16)
58    tokens[token] = user_id
59    return {"user_id": user_id, "token": token}
60 
61@app.post("/login")
62def login(req: LoginRequest):
63    for u in users.values():
64        if u["username"] == req.username and u["password"] == req.password:
65            token = secrets.token_hex(16)
66            tokens[token] = u["id"]
67            return {"token": token}
68    raise HTTPException(status_code=401, detail="Invalid credentials")
69 
70# Booking endpoints
71@app.get("/bookings/{booking_id}")
72def get_booking(booking_id: int, authorization: str = Header(None)):
73    get_user_id(authorization)
74    if booking_id not in bookings:
75        raise HTTPException(status_code=404, detail="Booking not found")
76    return bookings[booking_id]
77 
78@app.post("/bookings")
79def create_booking(booking: BookingCreate, authorization: str = Header(None)):
80    global booking_id_counter
81    get_user_id(authorization)
82    booking_id = booking_id_counter
83    booking_id_counter += 1
84    bookings[booking_id] = {
85        "id": booking_id,
86        "member_name": booking.member_name,
87        "desk_number": booking.desk_number,
88        "time_slot": booking.time_slot,
89        "membership_tier": booking.membership_tier,
90        "access_badge": booking.access_badge
91    }
92    return bookings[booking_id]
93 
94@app.patch("/bookings/{booking_id}")
95def update_booking(booking_id: int, update: BookingUpdate, authorization: str = Header(None)):
96    get_user_id(authorization)
97    if booking_id not in bookings:
98        raise HTTPException(status_code=404, detail="Booking not found")
99    b = bookings[booking_id]
100    if update.member_name is not None:
101        b["member_name"] = update.member_name
102    if update.desk_number is not None:
103        b["desk_number"] = update.desk_number
104    if update.time_slot is not None:
105        b["time_slot"] = update.time_slot
106    if update.membership_tier is not None:
107        b["membership_tier"] = update.membership_tier
108    if update.access_badge is not None:
109        b["access_badge"] = update.access_badge
110    return b

1fastapi
2uvicorn