fix: attribute error on org list page

Error for organisations with no AnswerSet
This commit is contained in:
James Graham
2021-03-09 17:43:46 +00:00
parent 989c8141b3
commit 97473a46e2

View File

@@ -71,6 +71,7 @@ class OrganisationListView(LoginRequiredMixin, ListView):
answers = organisation.current_answers answers = organisation.current_answers
country = 'Unknown' country = 'Unknown'
try:
if len(answers.countries) == 1: if len(answers.countries) == 1:
country = answers.countries[0].name country = answers.countries[0].name
@@ -80,6 +81,9 @@ class OrganisationListView(LoginRequiredMixin, ListView):
if answers.is_partner_organisation: if answers.is_partner_organisation:
country = f'{settings.PARENT_PROJECT_NAME} partners' country = f'{settings.PARENT_PROJECT_NAME} partners'
except AttributeError:
# Organisation has no AnswerSet - country is 'Unknown'
orgs = orgs_by_country.get(country, []) orgs = orgs_by_country.get(country, [])
orgs.append(organisation) orgs.append(organisation)
orgs_by_country[country] = orgs orgs_by_country[country] = orgs