GlitchTip/performance/filters.py

from django.db.models import Count
from django_filters import rest_framework as filters
from projects.models import Project
from .models import TransactionGroup, avg_transactionevent_time


class TransactionGroupFilter(filters.FilterSet):
    start = filters.IsoDateTimeFilter(
        field_name="transactionevent__created",
        lookup_expr="gte",
        label="Transaction start date",
    )
    end = filters.IsoDateTimeFilter(
        field_name="transactionevent__created",
        lookup_expr="lte",
        label="Transaction end date",
    )
    project = filters.ModelMultipleChoiceFilter(queryset=Project.objects.all())

    class Meta:
        model = TransactionGroup
        fields = ["project", "start", "end"]

    def filter_queryset(self, queryset):
        queryset = super().filter_queryset(queryset)
        # This annotation must be applied after any related transactionevent filter
        queryset = queryset.annotate(
            avg_duration=avg_transactionevent_time,
            transaction_count=Count("transactionevent"),
        )
        return queryset