Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import random
5import string
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11recipes = {}
12reviews = {}
13next_user_id = 1
14next_recipe_id = 1
15next_review_id = 1
16 
17def generate_token():
18    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
19 
20def get_current_user(authorization: Optional[str] = Header(None)):
21    if not authorization:
22        raise HTTPException(status_code=401, detail="Missing auth token")
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 
28class SignupRequest(BaseModel):
29    username: str
30    password: str
31 
32class LoginRequest(BaseModel):
33    username: str
34    password: str
35 
36class RecipeCreate(BaseModel):
37    name: str
38    ingredients: List[str]
39    cook_time: int
40 
41class ReviewCreate(BaseModel):
42    recipe_id: int
43    rating: int
44 
45@app.post("/signup")
46def signup(req: SignupRequest):
47    global next_user_id
48    user_id = next_user_id
49    next_user_id += 1
50    users[user_id] = {"username": req.username, "password": req.password}
51    token = generate_token()
52    tokens[token] = user_id
53    return {"user_id": user_id, "token": token}
54 
55@app.post("/login")
56def login(req: LoginRequest):
57    for uid, user in users.items():
58        if user["username"] == req.username and user["password"] == req.password:
59            token = generate_token()
60            tokens[token] = uid
61            return {"token": token}
62    raise HTTPException(status_code=401, detail="Invalid credentials")
63 
64@app.get("/recipes/{recipe_id}")
65def get_recipe(recipe_id: int, authorization: Optional[str] = Header(None)):
66    get_current_user(authorization)
67    if recipe_id not in recipes:
68        raise HTTPException(status_code=404, detail="Recipe not found")
69    return recipes[recipe_id]
70 
71@app.post("/recipes")
72def create_recipe(recipe: RecipeCreate, authorization: Optional[str] = Header(None)):
73    get_current_user(authorization)
74    global next_recipe_id
75    recipe_id = next_recipe_id
76    next_recipe_id += 1
77    recipes[recipe_id] = {
78        "id": recipe_id,
79        "name": recipe.name,
80        "ingredients": recipe.ingredients,
81        "cook_time": recipe.cook_time
82    }
83    return recipes[recipe_id]
84 
85@app.get("/recipes")
86def search_recipes(ingredients: str, authorization: Optional[str] = Header(None)):
87    get_current_user(authorization)
88    user_ingredients = set(i.strip().lower() for i in ingredients.split(","))
89    results = []
90    for rid, recipe in recipes.items():
91        recipe_ingredients = set(i.lower() for i in recipe["ingredients"])
92        missing = recipe_ingredients - user_ingredients
93        results.append({
94            "id": rid,
95            "name": recipe["name"],
96            "missing_items": list(missing),
97            "missing_count": len(missing),
98            "total_cook_time": recipe["cook_time"]
99        })
100    results.sort(key=lambda x: x["missing_count"])
101    return results
102 
103@app.post("/rate")
104def rate_recipe(review: ReviewCreate, authorization: Optional[str] = Header(None)):
105    user_id = get_current_user(authorization)
106    if review.recipe_id not in recipes:
107        raise HTTPException(status_code=404, detail="Recipe not found")
108    if review.rating < 1 or review.rating > 5:
109        raise HTTPException(status_code=400, detail="Rating must be 1-5")
110    global next_review_id
111    review_id = next_review_id
112    next_review_id += 1
113    reviews[review_id] = {
114        "id": review_id,
115        "user_id": user_id,
116        "recipe_id": review.recipe_id,
117        "rating": review.rating
118    }
119    return reviews[review_id]

1fastapi
2uvicorn