Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10warehouses = {}
11 
12# Auth helper
13def get_user_from_token(authorization: str = Header(None)):
14    if not authorization:
15        raise HTTPException(401, "No auth header")
16    token = authorization.replace("Bearer ", "")
17    user_id = tokens.get(token)
18    if not user_id:
19        raise HTTPException(401, "Invalid token")
20    return user_id
21 
22class SignupRequest(BaseModel):
23    username: str
24    password: str
25 
26class LoginRequest(BaseModel):
27    username: str
28    password: str
29 
30class WarehouseCreate(BaseModel):
31    name: str
32 
33class InventoryUpdate(BaseModel):
34    quantity: Optional[int] = None
35    location: Optional[str] = None
36    role: Optional[str] = None
37    admin_access: Optional[bool] = None
38 
39class InventoryItemCreate(BaseModel):
40    item_id: int
41    quantity: int = 0
42    location: str = ""
43    role: str = ""
44    admin_access: bool = False
45 
46@app.post("/signup")
47def signup(req: SignupRequest):
48    user_id = len(users) + 1
49    users[user_id] = {"username": req.username, "password": req.password}
50    token = secrets.token_hex(16)
51    tokens[token] = user_id
52    return {"user_id": user_id, "token": token}
53 
54@app.post("/login")
55def login(req: LoginRequest):
56    for uid, u in users.items():
57        if u["username"] == req.username and u["password"] == req.password:
58            token = secrets.token_hex(16)
59            tokens[token] = uid
60            return {"token": token}
61    raise HTTPException(401, "Invalid credentials")
62 
63@app.post("/warehouses")
64def create_warehouse(req: WarehouseCreate, authorization: str = Header(None)):
65    get_user_from_token(authorization)
66    wid = len(warehouses) + 1
67    warehouses[wid] = {"id": wid, "name": req.name, "inventory": {}}
68    return warehouses[wid]
69 
70@app.get("/warehouses/{wid}")
71def get_warehouse(wid: int, authorization: str = Header(None)):
72    get_user_from_token(authorization)
73    w = warehouses.get(wid)
74    if not w:
75        raise HTTPException(404, "Warehouse not found")
76    return w
77 
78@app.put("/warehouses/{wid}/inventory/{item_id}")
79def update_inventory(wid: int, item_id: int, update: InventoryUpdate, authorization: str = Header(None)):
80    get_user_from_token(authorization)
81    w = warehouses.get(wid)
82    if not w:
83        raise HTTPException(404, "Warehouse not found")
84    inv = w["inventory"]
85    if item_id not in inv:
86        inv[item_id] = {"item_id": item_id, "quantity": 0, "location": "", "role": "", "admin_access": False}
87    item = inv[item_id]
88    if update.quantity is not None:
89        item["quantity"] = update.quantity
90    if update.location is not None:
91        item["location"] = update.location
92    if update.role is not None:
93        item["role"] = update.role
94    if update.admin_access is not None:
95        item["admin_access"] = update.admin_access
96    return item
97 
98@app.post("/warehouses/{wid}/inventory")
99def add_inventory_item(wid: int, item: InventoryItemCreate, authorization: str = Header(None)):
100    get_user_from_token(authorization)
101    w = warehouses.get(wid)
102    if not w:
103        raise HTTPException(404, "Warehouse not found")
104    w["inventory"][item.item_id] = {
105        "item_id": item.item_id,
106        "quantity": item.quantity,
107        "location": item.location,
108        "role": item.role,
109        "admin_access": item.admin_access
110    }
111    return w["inventory"][item.item_id]
112 
113@app.get("/warehouses/{wid}/inventory/{item_id}")
114def get_inventory_item(wid: int, item_id: int, authorization: str = Header(None)):
115    get_user_from_token(authorization)
116    w = warehouses.get(wid)
117    if not w:
118        raise HTTPException(404, "Warehouse not found")
119    item = w["inventory"].get(item_id)
120    if not item:
121        raise HTTPException(404, "Item not found in warehouse")
122    return item

1fastapi
2uvicorn