Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from datetime import datetime, timedelta
3from typing import Optional
4import hashlib
5import secrets
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11tables = {}
12reservations = {}
13table_id_counter = 1
14reservation_id_counter = 1
15user_id_counter = 1
16 
17# Seed some tables
18for section in ["main", "patio", "bar"]:
19    for capacity in [2, 4, 6]:
20        for _ in range(3):
21            tables[table_id_counter] = {
22                "id": table_id_counter,
23                "capacity": capacity,
24                "section": section,
25                "reserved_slots": []
26            }
27            table_id_counter += 1
28 
29def get_current_user(authorization: str = Header(None)):
30    if not authorization:
31        raise HTTPException(status_code=401, detail="Missing auth token")
32    token = authorization.replace("Bearer ", "")
33    if token not in tokens:
34        raise HTTPException(status_code=401, detail="Invalid token")
35    return tokens[token]
36 
37@app.post("/signup")
38def signup(username: str, password: str):
39    global user_id_counter
40    if any(u["username"] == username for u in users.values()):
41        raise HTTPException(status_code=400, detail="Username taken")
42    user_id = user_id_counter
43    user_id_counter += 1
44    users[user_id] = {
45        "id": user_id,
46        "username": username,
47        "password": hashlib.sha256(password.encode()).hexdigest()
48    }
49    return {"id": user_id, "username": username}
50 
51@app.post("/login")
52def login(username: str, password: str):
53    for u in users.values():
54        if u["username"] == username and u["password"] == hashlib.sha256(password.encode()).hexdigest():
55            token = secrets.token_hex(16)
56            tokens[token] = u["id"]
57            return {"token": token}
58    raise HTTPException(status_code=401, detail="Invalid credentials")
59 
60@app.get("/tables")
61def get_tables(time: str, authorization: str = Header(None)):
62    get_current_user(authorization)
63    try:
64        dt = datetime.fromisoformat(time)
65    except:
66        raise HTTPException(status_code=400, detail="Invalid time format. Use ISO format like 2025-01-01T19:00")
67    
68    available = []
69    for t in tables.values():
70        if all(abs((dt - slot).total_seconds()) >= 7200 for slot in t["reserved_slots"]):
71            available.append({
72                "id": t["id"],
73                "capacity": t["capacity"],
74                "section": t["section"]
75            })
76    return available
77 
78@app.post("/tables/book")
79def book_table(table_id: int, time: str, authorization: str = Header(None)):
80    user_id = get_current_user(authorization)
81    global reservation_id_counter
82    
83    if table_id not in tables:
84        raise HTTPException(status_code=404, detail="Table not found")
85    
86    try:
87        dt = datetime.fromisoformat(time)
88    except:
89        raise HTTPException(status_code=400, detail="Invalid time format")
90    
91    table = tables[table_id]
92    if any(abs((dt - slot).total_seconds()) < 7200 for slot in table["reserved_slots"]):
93        raise HTTPException(status_code=409, detail="Table already booked for this time")
94    
95    res_id = reservation_id_counter
96    reservation_id_counter += 1
97    
98    table["reserved_slots"].append(dt)
99    reservations[res_id] = {
100        "id": res_id,
101        "user_id": user_id,
102        "table_id": table_id,
103        "time": dt.isoformat(),
104        "confirmation": f"CONF-{res_id:04d}"
105    }
106    return {"confirmation_number": reservations[res_id]["confirmation"], "reservation_id": res_id}
107 
108@app.get("/reservations/{reservation_id}")
109def get_reservation(reservation_id: int, authorization: str = Header(None)):
110    get_current_user(authorization)
111    if reservation_id not in reservations:
112        raise HTTPException(status_code=404, detail="Reservation not found")
113    return reservations[reservation_id]
114 
115@app.post("/reservations")
116def create_reservation(table_id: int, time: str, authorization: str = Header(None)):
117    return book_table(table_id, time, authorization)
118 
119@app.get("/users/{user_id}")
120def get_user(user_id: int, authorization: str = Header(None)):
121    get_current_user(authorization)
122    if user_id not in users:
123        raise HTTPException(status_code=404, detail="User not found")
124    return {"id": users[user_id]["id"], "username": users[user_id]["username"]}

1fastapi
2uvicorn