mirror of
https://github.com/sstent/foodplanner.git
synced 2025-12-06 08:01:47 +00:00
initial
This commit is contained in:
303
main.py
Normal file
303
main.py
Normal file
@@ -0,0 +1,303 @@
|
||||
# Meal Planner FastAPI Application
|
||||
# Run with: uvicorn main:app --reload
|
||||
|
||||
from fastapi import FastAPI, Depends, HTTPException, Request, Form
|
||||
from fastapi.templating import Jinja2Templates
|
||||
from fastapi.staticfiles import StaticFiles
|
||||
from fastapi.responses import HTMLResponse
|
||||
from sqlalchemy import create_engine, Column, Integer, String, Float, DateTime, ForeignKey, Text, Date
|
||||
from sqlalchemy.ext.declarative import declarative_base
|
||||
from sqlalchemy.orm import sessionmaker, Session, relationship
|
||||
from pydantic import BaseModel
|
||||
from typing import List, Optional
|
||||
from datetime import date, datetime
|
||||
import os
|
||||
|
||||
# Database setup - Use SQLite for easier setup
|
||||
DATABASE_URL = "sqlite:///./meal_planner.db"
|
||||
# For production, use PostgreSQL: DATABASE_URL = "postgresql://username:password@localhost/meal_planner"
|
||||
|
||||
engine = create_engine(DATABASE_URL, connect_args={"check_same_thread": False} if "sqlite" in DATABASE_URL else {})
|
||||
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
||||
Base = declarative_base()
|
||||
|
||||
# Initialize FastAPI app
|
||||
app = FastAPI(title="Meal Planner")
|
||||
templates = Jinja2Templates(directory="templates")
|
||||
|
||||
# Database Models
|
||||
class Food(Base):
|
||||
__tablename__ = "foods"
|
||||
|
||||
id = Column(Integer, primary_key=True, index=True)
|
||||
name = Column(String, unique=True, index=True)
|
||||
serving_size = Column(String)
|
||||
serving_unit = Column(String)
|
||||
calories = Column(Float)
|
||||
protein = Column(Float)
|
||||
carbs = Column(Float)
|
||||
fat = Column(Float)
|
||||
fiber = Column(Float, default=0)
|
||||
sugar = Column(Float, default=0)
|
||||
sodium = Column(Float, default=0)
|
||||
calcium = Column(Float, default=0)
|
||||
|
||||
class Meal(Base):
|
||||
__tablename__ = "meals"
|
||||
|
||||
id = Column(Integer, primary_key=True, index=True)
|
||||
name = Column(String, index=True)
|
||||
meal_type = Column(String) # breakfast, lunch, dinner, snack
|
||||
|
||||
# Relationship to meal foods
|
||||
meal_foods = relationship("MealFood", back_populates="meal")
|
||||
|
||||
class MealFood(Base):
|
||||
__tablename__ = "meal_foods"
|
||||
|
||||
id = Column(Integer, primary_key=True, index=True)
|
||||
meal_id = Column(Integer, ForeignKey("meals.id"))
|
||||
food_id = Column(Integer, ForeignKey("foods.id"))
|
||||
quantity = Column(Float)
|
||||
|
||||
meal = relationship("Meal", back_populates="meal_foods")
|
||||
food = relationship("Food")
|
||||
|
||||
class Plan(Base):
|
||||
__tablename__ = "plans"
|
||||
|
||||
id = Column(Integer, primary_key=True, index=True)
|
||||
person = Column(String, index=True) # Person A or Person B
|
||||
date = Column(Date, index=True)
|
||||
meal_id = Column(Integer, ForeignKey("meals.id"))
|
||||
|
||||
meal = relationship("Meal")
|
||||
|
||||
# Pydantic models
|
||||
class FoodCreate(BaseModel):
|
||||
name: str
|
||||
serving_size: str
|
||||
serving_unit: str
|
||||
calories: float
|
||||
protein: float
|
||||
carbs: float
|
||||
fat: float
|
||||
fiber: float = 0
|
||||
sugar: float = 0
|
||||
sodium: float = 0
|
||||
calcium: float = 0
|
||||
|
||||
class FoodResponse(BaseModel):
|
||||
id: int
|
||||
name: str
|
||||
serving_size: str
|
||||
serving_unit: str
|
||||
calories: float
|
||||
protein: float
|
||||
carbs: float
|
||||
fat: float
|
||||
fiber: float
|
||||
sugar: float
|
||||
sodium: float
|
||||
calcium: float
|
||||
|
||||
class Config:
|
||||
from_attributes = True
|
||||
|
||||
class MealCreate(BaseModel):
|
||||
name: str
|
||||
meal_type: str
|
||||
foods: List[dict] # [{"food_id": 1, "quantity": 1.5}]
|
||||
|
||||
# Database dependency
|
||||
def get_db():
|
||||
db = SessionLocal()
|
||||
try:
|
||||
yield db
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
# Create tables
|
||||
Base.metadata.create_all(bind=engine)
|
||||
|
||||
# Utility functions
|
||||
def calculate_meal_nutrition(meal, db: Session):
|
||||
"""Calculate total nutrition for a meal"""
|
||||
totals = {
|
||||
'calories': 0, 'protein': 0, 'carbs': 0, 'fat': 0,
|
||||
'fiber': 0, 'sugar': 0, 'sodium': 0, 'calcium': 0
|
||||
}
|
||||
|
||||
for meal_food in meal.meal_foods:
|
||||
food = meal_food.food
|
||||
quantity = meal_food.quantity
|
||||
|
||||
totals['calories'] += food.calories * quantity
|
||||
totals['protein'] += food.protein * quantity
|
||||
totals['carbs'] += food.carbs * quantity
|
||||
totals['fat'] += food.fat * quantity
|
||||
totals['fiber'] += food.fiber * quantity
|
||||
totals['sugar'] += food.sugar * quantity
|
||||
totals['sodium'] += food.sodium * quantity
|
||||
totals['calcium'] += food.calcium * quantity
|
||||
|
||||
# Calculate percentages
|
||||
total_cals = totals['calories']
|
||||
if total_cals > 0:
|
||||
totals['protein_pct'] = round((totals['protein'] * 4 / total_cals) * 100, 1)
|
||||
totals['carbs_pct'] = round((totals['carbs'] * 4 / total_cals) * 100, 1)
|
||||
totals['fat_pct'] = round((totals['fat'] * 9 / total_cals) * 100, 1)
|
||||
totals['net_carbs'] = totals['carbs'] - totals['fiber']
|
||||
|
||||
return totals
|
||||
|
||||
def calculate_day_nutrition(plans, db: Session):
|
||||
"""Calculate total nutrition for a day's worth of meals"""
|
||||
day_totals = {
|
||||
'calories': 0, 'protein': 0, 'carbs': 0, 'fat': 0,
|
||||
'fiber': 0, 'sugar': 0, 'sodium': 0, 'calcium': 0
|
||||
}
|
||||
|
||||
for plan in plans:
|
||||
meal_nutrition = calculate_meal_nutrition(plan.meal, db)
|
||||
for key in day_totals:
|
||||
if key in meal_nutrition:
|
||||
day_totals[key] += meal_nutrition[key]
|
||||
|
||||
# Calculate percentages
|
||||
total_cals = day_totals['calories']
|
||||
if total_cals > 0:
|
||||
day_totals['protein_pct'] = round((day_totals['protein'] * 4 / total_cals) * 100, 1)
|
||||
day_totals['carbs_pct'] = round((day_totals['carbs'] * 4 / total_cals) * 100, 1)
|
||||
day_totals['fat_pct'] = round((day_totals['fat'] * 9 / total_cals) * 100, 1)
|
||||
day_totals['net_carbs'] = day_totals['carbs'] - day_totals['fiber']
|
||||
|
||||
return day_totals
|
||||
|
||||
# Routes
|
||||
@app.get("/", response_class=HTMLResponse)
|
||||
async def root(request: Request):
|
||||
return templates.TemplateResponse("index.html", {"request": request})
|
||||
|
||||
# Foods tab
|
||||
@app.get("/foods", response_class=HTMLResponse)
|
||||
async def foods_page(request: Request, db: Session = Depends(get_db)):
|
||||
foods = db.query(Food).all()
|
||||
return templates.TemplateResponse("foods.html", {"request": request, "foods": foods})
|
||||
|
||||
@app.post("/foods/add")
|
||||
async def add_food(request: Request, db: Session = Depends(get_db),
|
||||
name: str = Form(...), serving_size: str = Form(...),
|
||||
serving_unit: str = Form(...), calories: float = Form(...),
|
||||
protein: float = Form(...), carbs: float = Form(...),
|
||||
fat: float = Form(...), fiber: float = Form(0),
|
||||
sugar: float = Form(0), sodium: float = Form(0),
|
||||
calcium: float = Form(0)):
|
||||
|
||||
food = Food(
|
||||
name=name, serving_size=serving_size, serving_unit=serving_unit,
|
||||
calories=calories, protein=protein, carbs=carbs, fat=fat,
|
||||
fiber=fiber, sugar=sugar, sodium=sodium, calcium=calcium
|
||||
)
|
||||
db.add(food)
|
||||
db.commit()
|
||||
return {"status": "success", "message": "Food added successfully"}
|
||||
|
||||
# Meals tab
|
||||
@app.get("/meals", response_class=HTMLResponse)
|
||||
async def meals_page(request: Request, db: Session = Depends(get_db)):
|
||||
meals = db.query(Meal).all()
|
||||
foods = db.query(Food).all()
|
||||
return templates.TemplateResponse("meals.html",
|
||||
{"request": request, "meals": meals, "foods": foods})
|
||||
|
||||
@app.post("/meals/add")
|
||||
async def add_meal(request: Request, db: Session = Depends(get_db),
|
||||
name: str = Form(...), meal_type: str = Form(...)):
|
||||
|
||||
meal = Meal(name=name, meal_type=meal_type)
|
||||
db.add(meal)
|
||||
db.commit()
|
||||
db.refresh(meal)
|
||||
return {"status": "success", "meal_id": meal.id}
|
||||
|
||||
@app.post("/meals/{meal_id}/add_food")
|
||||
async def add_food_to_meal(meal_id: int, food_id: int = Form(...),
|
||||
quantity: float = Form(...), db: Session = Depends(get_db)):
|
||||
|
||||
meal_food = MealFood(meal_id=meal_id, food_id=food_id, quantity=quantity)
|
||||
db.add(meal_food)
|
||||
db.commit()
|
||||
return {"status": "success"}
|
||||
|
||||
# Plan tab
|
||||
@app.get("/plan", response_class=HTMLResponse)
|
||||
async def plan_page(request: Request, person: str = "Person A", db: Session = Depends(get_db)):
|
||||
from datetime import date, timedelta
|
||||
|
||||
# Get 2 weeks starting from today
|
||||
start_date = date.today()
|
||||
dates = [start_date + timedelta(days=i) for i in range(14)]
|
||||
|
||||
# Get plans for the person
|
||||
plans = {}
|
||||
for d in dates:
|
||||
day_plans = db.query(Plan).filter(Plan.person == person, Plan.date == d).all()
|
||||
plans[d] = day_plans
|
||||
|
||||
# Calculate daily totals
|
||||
daily_totals = {}
|
||||
for d in dates:
|
||||
daily_totals[d] = calculate_day_nutrition(plans[d], db)
|
||||
|
||||
meals = db.query(Meal).all()
|
||||
|
||||
return templates.TemplateResponse("plan.html", {
|
||||
"request": request, "person": person, "dates": dates,
|
||||
"plans": plans, "daily_totals": daily_totals, "meals": meals
|
||||
})
|
||||
|
||||
@app.post("/plan/add")
|
||||
async def add_to_plan(request: Request, person: str = Form(...),
|
||||
plan_date: str = Form(...), meal_id: int = Form(...),
|
||||
db: Session = Depends(get_db)):
|
||||
|
||||
plan = Plan(person=person, date=datetime.strptime(plan_date, "%Y-%m-%d").date(), meal_id=meal_id)
|
||||
db.add(plan)
|
||||
db.commit()
|
||||
return {"status": "success"}
|
||||
|
||||
# Detailed planner tab
|
||||
@app.get("/detailed", response_class=HTMLResponse)
|
||||
async def detailed_page(request: Request, person: str = "Person A",
|
||||
plan_date: str = None, db: Session = Depends(get_db)):
|
||||
|
||||
if not plan_date:
|
||||
plan_date = date.today().strftime("%Y-%m-%d")
|
||||
|
||||
selected_date = datetime.strptime(plan_date, "%Y-%m-%d").date()
|
||||
|
||||
# Get all plans for the selected day
|
||||
plans = db.query(Plan).filter(Plan.person == person, Plan.date == selected_date).all()
|
||||
|
||||
# Group by meal type and calculate nutrition
|
||||
meal_details = []
|
||||
for plan in plans:
|
||||
meal_nutrition = calculate_meal_nutrition(plan.meal, db)
|
||||
meal_details.append({
|
||||
'plan': plan,
|
||||
'nutrition': meal_nutrition,
|
||||
'foods': plan.meal.meal_foods
|
||||
})
|
||||
|
||||
# Calculate day totals
|
||||
day_totals = calculate_day_nutrition(plans, db)
|
||||
|
||||
return templates.TemplateResponse("detailed.html", {
|
||||
"request": request, "person": person, "selected_date": selected_date,
|
||||
"meal_details": meal_details, "day_totals": day_totals
|
||||
})
|
||||
|
||||
if __name__ == "__main__":
|
||||
import uvicorn
|
||||
uvicorn.run(app, host="0.0.0.0", port=8999)
|
||||
Reference in New Issue
Block a user