from django.core.management.base import BaseCommand
from django.utils import timezone
from ...models import Message

class Command(BaseCommand):
    help = 'Clean up messages older than 7 days'

    def handle(self, *args, **options):
        one_week_ago = timezone.now() - timezone.timedelta(days=7)
        deleted_count, _ = Message.objects.filter(created_at__lt=one_week_ago).delete()
        self.stdout.write(self.style.SUCCESS(f'Successfully deleted {deleted_count} old messages'))