Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import hashlib
5import secrets
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11restaurants = {
12    1: {
13        "id": 1,
14        "name": "Burger Palace",
15        "menu": [
16            {"id": 1, "name": "Classic Burger", "price": 9.99, "popularity": 95, "dietary": ["meat"]},
17            {"id": 2, "name": "Veggie Burger", "price": 8.99, "popularity": 80, "dietary": ["vegetarian"]},
18            {"id": 3, "name": "Chicken Sandwich", "price": 10.99, "popularity": 88, "dietary": ["meat"]},
19            {"id": 4, "name": "Fries", "price": 3.99, "popularity": 75, "dietary": ["vegan"]},
20            {"id": 5, "name": "Onion Rings", "price": 4.99, "popularity": 70, "dietary": ["vegetarian"]}
21        ]
22    },
23    2: {
24        "id": 2,
25        "name": "Pizza Planet",
26        "menu": [
27            {"id": 6, "name": "Pepperoni Pizza", "price": 12.99, "popularity": 92, "dietary": ["meat"]},
28            {"id": 7, "name": "Margherita Pizza", "price": 10.99, "popularity": 85, "dietary": ["vegetarian"]},
29            {"id": 8, "name": "Veggie Supreme", "price": 11.99, "popularity": 78, "dietary": ["vegan"]},
30            {"id": 9, "name": "Garlic Bread", "price": 4.99, "popularity": 72, "dietary": ["vegetarian"]}
31        ]
32    }
33}
34orders = {}
35next_order_id = 1
36next_user_id = 1
37 
38class SignupRequest(BaseModel):
39    username: str
40    password: str
41 
42class LoginRequest(BaseModel):
43    username: str
44    password: str
45 
46class OrderItem(BaseModel):
47    dish_id: int
48    quantity: int = 1
49    modifications: Optional[str] = None
50 
51class OrderRequest(BaseModel):
52    restaurant_id: int
53    items: List[OrderItem]
54 
55def verify_token(authorization: str = Header(None)):
56    if not authorization:
57        raise HTTPException(status_code=401, detail="Missing auth header")
58    token = authorization.replace("Bearer ", "")
59    if token not in tokens:
60        raise HTTPException(status_code=401, detail="Invalid token")
61    return tokens[token]
62 
63@app.post("/signup")
64def signup(req: SignupRequest):
65    global next_user_id
66    if req.username in users:
67        raise HTTPException(status_code=400, detail="Username taken")
68    user_id = next_user_id
69    next_user_id += 1
70    users[req.username] = {"id": user_id, "password": req.password}
71    token = secrets.token_hex(16)
72    tokens[token] = req.username
73    return {"user_id": user_id, "token": token}
74 
75@app.post("/login")
76def login(req: LoginRequest):
77    if req.username not in users or users[req.username]["password"] != req.password:
78        raise HTTPException(status_code=401, detail="Invalid credentials")
79    token = secrets.token_hex(16)
80    tokens[token] = req.username
81    return {"token": token}
82 
83@app.get("/restaurants/{restaurant_id}")
84def get_restaurant(restaurant_id: int):
85    if restaurant_id not in restaurants:
86        raise HTTPException(status_code=404, detail="Restaurant not found")
87    return restaurants[restaurant_id]
88 
89@app.get("/restaurants/{restaurant_id}/recommend")
90def recommend_dishes(restaurant_id: int, dietary: Optional[str] = None):
91    if restaurant_id not in restaurants:
92        raise HTTPException(status_code=404, detail="Restaurant not found")
93    menu = restaurants[restaurant_id]["menu"]
94    if dietary:
95        diet_list = [d.strip().lower() for d in dietary.split(",")]
96        menu = [d for d in menu if any(diet in d["dietary"] for diet in diet_list)]
97    menu = sorted(menu, key=lambda x: x["popularity"], reverse=True)
98    return {"restaurant_id": restaurant_id, "recommended": menu}
99 
100@app.post("/orders/place")
101def place_order(order: OrderRequest, authorization: str = Header(None)):
102    user = verify_token(authorization)
103    if order.restaurant_id not in restaurants:
104        raise HTTPException(status_code=404, detail="Restaurant not found")
105    global next_order_id
106    order_id = next_order_id
107    next_order_id += 1
108    order_data = {
109        "id": order_id,
110        "user": user,
111        "restaurant_id": order.restaurant_id,
112        "items": [item.dict() for item in order.items],
113        "status": "confirmed"
114    }
115    orders[order_id] = order_data
116    return {"order_id": order_id, "status": "confirmed"}
117 
118@app.get("/orders/{order_id}")
119def get_order(order_id: int, authorization: str = Header(None)):
120    verify_token(authorization)
121    if order_id not in orders:
122        raise HTTPException(status_code=404, detail="Order not found")
123    return orders[order_id]

1fastapi
2uvicorn