Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import uuid
4 
5app = FastAPI()
6 
7users = {}
8stores = {}
9products = {}
10reorder_history = {}
11tokens = {}
12next_user_id = 1
13next_store_id = 1
14next_product_id = 1
15next_reorder_id = 1
16 
17def get_current_user(authorization: Optional[str] = Header(None)):
18    if not authorization:
19        raise HTTPException(status_code=401, detail="No token provided")
20    token = authorization.replace("Bearer ", "")
21    if token not in tokens:
22        raise HTTPException(status_code=401, detail="Invalid token")
23    return tokens[token]
24 
25@app.post("/signup")
26def signup(username: str, password: str):
27    global next_user_id
28    if username in users:
29        raise HTTPException(status_code=400, detail="User already exists")
30    user_id = next_user_id
31    users[username] = {"id": user_id, "password": password, "role": "user"}
32    next_user_id += 1
33    return {"id": user_id, "username": username}
34 
35@app.post("/login")
36def login(username: str, password: str):
37    if username not in users or users[username]["password"] != password:
38        raise HTTPException(status_code=401, detail="Invalid credentials")
39    token = str(uuid.uuid4())
40    tokens[token] = users[username]
41    return {"token": token}
42 
43@app.post("/store")
44def create_store(name: str, authorization: Optional[str] = Header(None)):
45    user = get_current_user(authorization)
46    global next_store_id
47    store_id = next_store_id
48    stores[store_id] = {"id": store_id, "name": name, "admin_id": user["id"], "staff": []}
49    next_store_id += 1
50    return stores[store_id]
51 
52@app.get("/store/{store_id}")
53def get_store(store_id: int, authorization: Optional[str] = Header(None)):
54    get_current_user(authorization)
55    if store_id not in stores:
56        raise HTTPException(status_code=404, detail="Store not found")
57    return stores[store_id]
58 
59@app.post("/store/{store_id}/promote")
60def promote_to_inventory_manager(store_id: int, username: str, authorization: Optional[str] = Header(None)):
61    user = get_current_user(authorization)
62    if store_id not in stores:
63        raise HTTPException(status_code=404, detail="Store not found")
64    store = stores[store_id]
65    if store["admin_id"] != user["id"]:
66        raise HTTPException(status_code=403, detail="Only store admin can promote")
67    if username not in users:
68        raise HTTPException(status_code=404, detail="User not found")
69    if username not in store["staff"]:
70        store["staff"].append(username)
71    return {"message": f"{username} promoted to inventory manager"}
72 
73@app.post("/product")
74def create_product(sku: str, store_id: int, stock: int = 0, authorization: Optional[str] = Header(None)):
75    user = get_current_user(authorization)
76    if store_id not in stores:
77        raise HTTPException(status_code=404, detail="Store not found")
78    store = stores[store_id]
79    if user["id"] != store["admin_id"] and user["username"] not in store["staff"]:
80        raise HTTPException(status_code=403, detail="Not authorized")
81    global next_product_id
82    product_id = next_product_id
83    products[product_id] = {"id": product_id, "sku": sku, "store_id": store_id, "stock": stock}
84    next_product_id += 1
85    return products[product_id]
86 
87@app.get("/product/{product_id}")
88def get_product(product_id: int, authorization: Optional[str] = Header(None)):
89    get_current_user(authorization)
90    if product_id not in products:
91        raise HTTPException(status_code=404, detail="Product not found")
92    return products[product_id]
93 
94@app.post("/reorder")
95def create_reorder(product_id: int, quantity: int, authorization: Optional[str] = Header(None)):
96    user = get_current_user(authorization)
97    if product_id not in products:
98        raise HTTPException(status_code=404, detail="Product not found")
99    product = products[product_id]
100    store = stores[product["store_id"]]
101    if user["id"] != store["admin_id"] and user["username"] not in store["staff"]:
102        raise HTTPException(status_code=403, detail="Not authorized")
103    global next_reorder_id
104    reorder_id = next_reorder_id
105    from datetime import datetime
106    reorder_history[reorder_id] = {
107        "id": reorder_id,
108        "product_id": product_id,
109        "quantity": quantity,
110        "timestamp": datetime.now().isoformat()
111    }
112    product["stock"] += quantity
113    next_reorder_id += 1
114    return reorder_history[reorder_id]
115 
116@app.get("/reorder/{reorder_id}")
117def get_reorder(reorder_id: int, authorization: Optional[str] = Header(None)):
118    get_current_user(authorization)
119    if reorder_id not in reorder_history:
120        raise HTTPException(status_code=404, detail="Reorder not found")
121    return reorder_history[reorder_id]

1fastapi
2uvicorn