refactor: prep for fallback to organisation latlng

This commit is contained in:
James Graham
2021-01-15 12:31:49 +00:00
parent 90c40b2e2e
commit 95fda6a3d5
3 changed files with 83 additions and 51 deletions

View File

@@ -106,15 +106,21 @@ def get_map_data(person: models.Person) -> typing.Dict[str, typing.Any]:
try:
latitude = answer_set.latitude or None
longitude = answer_set.longitude or None
organisation = answer_set.organisation.name or None
country = answer_set.country_of_residence.name or None
except AttributeError:
latitude = None
longitude = None
organisation = None
country = None
return {
'name': person.name,
'lat': latitude,
'lng': longitude,
'organisation': organisation,
'country': country,
'url': reverse('people:person.detail', kwargs={'pk': person.pk})
}