Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List, Dict
4import secrets
5import uuid
6 
7app = FastAPI()
8 
9# In-memory stores
10users = {}
11guests = {}
12tables = {}
13rsvps = {}
14seating_charts = {}
15tokens = {}
16next_user_id = 1
17next_guest_id = 1
18next_table_id = 1
19next_rsvp_id = 1
20next_chart_id = 1
21 
22# Auth helpers
23def get_user_id_from_token(authorization: str = Header(None)):
24    if not authorization:
25        raise HTTPException(status_code=401, detail="Missing auth header")
26    token = authorization.replace("Bearer ", "")
27    if token not in tokens:
28        raise HTTPException(status_code=401, detail="Invalid token")
29    return tokens[token]
30 
31# Models
32class SignupRequest(BaseModel):
33    username: str
34    password: str
35 
36class LoginRequest(BaseModel):
37    username: str
38    password: str
39 
40class GuestCreate(BaseModel):
41    name: str
42    meal_preference: Optional[str] = None
43 
44class GuestUpdate(BaseModel):
45    name: Optional[str] = None
46    meal_preference: Optional[str] = None
47 
48class TableCreate(BaseModel):
49    table_number: int
50    capacity: int
51 
52class RSVPCreate(BaseModel):
53    guest_id: int
54    status: str  # "accepted", "declined", "pending"
55 
56class SeatingChartCreate(BaseModel):
57    name: str
58    table_assignments: Dict[int, List[int]]  # table_id -> list of guest_ids
59 
60# --- Auth Endpoints ---
61@app.post("/signup")
62def signup(req: SignupRequest):
63    global next_user_id
64    for user in users.values():
65        if user["username"] == req.username:
66            raise HTTPException(status_code=400, detail="Username taken")
67    user_id = next_user_id
68    next_user_id += 1
69    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
70    token = secrets.token_hex(32)
71    tokens[token] = user_id
72    return {"user_id": user_id, "token": token}
73 
74@app.post("/login")
75def login(req: LoginRequest):
76    for user in users.values():
77        if user["username"] == req.username and user["password"] == req.password:
78            token = secrets.token_hex(32)
79            tokens[token] = user["id"]
80            return {"token": token}
81    raise HTTPException(status_code=401, detail="Invalid credentials")
82 
83# --- Guest Endpoints ---
84@app.get("/guests/{guest_id}")
85def get_guest(guest_id: int, authorization: str = Header(None)):
86    get_user_id_from_token(authorization)
87    if guest_id not in guests:
88        raise HTTPException(status_code=404, detail="Guest not found")
89    return guests[guest_id]
90 
91@app.post("/guests")
92def create_guest(guest: GuestCreate, authorization: str = Header(None)):
93    get_user_id_from_token(authorization)
94    global next_guest_id
95    guest_id = next_guest_id
96    next_guest_id += 1
97    guests[guest_id] = {"id": guest_id, "name": guest.name, "meal_preference": guest.meal_preference}
98    return guests[guest_id]
99 
100@app.get("/guests")
101def list_guests(authorization: str = Header(None)):
102    get_user_id_from_token(authorization)
103    return list(guests.values())
104 
105# --- Table Endpoints ---
106@app.get("/tables/{table_id}")
107def get_table(table_id: int, authorization: str = Header(None)):
108    get_user_id_from_token(authorization)
109    if table_id not in tables:
110        raise HTTPException(status_code=404, detail="Table not found")
111    return tables[table_id]
112 
113@app.post("/tables")
114def create_table(table: TableCreate, authorization: str = Header(None)):
115    get_user_id_from_token(authorization)
116    global next_table_id
117    table_id = next_table_id
118    next_table_id += 1
119    tables[table_id] = {"id": table_id, "table_number": table.table_number, "capacity": table.capacity, "assigned_guests": []}
120    return tables[table_id]
121 
122@app.get("/tables")
123def list_tables(authorization: str = Header(None)):
124    get_user_id_from_token(authorization)
125    return list(tables.values())
126 
127# --- RSVP Endpoints ---
128@app.get("/rsvps/{rsvp_id}")
129def get_rsvp(rsvp_id: int, authorization: str = Header(None)):
130    get_user_id_from_token(authorization)
131    if rsvp_id not in rsvps:
132        raise HTTPException(status_code=404, detail="RSVP not found")
133    return rsvps[rsvp_id]
134 
135@app.post("/rsvps")
136def create_rsvp(rsvp: RSVPCreate, authorization: str = Header(None)):
137    get_user_id_from_token(authorization)
138    global next_rsvp_id
139    if rsvp.guest_id not in guests:
140        raise HTTPException(status_code=404, detail="Guest not found")
141    rsvp_id = next_rsvp_id
142    next_rsvp_id += 1
143    rsvps[rsvp_id] = {"id": rsvp_id, "guest_id": rsvp.guest_id, "status": rsvp.status}
144    guests[rsvp.guest_id]["rsvp_status"] = rsvp.status
145    return rsvps[rsvp_id]
146 
147@app.get("/rsvps")
148def list_rsvps(authorization: str = Header(None)):
149    get_user_id_from_token(authorization)
150    return list(rsvps.values())
151 
152# --- Seating Chart Endpoints ---
153@app.get("/seating_charts/{chart_id}")
154def get_seating_chart(chart_id: int, authorization: str = Header(None)):
155    get_user_id_from_token(authorization)
156    if chart_id not in seating_charts:
157        raise HTTPException(status_code=404, detail="Seating chart not found")
158    return seating_charts[chart_id]
159 
160@app.post("/seating_charts")
161def create_seating_chart(chart: SeatingChartCreate, authorization: str = Header(None)):
162    get_user_id_from_token(authorization)
163    global next_chart_id
164    # Validate all tables and guests exist
165    for table_id in chart.table_assignments:
166        if table_id not in tables:
167            raise HTTPException(status_code=404, detail=f"Table {table_id} not found")
168        for guest_id in chart.table_assignments[table_id]:
169            if guest_id not in guests:
170                raise HTTPException(status_code=404, detail=f"Guest {guest_id} not found")
171    chart_id = next_chart_id
172    next_chart_id += 1
173    seating_charts[chart_id] = {"id": chart_id, "name": chart.name, "table_assignments": chart.table_assignments}
174    return seating_charts[chart_id]
175 
176@app.get("/seating_charts")
177def list_seating_charts(authorization: str = Header(None)):
178    get_user_id_from_token(authorization)
179    return list(seating_charts.values())

1fastapi
2uvicorn