Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4from datetime import datetime, timedelta
5import hashlib
6import secrets
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12desks = {}
13reservations = {}
14desk_id_counter = 1
15reservation_id_counter = 1
16user_id_counter = 1
17 
18class SignupRequest(BaseModel):
19    username: str
20    password: str
21 
22class LoginRequest(BaseModel):
23    username: str
24    password: str
25 
26class DeskCreate(BaseModel):
27    desk_type: str
28    price_per_hour: float
29    window_view: bool
30    location: str
31    amenities: list[str] = []
32 
33class ReservationCreate(BaseModel):
34    desk_id: int
35    date: str
36    hour: int
37 
38def get_current_user(authorization: Optional[str] = Header(None)):
39    if not authorization:
40        raise HTTPException(status_code=401, detail="Missing auth header")
41    token = authorization.replace("Bearer ", "")
42    if token not in tokens:
43        raise HTTPException(status_code=401, detail="Invalid token")
44    return tokens[token]
45 
46@app.post("/signup")
47def signup(req: SignupRequest):
48    global user_id_counter
49    if req.username in users:
50        raise HTTPException(status_code=400, detail="User exists")
51    user_id = user_id_counter
52    user_id_counter += 1
53    users[req.username] = {"id": user_id, "password": req.password}
54    return {"id": user_id, "username": req.username}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    if req.username not in users or users[req.username]["password"] != req.password:
59        raise HTTPException(status_code=401, detail="Invalid credentials")
60    token = secrets.token_hex(16)
61    tokens[token] = req.username
62    return {"token": token}
63 
64@app.post("/desks")
65def create_desk(desk: DeskCreate, authorization: Optional[str] = Header(None)):
66    get_current_user(authorization)
67    global desk_id_counter
68    desk_id = desk_id_counter
69    desk_id_counter += 1
70    desks[desk_id] = {
71        "id": desk_id,
72        "desk_type": desk.desk_type,
73        "price_per_hour": desk.price_per_hour,
74        "window_view": desk.window_view,
75        "location": desk.location,
76        "amenities": desk.amenities
77    }
78    return desks[desk_id]
79 
80@app.get("/desks/{desk_id}")
81def get_desk(desk_id: int, authorization: Optional[str] = Header(None)):
82    get_current_user(authorization)
83    if desk_id not in desks:
84        raise HTTPException(status_code=404, detail="Desk not found")
85    return desks[desk_id]
86 
87@app.get("/desks")
88def list_desks(
89    location: Optional[str] = None,
90    date: Optional[str] = None,
91    amenity: Optional[str] = None,
92    desk_type: Optional[str] = None,
93    authorization: Optional[str] = Header(None)
94):
95    get_current_user(authorization)
96    result = []
97    for desk in desks.values():
98        if location and desk["location"] != location:
99            continue
100        if desk_type and desk["desk_type"] != desk_type:
101            continue
102        if amenity and amenity not in desk["amenities"]:
103            continue
104        result.append({
105            "id": desk["id"],
106            "desk_type": desk["desk_type"],
107            "price_per_hour": desk["price_per_hour"],
108            "window_view": desk["window_view"],
109            "location": desk["location"],
110            "amenities": desk["amenities"]
111        })
112    return result
113 
114@app.post("/reservations")
115def create_reservation(res: ReservationCreate, authorization: Optional[str] = Header(None)):
116    username = get_current_user(authorization)
117    global reservation_id_counter
118    if res.desk_id not in desks:
119        raise HTTPException(status_code=404, detail="Desk not found")
120    rid = reservation_id_counter
121    reservation_id_counter += 1
122    reservations[rid] = {
123        "id": rid,
124        "desk_id": res.desk_id,
125        "date": res.date,
126        "hour": res.hour,
127        "username": username
128    }
129    return reservations[rid]
130 
131@app.get("/reservations/{reservation_id}")
132def get_reservation(reservation_id: int, authorization: Optional[str] = Header(None)):
133    get_current_user(authorization)
134    if reservation_id not in reservations:
135        raise HTTPException(status_code=404, detail="Reservation not found")
136    return reservations[reservation_id]

1fastapi
2uvicorn