You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
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())
|