Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import uuid
5import hashlib
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11recipes = {}
12next_user_id = 1
13next_recipe_id = 1
14 
15class SignupRequest(BaseModel):
16    username: str
17    password: str
18    name: Optional[str] = ""
19    dietary_preferences: Optional[List[str]] = []
20    favorite_cuisines: Optional[List[str]] = []
21    account_tier: Optional[str] = "free"
22    is_admin: Optional[bool] = False
23 
24class LoginRequest(BaseModel):
25    username: str
26    password: str
27 
28class UserUpdateRequest(BaseModel):
29    name: Optional[str] = None
30    dietary_preferences: Optional[List[str]] = None
31    favorite_cuisines: Optional[List[str]] = None
32    account_tier: Optional[str] = None
33    is_admin: Optional[bool] = None
34 
35class RecipeCreateRequest(BaseModel):
36    title: str
37    ingredients: List[str]
38    instructions: str
39 
40def get_current_user(authorization: str = Header(...)):
41    if not authorization.startswith("Bearer "):
42        raise HTTPException(status_code=401, detail="Invalid auth header")
43    token = authorization.split(" ")[1]
44    if token not in tokens:
45        raise HTTPException(status_code=401, detail="Invalid token")
46    return tokens[token]
47 
48@app.post("/signup")
49def signup(req: SignupRequest):
50    global next_user_id
51    user_id = next_user_id
52    next_user_id += 1
53    password_hash = hashlib.sha256(req.password.encode()).hexdigest()
54    users[user_id] = {
55        "id": user_id,
56        "username": req.username,
57        "password_hash": password_hash,
58        "name": req.name,
59        "dietary_preferences": req.dietary_preferences,
60        "favorite_cuisines": req.favorite_cuisines,
61        "account_tier": req.account_tier,
62        "is_admin": req.is_admin
63    }
64    return {"id": user_id, "username": req.username}
65 
66@app.post("/login")
67def login(req: LoginRequest):
68    for uid, user in users.items():
69        if user["username"] == req.username:
70            password_hash = hashlib.sha256(req.password.encode()).hexdigest()
71            if user["password_hash"] == password_hash:
72                token = str(uuid.uuid4())
73                tokens[token] = uid
74                return {"token": token}
75    raise HTTPException(status_code=401, detail="Invalid credentials")
76 
77@app.get("/users/{user_id}")
78def get_user(user_id: int):
79    if user_id not in users:
80        raise HTTPException(status_code=404, detail="User not found")
81    return users[user_id]
82 
83@app.post("/users")
84def create_user(req: SignupRequest):
85    return signup(req)
86 
87@app.patch("/users/{user_id}")
88def update_user(user_id: int, req: UserUpdateRequest, authorization: str = Header(...)):
89    current_user_id = get_current_user(authorization)
90    if current_user_id != user_id and not users[current_user_id]["is_admin"]:
91        raise HTTPException(status_code=403, detail="Not authorized")
92    if user_id not in users:
93        raise HTTPException(status_code=404, detail="User not found")
94    user = users[user_id]
95    if req.name is not None:
96        user["name"] = req.name
97    if req.dietary_preferences is not None:
98        user["dietary_preferences"] = req.dietary_preferences
99    if req.favorite_cuisines is not None:
100        user["favorite_cuisines"] = req.favorite_cuisines
101    if req.account_tier is not None:
102        user["account_tier"] = req.account_tier
103    if req.is_admin is not None:
104        if users[current_user_id]["is_admin"]:
105            user["is_admin"] = req.is_admin
106        else:
107            raise HTTPException(status_code=403, detail="Only admins can change admin status")
108    return user
109 
110@app.get("/recipes/{recipe_id}")
111def get_recipe(recipe_id: int):
112    if recipe_id not in recipes:
113        raise HTTPException(status_code=404, detail="Recipe not found")
114    return recipes[recipe_id]
115 
116@app.post("/recipes")
117def create_recipe(req: RecipeCreateRequest, authorization: str = Header(...)):
118    current_user_id = get_current_user(authorization)
119    global next_recipe_id
120    recipe_id = next_recipe_id
121    next_recipe_id += 1
122    recipes[recipe_id] = {
123        "id": recipe_id,
124        "title": req.title,
125        "ingredients": req.ingredients,
126        "instructions": req.instructions,
127        "author_id": current_user_id
128    }
129    return recipes[recipe_id]

1fastapi
2uvicorn