ccvpn/lambdainst/management/commands/expire_notify.py

from django.core.management.base import BaseCommand

from datetime import timedelta

from django.db.models import Q, F
from django.conf import settings
from django.utils import timezone
from django.template.loader import get_template
from django.core.mail import send_mass_mail

from lambdainst.models import VPNUser

ROOT_URL = settings.ROOT_URL
SITE_NAME = settings.TICKETS_SITE_NAME

NOTIFY_DAYS_BEFORE = settings.NOTIFY_DAYS_BEFORE
assert isinstance(NOTIFY_DAYS_BEFORE, (list, tuple, set))


def get_next_expirations(days=3):
    """ Gets users whose subscription will expire in some days """

    limit_date = timezone.now() + timedelta(days=days)

    users = VPNUser.objects.exclude(user__email__exact='')

    users = users.filter(expiration__gt=timezone.now())  # Not expired
    users = users.filter(expiration__lt=limit_date)  # Expire in a few days

    # Make sure we dont send the notice twice
    users = users.filter(Q(last_expiry_notice__isnull=True)
                         | Q(expiration__gt=F('last_expiry_notice')
                             + timedelta(days=days)))
    return users


class Command(BaseCommand):
    help = "Notify users near the end of their subscription"

    def handle(self, *args, **options):
        from_email = settings.DEFAULT_FROM_EMAIL

        for v in NOTIFY_DAYS_BEFORE:
            emails = []
            qs = get_next_expirations(v)
            users = list(qs)
            for u in users:
                # Ignore users with active subscriptions
                # They will get notified only if it gets cancelled (payments
                # processors will cancel after a few failed payments)
                if u.get_subscription():
                    continue

                ctx = dict(site_name=SITE_NAME, user=u.user,
                           exp=u.expiration, url=ROOT_URL)
                text = get_template('lambdainst/mail_expire_soon.txt').render(ctx)
                emails.append(("CCVPN Expiration", text, from_email, [u.user.email]))
                self.stdout.write("sending -%d days notify to %s ..." % (v, u.user.email))

            send_mass_mail(emails)
            qs.update(last_expiry_notice=timezone.now())