Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import hashlib
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11ingredients = {}
12recipes = {}
13substitutions = {}
14compatibility_ratings = {}
15next_user_id = 1
16next_ingredient_id = 1
17next_recipe_id = 1
18next_substitution_id = 1
19next_rating_id = 1
20 
21def get_current_user(authorization: Optional[str] = Header(None)):
22    if not authorization:
23        raise HTTPException(status_code=401, detail="Missing auth token")
24    token = authorization.replace("Bearer ", "")
25    if token not in tokens:
26        raise HTTPException(status_code=401, detail="Invalid token")
27    return tokens[token]
28 
29class SignupRequest(BaseModel):
30    username: str
31    password: str
32 
33class LoginRequest(BaseModel):
34    username: str
35    password: str
36 
37class IngredientCreate(BaseModel):
38    name: str
39    category: Optional[str] = None
40 
41class RecipeCreate(BaseModel):
42    name: str
43    primary_ingredient_id: int
44    instructions: Optional[str] = None
45 
46class SubstitutionRequest(BaseModel):
47    ingredient_id: int
48    substitute_ingredient_id: int
49 
50class CompatibilityRating(BaseModel):
51    substitution_id: int
52    rating: float
53 
54@app.post("/signup")
55def signup(req: SignupRequest):
56    global next_user_id
57    if req.username in users:
58        raise HTTPException(status_code=400, detail="Username already exists")
59    user_id = next_user_id
60    next_user_id += 1
61    users[req.username] = {"id": user_id, "password": hashlib.sha256(req.password.encode()).hexdigest()}
62    return {"id": user_id, "username": req.username}
63 
64@app.post("/login")
65def login(req: LoginRequest):
66    if req.username not in users:
67        raise HTTPException(status_code=401, detail="Invalid credentials")
68    if users[req.username]["password"] != hashlib.sha256(req.password.encode()).hexdigest():
69        raise HTTPException(status_code=401, detail="Invalid credentials")
70    token = secrets.token_hex(32)
71    tokens[token] = users[req.username]["id"]
72    return {"token": token}
73 
74@app.post("/ingredients")
75def create_ingredient(ing: IngredientCreate, authorization: Optional[str] = Header(None)):
76    get_current_user(authorization)
77    global next_ingredient_id
78    ing_id = next_ingredient_id
79    next_ingredient_id += 1
80    ingredients[ing_id] = {"id": ing_id, "name": ing.name, "category": ing.category}
81    return ingredients[ing_id]
82 
83@app.get("/ingredients/{ingredient_id}")
84def get_ingredient(ingredient_id: int, authorization: Optional[str] = Header(None)):
85    get_current_user(authorization)
86    if ingredient_id not in ingredients:
87        raise HTTPException(status_code=404, detail="Ingredient not found")
88    return ingredients[ingredient_id]
89 
90@app.post("/recipes")
91def create_recipe(rec: RecipeCreate, authorization: Optional[str] = Header(None)):
92    get_current_user(authorization)
93    if rec.primary_ingredient_id not in ingredients:
94        raise HTTPException(status_code=400, detail="Primary ingredient not found")
95    global next_recipe_id
96    rec_id = next_recipe_id
97    next_recipe_id += 1
98    recipes[rec_id] = {"id": rec_id, "name": rec.name, "primary_ingredient_id": rec.primary_ingredient_id, "instructions": rec.instructions}
99    return recipes[rec_id]
100 
101@app.get("/recipes/{recipe_id}")
102def get_recipe(recipe_id: int, authorization: Optional[str] = Header(None)):
103    get_current_user(authorization)
104    if recipe_id not in recipes:
105        raise HTTPException(status_code=404, detail="Recipe not found")
106    return recipes[recipe_id]
107 
108@app.post("/substitutions")
109def create_substitution(sub: SubstitutionRequest, authorization: Optional[str] = Header(None)):
110    get_current_user(authorization)
111    if sub.ingredient_id not in ingredients:
112        raise HTTPException(status_code=400, detail="Ingredient not found")
113    if sub.substitute_ingredient_id not in ingredients:
114        raise HTTPException(status_code=400, detail="Substitute ingredient not found")
115    global next_substitution_id
116    sub_id = next_substitution_id
117    next_substitution_id += 1
118    substitutions[sub_id] = {"id": sub_id, "ingredient_id": sub.ingredient_id, "substitute_ingredient_id": sub.substitute_ingredient_id}
119    return substitutions[sub_id]
120 
121@app.get("/substitutions/{substitution_id}")
122def get_substitution(substitution_id: int, authorization: Optional[str] = Header(None)):
123    get_current_user(authorization)
124    if substitution_id not in substitutions:
125        raise HTTPException(status_code=404, detail="Substitution not found")
126    return substitutions[substitution_id]
127 
128@app.post("/compatibility-ratings")
129def create_compatibility_rating(rate: CompatibilityRating, authorization: Optional[str] = Header(None)):
130    get_current_user(authorization)
131    if rate.substitution_id not in substitutions:
132        raise HTTPException(status_code=400, detail="Substitution not found")
133    global next_rating_id
134    rate_id = next_rating_id
135    next_rating_id += 1
136    compatibility_ratings[rate_id] = {"id": rate_id, "substitution_id": rate.substitution_id, "rating": rate.rating}
137    return compatibility_ratings[rate_id]
138 
139@app.get("/compatibility-ratings/{rating_id}")
140def get_compatibility_rating(rating_id: int, authorization: Optional[str] = Header(None)):
141    get_current_user(authorization)
142    if rating_id not in compatibility_ratings:
143        raise HTTPException(status_code=404, detail="Rating not found")
144    return compatibility_ratings[rating_id]
145 
146@app.get("/ingredients/{ingredient_id}/substitutions")
147def get_substitutions_for_ingredient(ingredient_id: int, authorization: Optional[str] = Header(None)):
148    get_current_user(authorization)
149    result = []
150    for sub in substitutions.values():
151        if sub["ingredient_id"] == ingredient_id:
152            result.append(sub)
153    return result

1fastapi
2uvicorn