From eaef0944440a7c5c514c4e7c85b130cc1f99061d Mon Sep 17 00:00:00 2001
From: iou1name <iou1name@national.shitposting.agency>
Date: Wed, 5 Sep 2018 12:22:45 -0400
Subject: [PATCH] page titles populate drop down on quest page

---
 create_quest/views.py         | 8 +++++++-
 quest/jinja2/quest/quest.html | 6 +++---
 quest/views.py                | 9 +++++----
 3 files changed, 15 insertions(+), 8 deletions(-)

diff --git a/create_quest/views.py b/create_quest/views.py
index 078c9c4..96ae897 100644
--- a/create_quest/views.py
+++ b/create_quest/views.py
@@ -6,7 +6,7 @@ from django.contrib import messages
 from django.shortcuts import redirect, render
 
 from .forms import QuestForm, PostForm
-from quest.models import Quest, Post
+from quest.models import Quest, Post, PageTitle
 
 def index(request):
 	"""
@@ -21,6 +21,12 @@ def index(request):
 			quest.save()
 			post.quest = quest
 			post.save()
+			page = PageTitle(
+				quest=quest,
+				page_num=1,
+				title="Page 1"
+			)
+			page.save()
 			return redirect('quest:quest', quest_id=quest.id)
 	else:
 		quest_form = QuestForm()
diff --git a/quest/jinja2/quest/quest.html b/quest/jinja2/quest/quest.html
index f46e7dc..82c1eb0 100644
--- a/quest/jinja2/quest/quest.html
+++ b/quest/jinja2/quest/quest.html
@@ -19,9 +19,9 @@
 {% endif %}
 	<li>
 		<select onChange="window.location.href=this.value">
-			{# for page in pages %}
-			<option value="{{ url_for('.quest', quest_id=quest_id, page_num=page[1]) }}"{% if page[1] == page_num %} selected="yes"{% endif %}>{{ page[2] }}</option>
-			{% endfor #}
+			{% for page in pages %}
+			<option value="{{ url('quest:quest', args=[quest_id, page.page_num]) }}"{% if page.page_num == page_num %} selected="yes"{% endif %}>{{ page.title }}</option>
+			{% endfor %}
 		</select>
 	</li>
 {% endblock %}
diff --git a/quest/views.py b/quest/views.py
index 509fb01..677ca6c 100644
--- a/quest/views.py
+++ b/quest/views.py
@@ -5,7 +5,7 @@ Quest and quest accessory views.
 from django.shortcuts import render
 from django.http import HttpResponse
 
-from .models import Quest, DiceRoll, PollOption, PollVote
+from .models import Quest, DiceRoll, PollOption, PollVote, PageTitle
 
 def index(request):
 	"""
@@ -20,11 +20,12 @@ def quest(request, quest_id, page_num=1):
 	Arbituary quest page view.
 	"""
 	quest = Quest.objects.get(id=quest_id)
+	pages = PageTitle.objects.filter(quest=quest).order_by('page_num')
 	messages = quest.message_set.all()
 	posts = quest.post_set.all()
-	dice_rolls = DiceRoll.objects.filter(dicecall__post__quest__id=quest_id)
-	poll_options = PollOption.objects.filter(poll__post__quest__id=quest_id)
-	poll_votes = PollVote.objects.filter(option__poll__post__quest__id=quest_id)
+	dice_rolls = DiceRoll.objects.filter(dicecall__post__quest=quest)
+	poll_options = PollOption.objects.filter(poll__post__quest=quest)
+	poll_votes = PollVote.objects.filter(option__poll__post__quest=quest)
 	ip_address = request.META['REMOTE_ADDR']
 	context = locals()
 	return render(request, 'quest/quest.html', context)