From 41dae13d44699f23a8861d7fcb8e42c9743cbb0f Mon Sep 17 00:00:00 2001
From: iou1name <iou1name@national.shitposting.agency>
Date: Thu, 26 Jul 2018 15:14:04 -0400
Subject: [PATCH] StrictUndefined + empty poll options issue fixed

---
 anonkun.py           | 2 +-
 templates/quest.html | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/anonkun.py b/anonkun.py
index e60eeb5..acfb8cc 100644
--- a/anonkun.py
+++ b/anonkun.py
@@ -59,7 +59,7 @@ app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024
 app.config['SESSION_TYPE'] = 'filesystem'
 app.jinja_env.trim_blocks = True
 app.jinja_env.lstrip_blocks = True
-#app.jinja_env.undefined = "StrictUndefined"
+app.jinja_env.undefined = "StrictUndefined"
 Session(app)
 socketio.init_app(app)
 paranoid = Paranoid(app)
diff --git a/templates/quest.html b/templates/quest.html
index f401c2f..e74bf7d 100644
--- a/templates/quest.html
+++ b/templates/quest.html
@@ -69,14 +69,14 @@
 							<col{% if quest_post[0] != open_post_id %} style="visibility: collapse;"{% endif %}/>
 							<col/>
 							<col/>
-							{% for option in options[quest_post[0]] %}
+							{% for option in options.get(quest_post[0], []) %}
 							<tr id="optionRow-{{ option[0] }}">
 								<td class="pollCheckBox">
-									<input type="checkbox" {% if ip_address in poll_votes[option[0]] %}checked="true"{% endif %} id="pollInput-{{ option[0] }}" onchange="pollVote({{ quest_post[0] }}, {{ option[0] }})"/>
+									<input type="checkbox" {% if ip_address in poll_votes.get(option[0], []) %}checked="true"{% endif %} id="pollInput-{{ option[0] }}" onchange="pollVote({{ quest_post[0] }}, {{ option[0] }})"/>
 									<label for="pollInput-{{ option[0] }}"></label>
 								</td>
 								<td class="option_text">{{ option[2] }}</td>
-								<td class="optionVotes">{{ poll_votes[option[0]]|length }}</td>
+								<td class="optionVotes">{{ poll_votes.get(option[0], [])|length }}</td>
 							</tr>
 							{% endfor %}
 						</table>