Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9sellers = {}
10listings = {}
11orders = {}
12next_user_id = 1
13next_seller_id = 1
14next_listing_id = 1
15next_order_id = 1
16 
17def get_current_user(authorization: Optional[str] = Header(None)):
18    if not authorization:
19        raise HTTPException(status_code=401, detail="Missing auth")
20    token = authorization.replace("Bearer ", "")
21    for uid, t in tokens.items():
22        if t == token:
23            return uid
24    raise HTTPException(status_code=401, detail="Invalid token")
25 
26@app.post("/signup")
27def signup(username: str, password: str):
28    global next_user_id
29    for u in users.values():
30        if u["username"] == username:
31            raise HTTPException(status_code=400, detail="Username taken")
32    uid = next_user_id
33    next_user_id += 1
34    users[uid] = {"id": uid, "username": username, "password": password, "is_admin": False}
35    return {"id": uid, "username": username}
36 
37@app.post("/login")
38def login(username: str, password: str):
39    for uid, u in users.items():
40        if u["username"] == username and u["password"] == password:
41            token = secrets.token_hex(16)
42            tokens[uid] = token
43            return {"token": token}
44    raise HTTPException(status_code=401, detail="Invalid credentials")
45 
46@app.post("/sellers")
47def create_seller(name: str, authorization: Optional[str] = Header(None)):
48    user_id = get_current_user(authorization)
49    global next_seller_id
50    sid = next_seller_id
51    next_seller_id += 1
52    sellers[sid] = {"id": sid, "name": name, "user_id": user_id, "is_moderator": False}
53    return {"id": sid, "name": name}
54 
55@app.get("/sellers/{seller_id}")
56def get_seller(seller_id: int):
57    if seller_id not in sellers:
58        raise HTTPException(status_code=404, detail="Seller not found")
59    return sellers[seller_id]
60 
61@app.post("/sellers/{seller_id}/make-moderator")
62def make_moderator(seller_id: int, authorization: Optional[str] = Header(None)):
63    user_id = get_current_user(authorization)
64    if not users[user_id].get("is_admin"):
65        raise HTTPException(status_code=403, detail="Only admins can promote")
66    if seller_id not in sellers:
67        raise HTTPException(status_code=404, detail="Seller not found")
68    sellers[seller_id]["is_moderator"] = True
69    return {"id": seller_id, "is_moderator": True}
70 
71@app.post("/listings")
72def create_listing(seller_id: int, name: str, price: float, stock: int, authorization: Optional[str] = Header(None)):
73    user_id = get_current_user(authorization)
74    if seller_id not in sellers:
75        raise HTTPException(status_code=404, detail="Seller not found")
76    if sellers[seller_id]["user_id"] != user_id:
77        raise HTTPException(status_code=403, detail="Not your seller account")
78    global next_listing_id
79    lid = next_listing_id
80    next_listing_id += 1
81    listings[lid] = {"id": lid, "seller_id": seller_id, "name": name, "price": price, "stock": stock}
82    return {"id": lid, "name": name}
83 
84@app.get("/listings/{listing_id}")
85def get_listing(listing_id: int):
86    if listing_id not in listings:
87        raise HTTPException(status_code=404, detail="Listing not found")
88    return listings[listing_id]
89 
90@app.post("/orders")
91def create_order(listing_id: int, quantity: int, authorization: Optional[str] = Header(None)):
92    user_id = get_current_user(authorization)
93    if listing_id not in listings:
94        raise HTTPException(status_code=404, detail="Listing not found")
95    if listings[listing_id]["stock"] < quantity:
96        raise HTTPException(status_code=400, detail="Not enough stock")
97    listings[listing_id]["stock"] -= quantity
98    global next_order_id
99    oid = next_order_id
100    next_order_id += 1
101    orders[oid] = {"id": oid, "listing_id": listing_id, "quantity": quantity, "user_id": user_id}
102    return {"id": oid, "listing_id": listing_id, "quantity": quantity}
103 
104@app.get("/orders/{order_id}")
105def get_order(order_id: int):
106    if order_id not in orders:
107        raise HTTPException(status_code=404, detail="Order not found")
108    return orders[order_id]

1fastapi
2uvicorn