Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import random
5import string
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11restaurants = {}
12reservations = {}
13restaurant_id_counter = 1
14reservation_id_counter = 1
15user_id_counter = 1
16 
17# Prepopulate some restaurants
18restaurants[1] = {"id": 1, "name": "Pasta Palace", "cuisine": "Italian", "rating": 4.5, "open_hours": "11:00-22:00"}
19restaurants[2] = {"id": 2, "name": "Sushi World", "cuisine": "Japanese", "rating": 4.8, "open_hours": "12:00-23:00"}
20restaurants[3] = {"id": 3, "name": "Taco Town", "cuisine": "Mexican", "rating": 4.2, "open_hours": "10:00-21:00"}
21restaurant_id_counter = 4
22 
23class SignupRequest(BaseModel):
24    username: str
25    password: str
26 
27class LoginRequest(BaseModel):
28    username: str
29    password: str
30 
31class RestaurantCreate(BaseModel):
32    name: str
33    cuisine: str
34    rating: float = 0.0
35    open_hours: str = ""
36 
37class ReservationCreate(BaseModel):
38    restaurant_id: int
39    party_size: int
40    date: str
41    special_requests: str = ""
42 
43def generate_token():
44    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
45 
46def get_current_user(authorization: str = Header(None)):
47    if not authorization:
48        raise HTTPException(status_code=401, detail="Missing auth header")
49    token = authorization.replace("Bearer ", "")
50    if token not in tokens:
51        raise HTTPException(status_code=401, detail="Invalid token")
52    return tokens[token]
53 
54@app.post("/signup")
55def signup(req: SignupRequest):
56    global user_id_counter
57    if any(u["username"] == req.username for u in users.values()):
58        raise HTTPException(status_code=400, detail="Username already exists")
59    user_id = user_id_counter
60    user_id_counter += 1
61    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
62    return {"id": user_id, "username": req.username}
63 
64@app.post("/login")
65def login(req: LoginRequest):
66    for u in users.values():
67        if u["username"] == req.username and u["password"] == req.password:
68            token = generate_token()
69            tokens[token] = u["id"]
70            return {"access_token": token, "token_type": "bearer"}
71    raise HTTPException(status_code=401, detail="Invalid credentials")
72 
73@app.post("/restaurants")
74def create_restaurant(restaurant: RestaurantCreate, authorization: str = Header(None)):
75    global restaurant_id_counter
76    user_id = get_current_user(authorization)
77    r_id = restaurant_id_counter
78    restaurant_id_counter += 1
79    restaurants[r_id] = {"id": r_id, "name": restaurant.name, "cuisine": restaurant.cuisine, "rating": restaurant.rating, "open_hours": restaurant.open_hours}
80    return restaurants[r_id]
81 
82@app.get("/restaurants")
83def list_restaurants():
84    return list(restaurants.values())
85 
86@app.get("/restaurants/{restaurant_id}")
87def get_restaurant(restaurant_id: int):
88    if restaurant_id not in restaurants:
89        raise HTTPException(status_code=404, detail="Restaurant not found")
90    return restaurants[restaurant_id]
91 
92@app.post("/reserve")
93def create_reservation(reservation: ReservationCreate, authorization: str = Header(None)):
94    global reservation_id_counter
95    user_id = get_current_user(authorization)
96    if reservation.restaurant_id not in restaurants:
97        raise HTTPException(status_code=404, detail="Restaurant not found")
98    res_id = reservation_id_counter
99    reservation_id_counter += 1
100    reservations[res_id] = {"id": res_id, "user_id": user_id, "restaurant_id": reservation.restaurant_id, "party_size": reservation.party_size, "date": reservation.date, "special_requests": reservation.special_requests}
101    return reservations[res_id]
102 
103@app.get("/reservations/{reservation_id}")
104def get_reservation(reservation_id: int, authorization: str = Header(None)):
105    user_id = get_current_user(authorization)
106    if reservation_id not in reservations:
107        raise HTTPException(status_code=404, detail="Reservation not found")
108    return reservations[reservation_id]

1fastapi
2uvicorn