Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import random
5import string
6 
7app = FastAPI()
8 
9# In-memory stores
10users = {}
11tokens = {}
12menu_items = {
13    1: {"id": 1, "name": "Margherita Pizza", "category": "Pizza", "price": 12.99, "dietary_tags": ["vegetarian"]},
14    2: {"id": 2, "name": "Pepperoni Pizza", "category": "Pizza", "price": 14.99, "dietary_tags": []},
15    3: {"id": 3, "name": "Caesar Salad", "category": "Salad", "price": 9.99, "dietary_tags": ["vegetarian", "gluten-free"]},
16    4: {"id": 4, "name": "Grilled Chicken", "category": "Main", "price": 16.99, "dietary_tags": ["gluten-free"]},
17    5: {"id": 5, "name": "Veggie Burger", "category": "Burger", "price": 13.99, "dietary_tags": ["vegetarian", "vegan"]},
18    6: {"id": 6, "name": "French Fries", "category": "Sides", "price": 5.99, "dietary_tags": ["vegetarian", "vegan", "gluten-free"]},
19    7: {"id": 7, "name": "Chocolate Cake", "category": "Dessert", "price": 7.99, "dietary_tags": ["vegetarian"]},
20    8: {"id": 8, "name": "Iced Tea", "category": "Drinks", "price": 2.99, "dietary_tags": ["vegetarian", "vegan", "gluten-free"]},
21}
22orders = {}
23next_user_id = 1
24next_order_id = 1
25 
26class SignupRequest(BaseModel):
27    username: str
28    password: str
29 
30class LoginRequest(BaseModel):
31    username: str
32    password: str
33 
34class OrderRequest(BaseModel):
35    table_number: int
36    item_ids: List[int]
37 
38def generate_token():
39    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
40 
41def get_current_user(authorization: str = Header(None)):
42    if not authorization:
43        raise HTTPException(status_code=401, detail="Missing auth token")
44    token = authorization.replace("Bearer ", "")
45    if token not in tokens:
46        raise HTTPException(status_code=401, detail="Invalid auth token")
47    return tokens[token]
48 
49@app.post("/signup")
50def signup(req: SignupRequest):
51    global next_user_id
52    user_id = next_user_id
53    next_user_id += 1
54    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
55    token = generate_token()
56    tokens[token] = user_id
57    return {"user_id": user_id, "token": token}
58 
59@app.post("/login")
60def login(req: LoginRequest):
61    for uid, user in users.items():
62        if user["username"] == req.username and user["password"] == req.password:
63            token = generate_token()
64            tokens[token] = uid
65            return {"user_id": uid, "token": token}
66    raise HTTPException(status_code=401, detail="Invalid credentials")
67 
68@app.get("/menu")
69def get_menu():
70    return list(menu_items.values())
71 
72@app.get("/menu/{item_id}")
73def get_menu_item(item_id: int):
74    if item_id not in menu_items:
75        raise HTTPException(status_code=404, detail="Menu item not found")
76    return menu_items[item_id]
77 
78@app.post("/order")
79def create_order(req: OrderRequest, authorization: str = Header(None)):
80    get_current_user(authorization)
81    global next_order_id
82    order_id = next_order_id
83    next_order_id += 1
84    items = []
85    for item_id in req.item_ids:
86        if item_id not in menu_items:
87            raise HTTPException(status_code=400, detail=f"Item {item_id} not found")
88        items.append(menu_items[item_id])
89    prep_time = len(items) * 5 + random.randint(0, 10)
90    orders[order_id] = {
91        "id": order_id,
92        "table_number": req.table_number,
93        "item_ids": req.item_ids,
94        "estimated_prep_time_minutes": prep_time
95    }
96    return orders[order_id]
97 
98@app.get("/order/{order_id}")
99def get_order(order_id: int, authorization: str = Header(None)):
100    get_current_user(authorization)
101    if order_id not in orders:
102        raise HTTPException(status_code=404, detail="Order not found")
103    return orders[order_id]

1fastapi
2uvicorn