sync - build workin

main.py

@@ -13,7 +13,6 @@ from fastapi.responses import HTMLResponse
 from sqlalchemy import create_engine, Column, Integer, String, Float, DateTime, ForeignKey, Text, Date, Boolean
 from sqlalchemy import or_
 from sqlalchemy.orm import sessionmaker, Session, relationship, declarative_base
-from sqlalchemy.orm import joinedload
 from pydantic import BaseModel, ConfigDict
 from typing import List, Optional
 from datetime import date, datetime
@@ -1567,6 +1566,113 @@ async def weekly_menu_page(request: Request, db: Session = Depends(get_db)):
         "templates": templates_list
     })
 
+@app.post("/weeklymenu/create")
+async def create_weekly_menu(request: Request, db: Session = Depends(get_db)):
+    """Create a new weekly menu with template assignments."""
+    try:
+        form_data = await request.form()
+        name = form_data.get("name")
+        template_assignments_str = form_data.get("template_assignments")
+
+        if not name:
+            return {"status": "error", "message": "Weekly menu name is required"}
+
+        # Check if weekly menu already exists
+        existing_weekly_menu = db.query(WeeklyMenu).filter(WeeklyMenu.name == name).first()
+        if existing_weekly_menu:
+            return {"status": "error", "message": f"Weekly menu with name '{name}' already exists"}
+
+        weekly_menu = WeeklyMenu(name=name)
+        db.add(weekly_menu)
+        db.flush() # To get the weekly_menu.id
+
+        if template_assignments_str:
+            assignments = template_assignments_str.split(',')
+            for assignment in assignments:
+                day_of_week_str, template_id_str = assignment.split(':', 1)
+                day_of_week = int(day_of_week_str)
+                template_id = int(template_id_str)
+
+                # Check if template exists
+                template = db.query(Template).filter(Template.id == template_id).first()
+                if not template:
+                    raise HTTPException(status_code=400, detail=f"Template with ID {template_id} not found.")
+
+                weekly_menu_day = WeeklyMenuDay(
+                    weekly_menu_id=weekly_menu.id,
+                    day_of_week=day_of_week,
+                    template_id=template_id
+                )
+                db.add(weekly_menu_day)
+        
+        db.commit()
+        return {"status": "success", "message": "Weekly menu created successfully"}
+
+    except Exception as e:
+        db.rollback()
+        logging.error(f"Error creating weekly menu: {e}")
+        return {"status": "error", "message": str(e)}
+
+@app.post("/weeklymenu/{weekly_menu_id}/apply")
+async def apply_weekly_menu(weekly_menu_id: int, request: Request, db: Session = Depends(get_db)):
+    """Apply a weekly menu to a person's plan for a specific week."""
+    try:
+        from datetime import datetime, timedelta
+        form_data = await request.form()
+        person = form_data.get("person")
+        week_start_date_str = form_data.get("week_start_date")
+        confirm_overwrite = form_data.get("confirm_overwrite") == "true"
+
+        if not person or not week_start_date_str:
+            return {"status": "error", "message": "Person and week start date are required."}
+
+        week_start_date = datetime.fromisoformat(week_start_date_str).date()
+
+        weekly_menu = db.query(WeeklyMenu).filter(WeeklyMenu.id == weekly_menu_id).first()
+        if not weekly_menu:
+            return {"status": "error", "message": "Weekly menu not found."}
+
+        # Check if there are existing plans for the target week
+        existing_plans = db.query(Plan).filter(
+            Plan.person == person,
+            Plan.date >= week_start_date,
+            Plan.date < (week_start_date + timedelta(days=7))
+        ).all()
+
+        if existing_plans and not confirm_overwrite:
+            return {"status": "confirm_overwrite", "message": "Meals already planned for this week. Do you want to overwrite them?"}
+
+        # If confirmed or no existing plans, delete existing plans for the week
+        if existing_plans:
+            db.query(Plan).filter(
+                Plan.person == person,
+                Plan.date >= week_start_date,
+                Plan.date < (week_start_date + timedelta(days=7))
+            ).delete()
+            db.flush()
+
+        # Apply each day of the weekly menu
+        for weekly_menu_day in weekly_menu.weekly_menu_days:
+            target_date = week_start_date + timedelta(days=weekly_menu_day.day_of_week)
+            template = weekly_menu_day.template
+
+            if template:
+                for template_meal in template.template_meals:
+                    plan = Plan(
+                        person=person,
+                        date=target_date,
+                        meal_id=template_meal.meal_id,
+                        meal_time=template_meal.meal_time
+                    )
+                    db.add(plan)
+        
+        db.commit()
+        return {"status": "success", "message": "Weekly menu applied successfully."}
+
+    except Exception as e:
+        db.rollback()
+        logging.error(f"Error applying weekly menu: {e}")
+        return {"status": "error", "message": str(e)}
 
 # Plan tab
 @app.get("/plan", response_class=HTMLResponse)
@@ -2199,7 +2305,7 @@ async def templates_page(request: Request, db: Session = Depends(get_db)):
 @app.get("/api/templates", response_model=List[TemplateDetail])
 async def get_templates_api(db: Session = Depends(get_db)):
     """API endpoint to get all templates with meal details."""
-    templates = db.query(Template).options(joinedload(Template.template_meals).joinedload(TemplateMeal.meal)).all()
+    templates = db.query(Template).options(orm.joinedload(Template.template_meals).joinedload(TemplateMeal.meal)).all()
     
     results = []
     for t in templates: