Guard logic with if condition to prevent attribute error

onaio · Aug 6, 2024 · 9b3af9b · 9b3af9b
1 parent 085884a
commit 9b3af9b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/tally_ho/apps/tally/views/reports/administrative_areas_reports.py b/tally_ho/apps/tally/views/reports/administrative_areas_reports.py
@@ -2025,6 +2025,8 @@ def create_results_power_point_headers(tally_id, filtered_electrol_races, qs):
         registrants_in_processed_stations = 0
         for station in station_ids_by_races:
             station_obj = stations_by_id.get(station.get('id'))
+            if station_obj is None:
+                continue
             # Calculate stations processed and total registrants
             form_states =\
                 ResultForm.objects.filter(

diff --git a/tally_ho/apps/tally/views/reports/turnout_reports_by_admin_areas.py b/tally_ho/apps/tally/views/reports/turnout_reports_by_admin_areas.py
@@ -181,6 +181,8 @@ def get_initial_queryset(self):
             registrants_in_processed_stations = 0
             for station in station_ids_by_race:
                 station_obj = stations_by_id.get(station.get('id'))
+                if station_obj is None:
+                    continue
                 # Calculate stations processed and total registrants
                 form_states = get_result_forms_for_station_in_admin_area(
                     tally_id,