from django.core.management.base import BaseCommand
from home.models import EventPage, EventOccurrence
from django.utils import timezone
from datetime import timedelta


class Command(BaseCommand):
    help = "Generate EventOccurrence objects for all recurring events (for daily cron/task). Also deletes occurrences older than 1 month."

    def handle(self, *args, **options):
        # Delete occurrences older than 1 month
        cutoff = timezone.now() - timedelta(days=30)
        deleted, _ = EventOccurrence.objects.filter(end__lt=cutoff).delete()
        self.stdout.write(self.style.WARNING(f"Deleted {deleted} old occurrences."))

        all_events = EventPage.objects.filter()
        recurrent_events = EventPage.objects.filter(recurrence_enabled=True)

        for event in all_events:
            event.generate_occurrences()

        self.stdout.write(
            self.style.SUCCESS(
                f"Generated occurrences for {all_events.count()} events."
            )
        )
        self.stdout.write(
            self.style.SUCCESS(
                f"Processed {recurrent_events.count()} recurring events."
            )
        )