From e6fbea7b5ac6ccb91e81be5ebc86d9f7480b5e25 Mon Sep 17 00:00:00 2001 From: Mike Olund Date: Mon, 27 Mar 2017 18:09:50 -0700 Subject: [PATCH] Fixed bug --- edivorce/apps/core/utils/user_response.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/edivorce/apps/core/utils/user_response.py b/edivorce/apps/core/utils/user_response.py index ce4166a4..cc07f546 100644 --- a/edivorce/apps/core/utils/user_response.py +++ b/edivorce/apps/core/utils/user_response.py @@ -1,3 +1,4 @@ +from django.db.models import Count from edivorce.apps.core.models import UserResponse, Question from edivorce.apps.core.utils.question_step_mapping import question_step_mapping @@ -107,7 +108,13 @@ def __get_data(bceid_user): MARRIED = 'Legally married' responses = UserResponse.objects.filter(bceid_user=bceid_user) - married = responses.get(question_id='married_marriage_like').value != COMMON_LAW + married_status = responses.filter(question_id='married_marriage_like') + + if married_status.count() > 0: + married = married_status[0].value != COMMON_LAW + else: + married = False + married_questions = list( Question.objects.filter(reveal_response=MARRIED).values_list("key", flat=True)) return married, married_questions, responses