From 97473a46e2495102712429fb7d53ac469a7ac075 Mon Sep 17 00:00:00 2001 From: James Graham Date: Tue, 9 Mar 2021 17:43:46 +0000 Subject: [PATCH] fix: attribute error on org list page Error for organisations with no AnswerSet --- people/views/organisation.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/people/views/organisation.py b/people/views/organisation.py index b561118..265f142 100644 --- a/people/views/organisation.py +++ b/people/views/organisation.py @@ -71,14 +71,18 @@ class OrganisationListView(LoginRequiredMixin, ListView): answers = organisation.current_answers country = 'Unknown' - if len(answers.countries) == 1: - country = answers.countries[0].name + try: + if len(answers.countries) == 1: + country = answers.countries[0].name - elif len(answers.countries) > 1: - country = 'International' + elif len(answers.countries) > 1: + country = 'International' - if answers.is_partner_organisation: - country = f'{settings.PARENT_PROJECT_NAME} partners' + if answers.is_partner_organisation: + country = f'{settings.PARENT_PROJECT_NAME} partners' + + except AttributeError: + # Organisation has no AnswerSet - country is 'Unknown' orgs = orgs_by_country.get(country, []) orgs.append(organisation)