darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-21 07:51:13 -05:00
Code Issues Packages Projects Releases Wiki Activity

group results by service using itertools

Browse Source 
allows us to nicely reuse the existing format_statistics function
This commit is contained in:
Leo Hemsted
2016-08-19 16:36:20 +01:00
parent ebb13a1251
commit 00d19f63f0
3 changed files with 14 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
app/service/rest.py
View File

@@ -1,3 +1,5 @@
import itertools
from flask import (
jsonify,
request,
@@ -21,7 +23,8 @@ from app.dao.services_dao import (
dao_remove_user_from_service,
dao_fetch_stats_for_service,
dao_fetch_todays_stats_for_service,
dao_fetch_weekly_historical_stats_for_service
dao_fetch_weekly_historical_stats_for_service,
dao_fetch_todays_stats_for_all_services
)
from app.dao import notifications_dao
from app.dao.provider_statistics_dao import get_fragment_count
@@ -51,9 +54,9 @@ def get_services():
user_id = request.args.get('user_id', None)
if user_id:
services = dao_fetch_all_services_by_user(user_id)
elif request.args.get('detailed') == 'True':
return get_detailed_services()
else:
if request.args.get('detailed') == 'True':
return get_detailed_services()
services = dao_fetch_all_services()
data = service_schema.dump(services, many=True).data
return jsonify(data=data)
@@ -251,12 +254,10 @@ def get_detailed_service(service_id, today_only=False):
def get_detailed_services():
services = {service.id: service for service in dao_fetch_all_services()}
stats = dao_fetch_todays_stats_for_all_services(service_id)
stats = dao_fetch_todays_stats_for_all_services()
for row in stats:
services[row.service_id].statistics
# todo: how do we separate rows of statistics by service?
service.statistics = statistics.format_statistics(stats)
for service_id, rows in itertools.groupby(stats, lambda x: x.service_id):
services[service_id].statistics = statistics.format_statistics(rows)
data = detailed_service_schema.dump(service).data
data = detailed_service_schema.dump(service, many=True).data
return jsonify(data=data)