Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9cellars = {}
10wines = {}
11next_user_id = 1
12next_cellar_id = 1
13next_wine_id = 1
14 
15def get_user_from_token(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 
24@app.post("/signup")
25def signup(username: str, password: str):
26    global next_user_id
27    for u in users.values():
28        if u["username"] == username:
29            raise HTTPException(status_code=400, detail="Username taken")
30    user_id = next_user_id
31    next_user_id += 1
32    users[user_id] = {"id": user_id, "username": username, "password": password, "role": "user"}
33    return {"user_id": user_id}
34 
35@app.post("/login")
36def login(username: str, password: str):
37    for u in users.values():
38        if u["username"] == username and u["password"] == password:
39            token = secrets.token_hex(16)
40            tokens[token] = u["id"]
41            return {"token": token}
42    raise HTTPException(status_code=401, detail="Invalid credentials")
43 
44@app.post("/cellar")
45def create_cellar(name: str, authorization: Optional[str] = Header(None)):
46    user_id = get_user_from_token(authorization)
47    global next_cellar_id
48    cellar_id = next_cellar_id
49    next_cellar_id += 1
50    cellars[cellar_id] = {
51        "id": cellar_id,
52        "name": name,
53        "sommelier_id": user_id,
54        "assistants": [],
55        "wines": []
56    }
57    users[user_id]["role"] = "sommelier"
58    return {"cellar_id": cellar_id}
59 
60@app.get("/cellar/{cellar_id}")
61def get_cellar(cellar_id: int, authorization: Optional[str] = Header(None)):
62    user_id = get_user_from_token(authorization)
63    cellar = cellars.get(cellar_id)
64    if not cellar:
65        raise HTTPException(status_code=404, detail="Cellar not found")
66    return cellar
67 
68@app.post("/cellar/{cellar_id}/promote")
69def promote_taster(cellar_id: int, taster_id: int, authorization: Optional[str] = Header(None)):
70    user_id = get_user_from_token(authorization)
71    cellar = cellars.get(cellar_id)
72    if not cellar:
73        raise HTTPException(status_code=404, detail="Cellar not found")
74    if cellar["sommelier_id"] != user_id:
75        raise HTTPException(status_code=403, detail="Only sommelier can promote")
76    taster = users.get(taster_id)
77    if not taster:
78        raise HTTPException(status_code=404, detail="Taster not found")
79    if taster_id in cellar["assistants"]:
80        raise HTTPException(status_code=400, detail="Already assistant")
81    cellar["assistants"].append(taster_id)
82    return {"status": "promoted"}
83 
84@app.post("/wine")
85def create_wine(name: str, vintage: int, tasting_notes: str, quantity: int, cellar_id: int, authorization: Optional[str] = Header(None)):
86    user_id = get_user_from_token(authorization)
87    cellar = cellars.get(cellar_id)
88    if not cellar:
89        raise HTTPException(status_code=404, detail="Cellar not found")
90    if cellar["sommelier_id"] != user_id and user_id not in cellar["assistants"]:
91        raise HTTPException(status_code=403, detail="Not authorized")
92    global next_wine_id
93    wine_id = next_wine_id
94    next_wine_id += 1
95    wine = {
96        "id": wine_id,
97        "name": name,
98        "vintage": vintage,
99        "tasting_notes": tasting_notes,
100        "quantity": quantity,
101        "cellar_id": cellar_id
102    }
103    wines[wine_id] = wine
104    cellar["wines"].append(wine_id)
105    return {"wine_id": wine_id}
106 
107@app.get("/wine/{wine_id}")
108def get_wine(wine_id: int, authorization: Optional[str] = Header(None)):
109    user_id = get_user_from_token(authorization)
110    wine = wines.get(wine_id)
111    if not wine:
112        raise HTTPException(status_code=404, detail="Wine not found")
113    return wine

1fastapi
2uvicorn