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 
9menu = {
10    1: {"name": "Burger", "price": 9.99, "description": "Juicy beef burger", "category": "Main"},
11    2: {"name": "Fries", "price": 3.99, "description": "Crispy golden fries", "category": "Sides"},
12    3: {"name": "Cola", "price": 1.99, "description": "Refreshing cola", "category": "Drinks"},
13    4: {"name": "Salad", "price": 7.99, "description": "Fresh garden salad", "category": "Main"},
14    5: {"name": "Ice Cream", "price": 4.99, "description": "Vanilla ice cream", "category": "Desserts"},
15}
16 
17orders = {}
18order_id_counter = 1
19 
20users = {}
21user_id_counter = 1
22tokens = {}  # token -> user_id
23 
24class SignupRequest(BaseModel):
25    username: str
26    password: str
27 
28class LoginRequest(BaseModel):
29    username: str
30    password: str
31 
32class OrderCreate(BaseModel):
33    table_number: int
34    item_ids: List[int]
35 
36@app.get("/menu")
37def get_menu():
38    categories = {}
39    for item_id, item in menu.items():
40        cat = item["category"]
41        if cat not in categories:
42            categories[cat] = []
43        categories[cat].append({
44            "id": item_id,
45            "name": item["name"],
46            "price": item["price"],
47            "description": item["description"]
48        })
49    return categories
50 
51@app.post("/signup")
52def signup(req: SignupRequest):
53    global user_id_counter
54    for u in users.values():
55        if u["username"] == req.username:
56            raise HTTPException(status_code=400, detail="User exists")
57    user_id = user_id_counter
58    user_id_counter += 1
59    users[user_id] = {
60        "id": user_id,
61        "username": req.username,
62        "password": hashlib.sha256(req.password.encode()).hexdigest()
63    }
64    token = secrets.token_hex(16)
65    tokens[token] = user_id
66    return {"user_id": user_id, "token": token}
67 
68@app.post("/login")
69def login(req: LoginRequest):
70    for uid, u in users.items():
71        if u["username"] == req.username and u["password"] == hashlib.sha256(req.password.encode()).hexdigest():
72            token = secrets.token_hex(16)
73            tokens[token] = uid
74            return {"token": token}
75    raise HTTPException(status_code=401, detail="Invalid credentials")
76 
77def get_current_user(authorization: Optional[str] = Header(None)):
78    if not authorization:
79        raise HTTPException(status_code=401, detail="No auth")
80    token = authorization.replace("Bearer ", "")
81    if token not in tokens:
82        raise HTTPException(status_code=401, detail="Invalid token")
83    return tokens[token]
84 
85@app.get("/menu/{item_id}")
86def get_menu_item(item_id: int):
87    if item_id not in menu:
88        raise HTTPException(status_code=404, detail="Not found")
89    return menu[item_id]
90 
91@app.post("/orders")
92def create_order(order: OrderCreate, authorization: Optional[str] = Header(None)):
93    get_current_user(authorization)
94    global order_id_counter
95    order_id = order_id_counter
96    order_id_counter += 1
97    items = []
98    for item_id in order.item_ids:
99        if item_id not in menu:
100            raise HTTPException(status_code=400, detail=f"Item {item_id} not found")
101        items.append(menu[item_id])
102    total = sum(item["price"] for item in items)
103    orders[order_id] = {
104        "id": order_id,
105        "table_number": order.table_number,
106        "items": items,
107        "total": total
108    }
109    return orders[order_id]
110 
111@app.get("/orders/{order_id}")
112def get_order(order_id: int, authorization: Optional[str] = Header(None)):
113    get_current_user(authorization)
114    if order_id not in orders:
115        raise HTTPException(status_code=404, detail="Not found")
116    return orders[order_id]

1fastapi
2uvicorn