From 40517817b95d34f1505e7b161cbe3f0745cd823e Mon Sep 17 00:00:00 2001 From: natdeanlewissoftwire Date: Thu, 14 Dec 2023 12:03:16 +0000 Subject: [PATCH] feat: optimise locations count --- app/helpers/schemes_helper.rb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/app/helpers/schemes_helper.rb b/app/helpers/schemes_helper.rb index 98bd2d62f..88022dd53 100644 --- a/app/helpers/schemes_helper.rb +++ b/app/helpers/schemes_helper.rb @@ -60,10 +60,10 @@ module SchemesHelper when "schemes" "You've selected #{pluralize(scheme_count, 'scheme')}." when "locations" - location_count = schemes.map(&:locations).flatten.count + location_count = Location.where(scheme: schemes).count "You've selected #{pluralize(location_count, 'location')} from #{pluralize(scheme_count, 'scheme')}." when "combined" - location_count = schemes.map(&:locations).flatten.count + location_count = Location.where(scheme: schemes).count "You've selected #{pluralize(scheme_count, 'scheme')} with #{pluralize(location_count, 'location')}. The CSV will have one location per row with scheme details listed for each location." end end