Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict
4import secrets
5import datetime
6 
7app = FastAPI()
8 
9users: Dict[int, dict] = {}
10ingredients: Dict[int, dict] = {}
11next_user_id = 1
12next_ingredient_id = 1
13tokens: Dict[str, int] = {}
14 
15def get_current_user(authorization: Optional[str] = Header(None)):
16    if not authorization:
17        raise HTTPException(status_code=401, detail="Missing auth header")
18    token = authorization.replace("Bearer ", "")
19    user_id = tokens.get(token)
20    if not user_id:
21        raise HTTPException(status_code=401, detail="Invalid token")
22    return user_id
23 
24class SignupRequest(BaseModel):
25    username: str
26    password: str
27 
28class LoginRequest(BaseModel):
29    username: str
30    password: str
31 
32class IngredientCreate(BaseModel):
33    name: str
34    quantity: int
35    expiration_date: str
36 
37class IngredientUpdate(BaseModel):
38    name: Optional[str] = None
39    quantity: Optional[int] = None
40    expiration_date: Optional[str] = None
41 
42@app.post("/signup")
43def signup(req: SignupRequest):
44    global next_user_id
45    user_id = next_user_id
46    next_user_id += 1
47    users[user_id] = {"username": req.username, "password": req.password}
48    return {"id": user_id, "username": req.username}
49 
50@app.post("/login")
51def login(req: LoginRequest):
52    for uid, u in users.items():
53        if u["username"] == req.username and u["password"] == req.password:
54            token = secrets.token_hex(16)
55            tokens[token] = uid
56            return {"token": token}
57    raise HTTPException(status_code=401, detail="Invalid credentials")
58 
59@app.get("/ingredients/{ingredient_id}")
60def get_ingredient(ingredient_id: int, authorization: Optional[str] = Header(None)):
61    get_current_user(authorization)
62    ing = ingredients.get(ingredient_id)
63    if not ing:
64        raise HTTPException(status_code=404, detail="Ingredient not found")
65    alert = ing["quantity"] < 10
66    return {**ing, "id": ingredient_id, "low_stock_alert": alert}
67 
68@app.post("/ingredients")
69def create_ingredient(req: IngredientCreate, authorization: Optional[str] = Header(None)):
70    get_current_user(authorization)
71    global next_ingredient_id
72    ing_id = next_ingredient_id
73    next_ingredient_id += 1
74    ingredients[ing_id] = {
75        "name": req.name,
76        "quantity": req.quantity,
77        "expiration_date": req.expiration_date
78    }
79    alert = req.quantity < 10
80    return {**ingredients[ing_id], "id": ing_id, "low_stock_alert": alert}
81 
82@app.put("/ingredients/{ingredient_id}")
83def update_ingredient(ingredient_id: int, req: IngredientUpdate, authorization: Optional[str] = Header(None)):
84    get_current_user(authorization)
85    ing = ingredients.get(ingredient_id)
86    if not ing:
87        raise HTTPException(status_code=404, detail="Ingredient not found")
88    if req.name is not None:
89        ing["name"] = req.name
90    if req.quantity is not None:
91        ing["quantity"] = req.quantity
92    if req.expiration_date is not None:
93        ing["expiration_date"] = req.expiration_date
94    alert = ing["quantity"] < 10
95    return {**ing, "id": ingredient_id, "low_stock_alert": alert}
96 
97@app.delete("/ingredients/{ingredient_id}")
98def delete_ingredient(ingredient_id: int, authorization: Optional[str] = Header(None)):
99    get_current_user(authorization)
100    if ingredient_id not in ingredients:
101        raise HTTPException(status_code=404, detail="Ingredient not found")
102    del ingredients[ingredient_id]
103    return {"ok": True}

1fastapi
2uvicorn