Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict, Any
4import secrets
5import time
6 
7app = FastAPI()
8 
9orders = {}
10users = {}
11tokens = {}
12user_counter = 0
13order_counter = 0
14 
15class SignupRequest(BaseModel):
16    username: str
17    password: str
18 
19class LoginRequest(BaseModel):
20    username: str
21    password: str
22 
23class OrderItem(BaseModel):
24    name: str
25    quantity: int
26 
27class OrderCreate(BaseModel):
28    items: list[OrderItem]
29    delivery_address: str
30    special_instructions: Optional[str] = None
31 
32class OrderUpdate(BaseModel):
33    items: Optional[list[OrderItem]] = None
34    delivery_address: Optional[str] = None
35    special_instructions: Optional[str] = None
36    status: Optional[str] = None
37 
38def get_current_user(authorization: Optional[str] = Header(None)):
39    if not authorization:
40        raise HTTPException(status_code=401, detail="Missing authorization header")
41    token = authorization.replace("Bearer ", "")
42    if token not in tokens:
43        raise HTTPException(status_code=401, detail="Invalid token")
44    return tokens[token]
45 
46@app.post("/signup")
47def signup(req: SignupRequest):
48    global user_counter
49    if req.username in users:
50        raise HTTPException(status_code=400, detail="Username already exists")
51    user_counter += 1
52    users[req.username] = {"id": user_counter, "username": req.username, "password": req.password}
53    return {"id": user_counter, "username": req.username}
54 
55@app.post("/login")
56def login(req: LoginRequest):
57    if req.username not in users or users[req.username]["password"] != req.password:
58        raise HTTPException(status_code=401, detail="Invalid credentials")
59    token = secrets.token_hex(32)
60    tokens[token] = req.username
61    return {"token": token}
62 
63@app.get("/orders/{order_id}")
64def get_order(order_id: int, authorization: Optional[str] = Header(None)):
65    get_current_user(authorization)
66    if order_id not in orders:
67        raise HTTPException(status_code=404, detail="Order not found")
68    return orders[order_id]
69 
70@app.post("/orders")
71def create_order(req: OrderCreate, authorization: Optional[str] = Header(None)):
72    global order_counter
73    get_current_user(authorization)
74    order_counter += 1
75    order = {
76        "id": order_counter,
77        "items": [item.dict() for item in req.items],
78        "delivery_address": req.delivery_address,
79        "special_instructions": req.special_instructions,
80        "status": "pending",
81        "created_at": time.time()
82    }
83    orders[order_counter] = order
84    return order
85 
86@app.put("/orders/{order_id}")
87def update_order(order_id: int, req: OrderUpdate, authorization: Optional[str] = Header(None)):
88    get_current_user(authorization)
89    if order_id not in orders:
90        raise HTTPException(status_code=404, detail="Order not found")
91    
92    order = orders[order_id]
93    
94    if req.items is not None:
95        order["items"] = [item.dict() for item in req.items]
96    if req.delivery_address is not None:
97        order["delivery_address"] = req.delivery_address
98    if req.special_instructions is not None:
99        order["special_instructions"] = req.special_instructions
100    if req.status is not None:
101        allowed_statuses = ["pending", "confirmed", "preparing", "out_for_delivery", "delivered", "cancelled"]
102        if req.status not in allowed_statuses:
103            raise HTTPException(status_code=400, detail=f"Invalid status. Allowed: {allowed_statuses}")
104        order["status"] = req.status
105    
106    orders[order_id] = order
107    return order

1fastapi
2uvicorn