Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import secrets
5import datetime
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11meal_plans = {}
12subscriptions = {}
13orders = {}
14next_user_id = 1
15next_meal_plan_id = 1
16next_subscription_id = 1
17next_order_id = 1
18 
19def get_current_user(authorization: str = Header(...)):
20    token = authorization.replace("Bearer ", "")
21    if token not in tokens:
22        raise HTTPException(status_code=401, detail="Invalid token")
23    return tokens[token]
24 
25class SignupRequest(BaseModel):
26    username: str
27    password: str
28 
29class LoginRequest(BaseModel):
30    username: str
31    password: str
32 
33class MealPlanCreate(BaseModel):
34    name: str
35    description: str
36    price: float
37 
38class SubscriptionCreate(BaseModel):
39    user_id: int
40    meal_plan_id: int
41    delivery_day: str
42 
43class OrderCreate(BaseModel):
44    subscription_id: int
45    delivery_date: str
46    items: List[str]
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    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
54    return {"user_id": user_id, "username": req.username}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    for uid, u in users.items():
59        if u["username"] == req.username and u["password"] == req.password:
60            token = secrets.token_hex(16)
61            tokens[token] = uid
62            return {"token": token, "user_id": uid}
63    raise HTTPException(status_code=401, detail="Invalid credentials")
64 
65@app.post("/meal_plans")
66def create_meal_plan(plan: MealPlanCreate, authorization: str = Header(...)):
67    get_current_user(authorization)
68    global next_meal_plan_id
69    pid = next_meal_plan_id
70    next_meal_plan_id += 1
71    meal_plans[pid] = {"id": pid, "name": plan.name, "description": plan.description, "price": plan.price}
72    return meal_plans[pid]
73 
74@app.get("/meal_plans/{plan_id}")
75def get_meal_plan(plan_id: int, authorization: str = Header(...)):
76    get_current_user(authorization)
77    if plan_id not in meal_plans:
78        raise HTTPException(status_code=404, detail="Meal plan not found")
79    return meal_plans[plan_id]
80 
81@app.post("/subscriptions")
82def create_subscription(sub: SubscriptionCreate, authorization: str = Header(...)):
83    get_current_user(authorization)
84    global next_subscription_id
85    sid = next_subscription_id
86    next_subscription_id += 1
87    subscriptions[sid] = {
88        "id": sid,
89        "user_id": sub.user_id,
90        "meal_plan_id": sub.meal_plan_id,
91        "delivery_day": sub.delivery_day,
92        "created_at": datetime.datetime.now().isoformat()
93    }
94    return subscriptions[sid]
95 
96@app.get("/subscriptions/{sub_id}")
97def get_subscription(sub_id: int, authorization: str = Header(...)):
98    get_current_user(authorization)
99    if sub_id not in subscriptions:
100        raise HTTPException(status_code=404, detail="Subscription not found")
101    sub = subscriptions[sub_id]
102    history = [o for oid, o in orders.items() if o["subscription_id"] == sub_id and o["delivery_date"] < datetime.datetime.now().isoformat()]
103    upcoming = [o for oid, o in orders.items() if o["subscription_id"] == sub_id and o["delivery_date"] >= datetime.datetime.now().isoformat()]
104    return {"subscription": sub, "history": history, "upcoming": upcoming}
105 
106@app.post("/orders")
107def create_order(order: OrderCreate, authorization: str = Header(...)):
108    get_current_user(authorization)
109    global next_order_id
110    oid = next_order_id
111    next_order_id += 1
112    orders[oid] = {
113        "id": oid,
114        "subscription_id": order.subscription_id,
115        "delivery_date": order.delivery_date,
116        "items": order.items,
117        "created_at": datetime.datetime.now().isoformat()
118    }
119    return orders[oid]
120 
121@app.get("/orders/{order_id}")
122def get_order(order_id: int, authorization: str = Header(...)):
123    get_current_user(authorization)
124    if order_id not in orders:
125        raise HTTPException(status_code=404, detail="Order not found")
126    return orders[order_id]

1fastapi
2uvicorn
3pydantic