Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import uvicorn
6 
7app = FastAPI()
8 
9users = {}
10orders = {}
11tokens = {}
12user_id_counter = 1
13order_id_counter = 1
14 
15class SignupRequest(BaseModel):
16    username: str
17    password: str
18 
19class LoginRequest(BaseModel):
20    username: str
21    password: str
22 
23class OrderCreate(BaseModel):
24    clothing_items: list[str] = []
25    pickup_time: Optional[str] = None
26    status: str = "pending"
27    delivery_speed: str = "standard"
28 
29class OrderUpdate(BaseModel):
30    clothing_items: Optional[list[str]] = None
31    pickup_time: Optional[str] = None
32    status: Optional[str] = None
33    delivery_speed: Optional[str] = None
34 
35def get_current_user(authorization: str = Header(...)):
36    if not authorization.startswith("Bearer "):
37        raise HTTPException(status_code=401, detail="Invalid auth header")
38    token = authorization.split(" ")[1]
39    if token not in tokens:
40        raise HTTPException(status_code=401, detail="Invalid token")
41    return tokens[token]
42 
43@app.post("/signup")
44def signup(req: SignupRequest):
45    global user_id_counter
46    if any(u["username"] == req.username for u in users.values()):
47        raise HTTPException(status_code=400, detail="Username taken")
48    user_id = user_id_counter
49    user_id_counter += 1
50    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
51    return {"id": user_id, "username": req.username}
52 
53@app.post("/login")
54def login(req: LoginRequest):
55    for uid, u in users.items():
56        if u["username"] == req.username and u["password"] == req.password:
57            token = secrets.token_hex(32)
58            tokens[token] = uid
59            return {"token": token}
60    raise HTTPException(status_code=401, detail="Invalid credentials")
61 
62@app.get("/orders/{order_id}")
63def get_order(order_id: int, authorization: str = Header(...)):
64    user_id = get_current_user(authorization)
65    if order_id not in orders:
66        raise HTTPException(status_code=404, detail="Order not found")
67    return orders[order_id]
68 
69@app.post("/orders")
70def create_order(req: OrderCreate, authorization: str = Header(...)):
71    global order_id_counter
72    user_id = get_current_user(authorization)
73    order_id = order_id_counter
74    order_id_counter += 1
75    orders[order_id] = {
76        "id": order_id,
77        "user_id": user_id,
78        "clothing_items": req.clothing_items,
79        "pickup_time": req.pickup_time,
80        "status": req.status,
81        "delivery_speed": req.delivery_speed
82    }
83    return orders[order_id]
84 
85@app.patch("/orders/{order_id}")
86def update_order(order_id: int, req: OrderUpdate, authorization: str = Header(...)):
87    user_id = get_current_user(authorization)
88    if order_id not in orders:
89        raise HTTPException(status_code=404, detail="Order not found")
90    order = orders[order_id]
91    if req.clothing_items is not None:
92        order["clothing_items"] = req.clothing_items
93    if req.pickup_time is not None:
94        order["pickup_time"] = req.pickup_time
95    if req.status is not None:
96        order["status"] = req.status
97    if req.delivery_speed is not None:
98        order["delivery_speed"] = req.delivery_speed
99    return order

1fastapi
2uvicorn