Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import uuid
5import hashlib
6 
7app = FastAPI()
8 
9cocktails = {}
10cocktail_id_counter = 1
11users = {}
12tokens = {}
13 
14class SignupRequest(BaseModel):
15    username: str
16    password: str
17 
18class LoginRequest(BaseModel):
19    username: str
20    password: str
21 
22class Cocktail(BaseModel):
23    name: str
24    ingredients: list[str]
25    instructions: str
26 
27class Rating(BaseModel):
28    rating: int
29 
30def hash_password(password: str) -> str:
31    return hashlib.sha256(password.encode()).hexdigest()
32 
33def get_user_from_token(authorization: str = Header(None)):
34    if not authorization:
35        raise HTTPException(status_code=401, detail="Missing auth header")
36    token = authorization.replace("Bearer ", "")
37    if token not in tokens:
38        raise HTTPException(status_code=401, detail="Invalid token")
39    return tokens[token]
40 
41@app.post("/signup")
42def signup(req: SignupRequest):
43    if req.username in users:
44        raise HTTPException(status_code=400, detail="User already exists")
45    users[req.username] = hash_password(req.password)
46    return {"message": "User created"}
47 
48@app.post("/login")
49def login(req: LoginRequest):
50    if req.username not in users or users[req.username] != hash_password(req.password):
51        raise HTTPException(status_code=401, detail="Invalid credentials")
52    token = str(uuid.uuid4())
53    tokens[token] = req.username
54    return {"token": token}
55 
56@app.get("/cocktails")
57def get_cocktails(authorization: str = Header(None)):
58    get_user_from_token(authorization)
59    return cocktails
60 
61@app.post("/cocktails")
62def create_cocktail(cocktail: Cocktail, authorization: str = Header(None)):
63    get_user_from_token(authorization)
64    global cocktail_id_counter
65    cid = cocktail_id_counter
66    cocktail_id_counter += 1
67    cocktails[cid] = {
68        "id": cid,
69        "name": cocktail.name,
70        "ingredients": cocktail.ingredients,
71        "instructions": cocktail.instructions,
72        "ratings": []
73    }
74    return cocktails[cid]
75 
76@app.get("/cocktails/{cocktail_id}")
77def get_cocktail(cocktail_id: int, authorization: str = Header(None)):
78    get_user_from_token(authorization)
79    if cocktail_id not in cocktails:
80        raise HTTPException(status_code=404, detail="Cocktail not found")
81    return cocktails[cocktail_id]
82 
83@app.post("/cocktails/{cocktail_id}/rate")
84def rate_cocktail(cocktail_id: int, rating: Rating, authorization: str = Header(None)):
85    get_user_from_token(authorization)
86    if cocktail_id not in cocktails:
87        raise HTTPException(status_code=404, detail="Cocktail not found")
88    if rating.rating < 1 or rating.rating > 5:
89        raise HTTPException(status_code=400, detail="Rating must be between 1 and 5")
90    cocktails[cocktail_id]["ratings"].append(rating.rating)
91    return {"message": "Rating added"}
92 
93@app.get("/cocktails/search")
94def search_cocktails(by: str = "ingredient", q: str = "", authorization: str = Header(None)):
95    get_user_from_token(authorization)
96    if by != "ingredient":
97        raise HTTPException(status_code=400, detail="Only ingredient search is supported")
98    results = []
99    for cid, c in cocktails.items():
100        if any(q.lower() in ing.lower() for ing in c["ingredients"]):
101            results.append(c)
102    return results

1fastapi
2uvicorn