notifications-admin/app/main/views/history.py

from collections import defaultdict
from operator import attrgetter

from flask import render_template, request

from app import current_service, format_date_numeric
from app.main import main
from app.models.event import APIKeyEvent, APIKeyEvents, ServiceEvents
from app.utils import user_has_permissions


@main.route("/services/<uuid:service_id>/history")
@user_has_permissions('manage_service')
def history(service_id):

    events = _get_events(current_service.id, request.args.get('selected'))

    return render_template(
        'views/temp-history.html',
        days=_chunk_events_by_day(events),
        show_navigation=request.args.get('selected') or any(
            isinstance(event, APIKeyEvent) for event in events
        ),
        user_getter=current_service.active_users.get_name_from_id,
    )


def _get_events(service_id, selected):
    if selected == 'api':
        return APIKeyEvents(service_id)
    if selected == 'service':
        return ServiceEvents(service_id)
    return APIKeyEvents(service_id) + ServiceEvents(service_id)


def _chunk_events_by_day(events):

    days = defaultdict(list)

    for event in sorted(events, key=attrgetter('time'), reverse=True):
        days[format_date_numeric(event.time)].append(event)

    return sorted(days.items(), reverse=True)