diff --git a/bigday/views.py b/bigday/views.py index 603f4a7..7d3a94d 100644 --- a/bigday/views.py +++ b/bigday/views.py @@ -1,5 +1,4 @@ from django.shortcuts import render, redirect -from django.urls import reverse from .models import InvitedGuest @@ -18,7 +17,13 @@ def rsvp(request): guest.attending = attending any_attending = any_attending or attending guest.save() - return redirect(reverse('thank-you') + '?attending=' + ('yes' if any_attending else 'no')) + if any_attending: + return redirect('thanks') + else: + return redirect('sorry') def thanks(request): - return render(request, 'thank-you.html', {'attending': request.GET.get('attending') == 'yes'}) \ No newline at end of file + return render(request, 'thank-you.html', {'attending': True}) + +def sorry(request): + return render(request, 'thank-you.html', {'attending': False}) diff --git a/djangoWedding/urls.py b/djangoWedding/urls.py index f8b13b2..a917736 100644 --- a/djangoWedding/urls.py +++ b/djangoWedding/urls.py @@ -6,5 +6,6 @@ urlpatterns = [ path('admin/', admin.site.urls), path('', bigday.views.index, name='index'), path('rsvp',bigday.views.rsvp, name='rsvp'), - path('thanks', bigday.views.thanks, name='thank-you') + path('thanks', bigday.views.thanks, name='thanks'), + path('sorry', bigday.views.sorry, name='sorry'), ]