Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import hashlib
5import secrets
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11wines = {}
12wine_id_counter = 1
13 
14class SignupRequest(BaseModel):
15    username: str
16    password: str
17 
18class LoginRequest(BaseModel):
19    username: str
20    password: str
21 
22class WineCreate(BaseModel):
23    varietal: str
24    vintage: int
25    region: str
26    rating: int
27    drink_window_start: int
28    bottle_count: int
29    storage_location: str
30 
31def get_current_user(authorization: str = Header(...)):
32    if not authorization.startswith("Bearer "):
33        raise HTTPException(status_code=401, detail="Invalid auth header")
34    token = authorization[7:]
35    user_id = tokens.get(token)
36    if user_id is None:
37        raise HTTPException(status_code=401, detail="Invalid token")
38    return user_id
39 
40@app.post("/signup")
41def signup(req: SignupRequest):
42    if req.username in users:
43        raise HTTPException(status_code=400, detail="User already exists")
44    user_id = len(users) + 1
45    users[req.username] = {"id": user_id, "password": hashlib.sha256(req.password.encode()).hexdigest()}
46    return {"id": user_id, "username": req.username}
47 
48@app.post("/login")
49def login(req: LoginRequest):
50    user = users.get(req.username)
51    if not user or user["password"] != hashlib.sha256(req.password.encode()).hexdigest():
52        raise HTTPException(status_code=401, detail="Invalid credentials")
53    token = secrets.token_hex(32)
54    tokens[token] = user["id"]
55    return {"token": token}
56 
57@app.get("/wines/{wine_id}")
58def get_wine(wine_id: int, authorization: str = Header(...)):
59    get_current_user(authorization)
60    wine = wines.get(wine_id)
61    if not wine:
62        raise HTTPException(status_code=404, detail="Wine not found")
63    return wine
64 
65@app.post("/wines")
66def create_wine(wine: WineCreate, authorization: str = Header(...)):
67    get_current_user(authorization)
68    global wine_id_counter
69    wine_id = wine_id_counter
70    wine_id_counter += 1
71    wines[wine_id] = {
72        "id": wine_id,
73        "varietal": wine.varietal,
74        "vintage": wine.vintage,
75        "region": wine.region,
76        "rating": wine.rating,
77        "drink_window_start": wine.drink_window_start,
78        "bottle_count": wine.bottle_count,
79        "storage_location": wine.storage_location
80    }
81    return wines[wine_id]
82 
83@app.get("/wines")
84def search_wines(
85    vintage: Optional[int] = None,
86    region: Optional[str] = None,
87    rating_min: Optional[int] = None,
88    varietal: Optional[str] = None,
89    authorization: str = Header(...)
90):
91    get_current_user(authorization)
92    results = []
93    for wine in wines.values():
94        if vintage is not None and wine["vintage"] != vintage:
95            continue
96        if region is not None and wine["region"].lower() != region.lower():
97            continue
98        if rating_min is not None and wine["rating"] < rating_min:
99            continue
100        if varietal is not None and wine["varietal"].lower() != varietal.lower():
101            continue
102        results.append({
103            "id": wine["id"],
104            "varietal": wine["varietal"],
105            "vintage": wine["vintage"],
106            "region": wine["region"],
107            "rating": wine["rating"],
108            "drink_window_start": wine["drink_window_start"],
109            "bottle_count": wine["bottle_count"],
110            "storage_location": wine["storage_location"]
111        })
112    return results

1fastapi
2uvicorn