Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import hashlib
5import uuid
6from datetime import datetime, date, time
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12reservations = {}
13reservation_id_counter = 1
14 
15class SignupRequest(BaseModel):
16    username: str
17    password: str
18 
19class LoginRequest(BaseModel):
20    username: str
21    password: str
22 
23class ReservationCreate(BaseModel):
24    party_size: int
25    date: str
26    time_slot: str
27 
28class ReservationCancel(BaseModel):
29    reason: str
30 
31def hash_password(password: str) -> str:
32    return hashlib.sha256(password.encode()).hexdigest()
33 
34def generate_token() -> str:
35    return str(uuid.uuid4())
36 
37def get_current_user(authorization: str = Header(...)):
38    if not authorization.startswith("Bearer "):
39        raise HTTPException(status_code=401, detail="Invalid authorization header")
40    token = authorization[7:]
41    if token not in tokens:
42        raise HTTPException(status_code=401, detail="Invalid token")
43    return tokens[token]
44 
45@app.post("/signup")
46def signup(req: SignupRequest):
47    if req.username in users:
48        raise HTTPException(status_code=400, detail="Username already exists")
49    users[req.username] = {"username": req.username, "password": hash_password(req.password)}
50    return {"message": "User created"}
51 
52@app.post("/login")
53def login(req: LoginRequest):
54    user = users.get(req.username)
55    if not user or user["password"] != hash_password(req.password):
56        raise HTTPException(status_code=401, detail="Invalid credentials")
57    token = generate_token()
58    tokens[token] = req.username
59    return {"token": token}
60 
61@app.post("/reservations")
62def create_reservation(req: ReservationCreate, authorization: str = Header(...)):
63    global reservation_id_counter
64    username = get_current_user(authorization)
65    res_id = reservation_id_counter
66    reservation_id_counter += 1
67    reservations[res_id] = {
68        "id": res_id,
69        "username": username,
70        "party_size": req.party_size,
71        "date": req.date,
72        "time_slot": req.time_slot,
73        "cancelled": False,
74        "cancel_reason": None
75    }
76    return {"id": res_id, "message": "Reservation created"}
77 
78@app.get("/reservations/{reservation_id}")
79def get_reservation(reservation_id: int, authorization: str = Header(...)):
80    username = get_current_user(authorization)
81    res = reservations.get(reservation_id)
82    if not res:
83        raise HTTPException(status_code=404, detail="Reservation not found")
84    return res
85 
86@app.get("/availability")
87def get_availability(date: str, party_size: int, authorization: str = Header(...)):
88    get_current_user(authorization)
89    total_slots = ["18:00", "19:00", "20:00", "21:00"]
90    available = []
91    for slot in total_slots:
92        booked = 0
93        for res in reservations.values():
94            if res["date"] == date and res["time_slot"] == slot and not res["cancelled"]:
95                booked += res["party_size"]
96        if booked + party_size <= 20:
97            available.append(slot)
98    return {"available_slots": available}
99 
100@app.delete("/reservations/{reservation_id}")
101def cancel_reservation(reservation_id: int, req: ReservationCancel, authorization: str = Header(...)):
102    username = get_current_user(authorization)
103    res = reservations.get(reservation_id)
104    if not res:
105        raise HTTPException(status_code=404, detail="Reservation not found")
106    if res["username"] != username:
107        raise HTTPException(status_code=403, detail="Not your reservation")
108    if res["cancelled"]:
109        raise HTTPException(status_code=400, detail="Already cancelled")
110    res["cancelled"] = True
111    res["cancel_reason"] = req.reason
112    return {"message": "Reservation cancelled"}

1fastapi
2uvicorn