twschiller/open-synthesis

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

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


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":
            self.report(send_digest_emails(DigestFrequency.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
                self.report(send_digest_emails(DigestFrequency.weekly))
            else:
                msg = "Skipping weekly digest until day %s (current=%s)" % (
                    digest_day,
                    current_day,
                )
                self.stdout.write(self.style.WARNING(msg))  # pylint: disable=no-member
        else:
            raise CommandError('Expected frequency "daily" or "weekly"')