From 4e7c5f9495ec1416549d63be238e01dafc81494d Mon Sep 17 00:00:00 2001
From: tourn <latzer.daniel@gmail.com>
Date: Mon, 30 Mar 2020 22:06:17 +0200
Subject: [PATCH] Add buttons to add a meal plan to a specific point in time

---
 cookbook/templates/meal_plan.html | 1 +
 cookbook/views/new.py             | 7 +++++++
 2 files changed, 8 insertions(+)

diff --git a/cookbook/templates/meal_plan.html b/cookbook/templates/meal_plan.html
index c2956916..0d14d5b1 100644
--- a/cookbook/templates/meal_plan.html
+++ b/cookbook/templates/meal_plan.html
@@ -54,6 +54,7 @@
                     <tr>
                         {% for day_key, days_value in plan_value.days.items %}
                             <td>
+                                <a href="{% url 'new_meal_plan' %}?date={{ day_key|date:'Y-m-d' }}&meal={{ plan_key }}"><i class="fas fa-plus"></i></a><br/>
                                 {% for mp in days_value %}
                                     <a href="{% url 'edit_meal_plan' mp.pk %}"><i class="fas fa-edit"></i></a>
                                     <a href="{% url 'view_recipe' mp.recipe.id %}">{{ mp.recipe.name }}</a><br/>
diff --git a/cookbook/views/new.py b/cookbook/views/new.py
index 7f441253..198aefc8 100644
--- a/cookbook/views/new.py
+++ b/cookbook/views/new.py
@@ -1,4 +1,5 @@
 import re
+from datetime import datetime
 
 from django.contrib import messages
 from django.contrib.auth.decorators import login_required
@@ -120,6 +121,12 @@ class MealPlanCreate(LoginRequiredMixin, CreateView):
     form_class = MealPlanForm
     success_url = reverse_lazy('view_plan')
 
+    def get_initial(self):
+        return dict(
+            meal=self.request.GET['meal'] if 'meal' in self.request.GET else None,
+            date=datetime.strptime(self.request.GET['date'], '%Y-%m-%d') if 'date' in self.request.GET else None
+        )
+
     def form_valid(self, form):
         obj = form.save(commit=False)
         obj.user = self.request.user