"""Django admin command to send email digests."""
from django.core.management.base import BaseCommand, CommandError
from django.conf import settings
from django.utils import timezone

from openach.models import DigestFrequency
from openach.digest import send_digest_emails

class Command(BaseCommand):
    """Django admin command to send out digest emails."""

    help = 'Send digest emails to users subscribed with the given frequency.'

    def add_arguments(self, parser):
        """Add 'frequency' and 'force' command arguments."""
        parser.add_argument('frequency', choices=['daily', 'weekly'])
        parser.add_argument('--force', nargs='?', dest='force', const=True, default=False,
                            help='Force the weekly digest to be sent regardless of day (default: False)')

    def report(self, cnt):
        """Report number of emails sent to STDOUT."""
        msg = 'Sent %s digest emails (%s skipped, %s failed)' % cnt
        self.stdout.write(self.style.SUCCESS(msg))  # pylint: disable=no-member

    def handle(self, *args, **options):
        """Handle the command invocation."""
        if options['frequency'] == 'daily':
        elif options['frequency'] == 'weekly':
            digest_day = getattr(settings, 'DIGEST_WEEKLY_DAY')
            current_day = timezone.now().weekday()
            if current_day == digest_day or options['force']:
                if current_day != digest_day and options['force']:
                    msg = 'Forcing weekly digest to be sent (scheduled=%s, current=%s)' % (digest_day, current_day)
                    self.stdout.write(self.style.WARNING(msg))  # pylint: disable=no-member
                msg = 'Skipping weekly digest until day %s (current=%s)' % (digest_day, current_day)
                self.stdout.write(self.style.WARNING(msg))  # pylint: disable=no-member
            raise CommandError('Expected frequency "daily" or "weekly"')