from django.core.management.base import BaseCommand
from django.db.models import Min

from hc.api.models import Notification, Check


class Command(BaseCommand):
    help = "Prune stored notifications"

    def handle(self, *args, **options):
        total = 0

        q = Check.objects.filter(n_pings__gt=100)
        q = q.annotate(min_ping_date=Min("ping__created"))
        for check in q:
            qq = Notification.objects.filter(
                owner_id=check.id, created__lt=check.min_ping_date
            )

            num_deleted, _ = qq.delete()
            total += num_deleted

        return "Done! Pruned %d notifications." % total
