Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import random
5import string
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11tables = {}
12reservations = {}
13table_id_counter = 1
14reservation_id_counter = 1
15user_id_counter = 1
16 
17# Pre-populate some tables
18for section in ["patio", "main", "bar"]:
19    for i in range(5):
20        tables[table_id_counter] = {
21            "id": table_id_counter,
22            "capacity": random.choice([2, 4, 6]),
23            "section": section,
24            "reserved_slots": []
25        }
26        table_id_counter += 1
27 
28class SignupRequest(BaseModel):
29    username: str
30    password: str
31 
32class LoginRequest(BaseModel):
33    username: str
34    password: str
35 
36class BookTableRequest(BaseModel):
37    table_id: int
38    time: str
39    party_size: int
40 
41def generate_token():
42    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
43 
44def get_current_user(authorization: Optional[str] = Header(None)):
45    if not authorization:
46        raise HTTPException(status_code=401, detail="Missing auth token")
47    token = authorization.replace("Bearer ", "")
48    if token not in tokens:
49        raise HTTPException(status_code=401, detail="Invalid token")
50    return tokens[token]
51 
52@app.post("/signup")
53def signup(req: SignupRequest):
54    global user_id_counter
55    if req.username in users:
56        raise HTTPException(status_code=400, detail="User already exists")
57    user_id = user_id_counter
58    user_id_counter += 1
59    users[req.username] = {"id": user_id, "username": req.username, "password": req.password}
60    return {"id": user_id, "username": req.username}
61 
62@app.post("/login")
63def login(req: LoginRequest):
64    user = users.get(req.username)
65    if not user or user["password"] != req.password:
66        raise HTTPException(status_code=401, detail="Invalid credentials")
67    token = generate_token()
68    tokens[token] = user["id"]
69    return {"token": token}
70 
71@app.get("/tables")
72def get_tables(time: str):
73    available = []
74    for t in tables.values():
75        if time not in t["reserved_slots"]:
76            available.append(t)
77    return available
78 
79@app.post("/tables/book")
80def book_table(req: BookTableRequest, authorization: Optional[str] = Header(None)):
81    global reservation_id_counter
82    user_id = get_current_user(authorization)
83    table = tables.get(req.table_id)
84    if not table:
85        raise HTTPException(status_code=404, detail="Table not found")
86    if req.time in table["reserved_slots"]:
87        raise HTTPException(status_code=400, detail="Table already reserved at this time")
88    if req.party_size > table["capacity"]:
89        raise HTTPException(status_code=400, detail="Party size exceeds table capacity")
90    reservation_id = reservation_id_counter
91    reservation_id_counter += 1
92    reservations[reservation_id] = {
93        "id": reservation_id,
94        "table_id": req.table_id,
95        "time": req.time,
96        "party_size": req.party_size,
97        "user_id": user_id
98    }
99    table["reserved_slots"].append(req.time)
100    return {"confirmation_number": reservation_id}
101 
102@app.get("/users/{user_id}")
103def get_user(user_id: int):
104    for u in users.values():
105        if u["id"] == user_id:
106            return u
107    raise HTTPException(status_code=404, detail="User not found")
108 
109@app.get("/reservations/{reservation_id}")
110def get_reservation(reservation_id: int):
111    res = reservations.get(reservation_id)
112    if not res:
113        raise HTTPException(status_code=404, detail="Reservation not found")
114    return res
115 
116@app.get("/tables/{table_id}")
117def get_table(table_id: int):
118    table = tables.get(table_id)
119    if not table:
120        raise HTTPException(status_code=404, detail="Table not found")
121    return table
122 
123@app.post("/users")
124def create_user(req: SignupRequest):
125    return signup(req)
126 
127@app.post("/reservations")
128def create_reservation(req: BookTableRequest, authorization: Optional[str] = Header(None)):
129    return book_table(req, authorization)
130 
131@app.post("/tables")
132def create_table(capacity: int, section: str):
133    global table_id_counter
134    table = {
135        "id": table_id_counter,
136        "capacity": capacity,
137        "section": section,
138        "reserved_slots": []
139    }
140    tables[table_id_counter] = table
141    table_id_counter += 1
142    return table

1fastapi
2uvicorn