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.
57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
8 years ago
|
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.template import Context
|
||
|
|
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:
|
||
|
|
ctx = Context(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]))
|
||
|
|
print("sending -%d days notify to %s ..." % (v, u.user.email))
|
||
|
|
|
||
|
|
send_mass_mail(emails)
|
||
|
|
qs.update(last_expiry_notice=timezone.now())
|