estructuring how to get total_notifications.

Signed-off-by: Cliff Hill <Clifford.hill@gsa.gov>
This commit is contained in:
Cliff Hill
2025-01-10 13:47:43 -05:00
parent b6337ad807
commit f1cc6aa400
3 changed files with 40 additions and 16 deletions

View File

@@ -230,9 +230,9 @@ def get_service_statistics_for_specific_days(service_id, start, days=1):
end_date = datetime.strptime(start, "%Y-%m-%d")
start_date = end_date - timedelta(days=days - 1)
results = dao_fetch_stats_for_service_from_days(service_id, start_date, end_date)
total_notifications, results = dao_fetch_stats_for_service_from_days(service_id, start_date, end_date,)
stats = get_specific_days_stats(results, start_date, days=days)
stats = get_specific_days_stats(results, start_date, days=days, total_notifications=total_notifications,)
return stats
@@ -678,7 +678,8 @@ def get_single_month_notification_stats_for_service(service_id):
month_year = datetime(year, month, 10, 00, 00, 00)
start_date, end_date = get_month_start_and_end_date_in_utc(month_year)
results = dao_fetch_stats_for_service_from_days(service_id, start_date, end_date)
# First element is total notifications used elsewhere.
__, results = dao_fetch_stats_for_service_from_days(service_id, start_date, end_date)
stats = get_specific_days_stats(results, start_date, end_date=end_date)
return jsonify(stats)