def edit_meal(request, meal_id):
context_dict = {}
user = User.objects.get(username=request.user)
# Check to see if meal exists.
meal = get_object_or_404(Meal, pk=meal_id)
# Check to see if user has access to that meal.
if user != meal.user:
raise PermissionDenied
if request.method == "POST":
form = MealForm(request.POST, instance=meal)
if form.is_valid():
meal = form.save()
messages.success(request, "Your meal has been updated.")
return HttpResponseRedirect("/meal_tracker/dashboard/")
else:
form = MealForm(instance=meal)
context_dict["form"] = form
context_dict["page_title"] = "Edit Meal"
context_dict["h1_header"] = "Edit Your Meal"
context_dict["form_action_tag"] = "edit_meal/" + meal_id
return render(request, "meal_tracker/add_edit_meal.html", context_dict)
def add_meal(request):
context_dict = {}
user = User.objects.get(username=request.user)
if request.method == "POST":
meal = Meal(user=user)
form = MealForm(request.POST, instance=meal)
if form.is_valid():
meal = form.save()
messages.success(request, "Your meal has been added.")
return HttpResponseRedirect("/meal_tracker/dashboard/")
else:
print form.errors
else:
form = MealForm()
context_dict["form"] = form
context_dict["page_title"] = "Add Meal"
context_dict["h1_header"] = "Add a New Meal"
context_dict["form_action_tag"] = "add_meal"
return render(request, "meal_tracker/add_edit_meal.html", context_dict)