From 6e07b03f01a57ebe4d8f4fbd533c00c9686c993b Mon Sep 17 00:00:00 2001 From: Melissa Autumn Date: Thu, 8 Aug 2024 09:21:26 -0700 Subject: [PATCH] Fix download data not working --- backend/src/appointment/controller/data.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/backend/src/appointment/controller/data.py b/backend/src/appointment/controller/data.py index 71e066c93..abf922e6d 100644 --- a/backend/src/appointment/controller/data.py +++ b/backend/src/appointment/controller/data.py @@ -11,7 +11,7 @@ def model_to_csv_buffer(models): """Dumps a DeclarationBase model to csv and returns an in-memory buffer""" - if len(models) == 0: + if len(models) == 0 or not models[0]: return StringIO() # Don't write out these columns @@ -46,7 +46,9 @@ def download(db, subscriber: Subscriber): schedules = repo.schedule.get_by_subscriber(db, subscriber.id) availability = [repo.schedule.get_availability(db, schedule.id) for schedule in schedules] invite = repo.invite.get_by_subscriber(db, subscriber.id) - waiting_list = invite.waiting_list + waiting_list = None + if invite: + waiting_list = invite.waiting_list # Convert models to csv attendee_buffer = model_to_csv_buffer(attendees)