Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import random
4import string
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10boxes = {}
11items = {}
12box_items = {}
13item_id_counter = 1
14box_id_counter = 1
15user_id_counter = 1
16 
17def generate_token():
18    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
19 
20def get_user_from_token(authorization: str = Header(None)):
21    if not authorization:
22        raise HTTPException(status_code=401, detail="Missing auth header")
23    token = authorization.replace("Bearer ", "")
24    if token not in tokens:
25        raise HTTPException(status_code=401, detail="Invalid token")
26    return tokens[token]
27 
28@app.post("/signup")
29def signup(username: str, password: str):
30    global user_id_counter
31    if any(u["username"] == username for u in users.values()):
32        raise HTTPException(status_code=400, detail="Username exists")
33    user_id = user_id_counter
34    user_id_counter += 1
35    users[user_id] = {"id": user_id, "username": username, "password": password}
36    return {"user_id": user_id}
37 
38@app.post("/login")
39def login(username: str, password: str):
40    for uid, u in users.items():
41        if u["username"] == username and u["password"] == password:
42            token = generate_token()
43            tokens[token] = uid
44            return {"token": token}
45    raise HTTPException(status_code=401, detail="Invalid credentials")
46 
47@app.get("/users/{user_id}")
48def get_user(user_id: int, authorization: str = Header(None)):
49    get_user_from_token(authorization)
50    if user_id not in users:
51        raise HTTPException(status_code=404, detail="User not found")
52    return users[user_id]
53 
54@app.post("/users")
55def create_user(username: str, password: str):
56    return signup(username, password)
57 
58@app.get("/boxes/{box_id}")
59def get_box(box_id: int, authorization: str = Header(None)):
60    get_user_from_token(authorization)
61    if box_id not in boxes:
62        raise HTTPException(status_code=404, detail="Box not found")
63    return boxes[box_id]
64 
65@app.post("/boxes")
66def create_box(name: str, authorization: str = Header(None)):
67    global box_id_counter
68    user_id = get_user_from_token(authorization)
69    box_id = box_id_counter
70    box_id_counter += 1
71    boxes[box_id] = {"id": box_id, "name": name, "user_id": user_id, "items": []}
72    return boxes[box_id]
73 
74@app.get("/boxes/{box_id}/contents")
75def get_box_contents(box_id: int, authorization: str = Header(None)):
76    get_user_from_token(authorization)
77    if box_id not in boxes:
78        raise HTTPException(status_code=404, detail="Box not found")
79    contents = []
80    for bi in box_items.get(box_id, []):
81        item = items.get(bi["item_id"])
82        if item:
83            contents.append({
84                "sku": item["sku"],
85                "quantity": bi["quantity"],
86                "expiration_date": item["expiration_date"]
87            })
88    return contents
89 
90@app.post("/boxes/customize")
91def customize_box(box_id: int, item_sku: str, new_item_sku: str, authorization: str = Header(None)):
92    user_id = get_user_from_token(authorization)
93    if box_id not in boxes:
94        raise HTTPException(status_code=404, detail="Box not found")
95    if boxes[box_id]["user_id"] != user_id:
96        raise HTTPException(status_code=403, detail="Not your box")
97    old_item_id = None
98    for iid, i in items.items():
99        if i["sku"] == item_sku:
100            old_item_id = iid
101            break
102    new_item_id = None
103    for iid, i in items.items():
104        if i["sku"] == new_item_sku:
105            new_item_id = iid
106            break
107    if not old_item_id or not new_item_id:
108        raise HTTPException(status_code=404, detail="Item not found")
109    box_items_list = box_items.get(box_id, [])
110    for bi in box_items_list:
111        if bi["item_id"] == old_item_id:
112            bi["item_id"] = new_item_id
113            return {"status": "swapped"}
114    raise HTTPException(status_code=404, detail="Item not in box")

1fastapi
2uvicorn