Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import random
4import string
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10reservations = {}
11waitlist = {}
12cancellations = {}
13reservation_id_counter = 1
14user_id_counter = 1
15 
16def generate_token():
17    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
18 
19def get_current_user(authorization: str = Header(None)):
20    if not authorization:
21        raise HTTPException(status_code=401, detail="Missing auth header")
22    token = authorization.replace("Bearer ", "")
23    for uid, t in tokens.items():
24        if t == token:
25            return uid
26    raise HTTPException(status_code=401, detail="Invalid token")
27 
28@app.post("/signup")
29def signup(username: str, password: str):
30    global user_id_counter
31    for u in users.values():
32        if u["username"] == username:
33            raise HTTPException(status_code=400, detail="Username taken")
34    uid = user_id_counter
35    user_id_counter += 1
36    users[uid] = {"id": uid, "username": username, "password": password}
37    return {"id": uid, "username": username}
38 
39@app.post("/login")
40def login(username: str, password: str):
41    for uid, u in users.items():
42        if u["username"] == username and u["password"] == password:
43            token = generate_token()
44            tokens[uid] = token
45            return {"token": token}
46    raise HTTPException(status_code=401, detail="Invalid credentials")
47 
48@app.get("/reservations/{reservation_id}")
49def get_reservation(reservation_id: int, authorization: str = Header(None)):
50    user_id = get_current_user(authorization)
51    if reservation_id not in reservations:
52        raise HTTPException(status_code=404, detail="Reservation not found")
53    return reservations[reservation_id]
54 
55@app.post("/reservations")
56def create_reservation(party_size: int, time: str, table_preference: str = "", special_requests: str = "", authorization: str = Header(None)):
57    global reservation_id_counter
58    user_id = get_current_user(authorization)
59    rid = reservation_id_counter
60    reservation_id_counter += 1
61    reservations[rid] = {
62        "id": rid,
63        "user_id": user_id,
64        "party_size": party_size,
65        "time": time,
66        "table_preference": table_preference,
67        "special_requests": special_requests,
68        "status": "confirmed"
69    }
70    return reservations[rid]
71 
72@app.put("/reservations/{reservation_id}")
73def update_reservation(reservation_id: int, party_size: Optional[int] = None, time: Optional[str] = None, table_preference: Optional[str] = None, special_requests: Optional[str] = None, authorization: str = Header(None)):
74    user_id = get_current_user(authorization)
75    if reservation_id not in reservations:
76        raise HTTPException(status_code=404, detail="Reservation not found")
77    res = reservations[reservation_id]
78    if res["user_id"] != user_id:
79        raise HTTPException(status_code=403, detail="Not your reservation")
80    if party_size is not None:
81        res["party_size"] = party_size
82    if time is not None:
83        res["time"] = time
84    if table_preference is not None:
85        res["table_preference"] = table_preference
86    if special_requests is not None:
87        res["special_requests"] = special_requests
88    return res
89 
90@app.post("/reservations/{reservation_id}/cancel")
91def cancel_reservation(reservation_id: int, authorization: str = Header(None)):
92    user_id = get_current_user(authorization)
93    if reservation_id not in reservations:
94        raise HTTPException(status_code=404, detail="Reservation not found")
95    res = reservations[reservation_id]
96    if res["user_id"] != user_id:
97        raise HTTPException(status_code=403, detail="Not your reservation")
98    res["status"] = "cancelled"
99    cancellations[reservation_id] = {
100        "reservation_id": reservation_id,
101        "user_id": user_id,
102        "cancelled_at": "now"
103    }
104    return {"message": "Cancelled", "reservation": res}
105 
106@app.post("/waitlist")
107def join_waitlist(reservation_id: int, authorization: str = Header(None)):
108    user_id = get_current_user(authorization)
109    if reservation_id not in reservations:
110        raise HTTPException(status_code=404, detail="Reservation not found")
111    if reservation_id not in waitlist:
112        waitlist[reservation_id] = []
113    waitlist[reservation_id].append(user_id)
114    return {"message": "Added to waitlist", "position": len(waitlist[reservation_id])}
115 
116@app.get("/waitlist/{reservation_id}")
117def get_waitlist(reservation_id: int, authorization: str = Header(None)):
118    get_current_user(authorization)
119    if reservation_id not in reservations:
120        raise HTTPException(status_code=404, detail="Reservation not found")
121    return {"waitlist": waitlist.get(reservation_id, [])}
122 
123@app.get("/cancellations/{reservation_id}")
124def get_cancellation(reservation_id: int, authorization: str = Header(None)):
125    get_current_user(authorization)
126    if reservation_id not in cancellations:
127        raise HTTPException(status_code=404, detail="Cancellation not found")
128    return cancellations[reservation_id]

1fastapi
2uvicorn