X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/db6ce7eb0542012566e8632cd8db45d647b1d652..d835427a08be1d36308131137e048c3fb441bfaa:/src/messaging/models.py diff --git a/src/messaging/models.py b/src/messaging/models.py index 4705b97bd..fccd64776 100644 --- a/src/messaging/models.py +++ b/src/messaging/models.py @@ -1,10 +1,15 @@ +from datetime import timedelta +from django.apps import apps from django.conf import settings from django.core.mail import send_mail from django.db import models from django.template import Template, Context +from django.urls import reverse +from django.utils.timezone import now, get_current_timezone from django.utils.translation import ugettext_lazy as _ from sentry_sdk import capture_exception -from .states import states +from catalogue.utils import get_random_hash +from .states import Level, states class EmailTemplate(models.Model): @@ -31,80 +36,104 @@ class EmailTemplate(models.Model): verbose_name_plural = _('email templates') def __str__(self): - return '%s (%+d)' % (self.get_state_display(), self.min_days_since) + return '%s (%+d)' % (self.get_state_display(), self.min_days_since or 0) + + @classmethod + def get_current(cls, time=None): + time = (time or now()).astimezone(get_current_timezone()) + weekday = time.isoweekday() + qs = cls.objects.filter(is_active=True) + qs = qs.exclude(min_hour__gt=time.hour).exclude(max_hour__lte=time.hour) + qs = qs.exclude(min_day_of_month__gt=time.day).exclude(max_day_of_month__lte=time.day) + qs = qs.exclude(**{f'dow_{weekday}': False}) + return qs def run(self, time=None, verbose=False, dry_run=False): - state = self.get_state() - recipients = state(time=time, offset=self.days).get_recipients() - hash_values = set(recipient.hash_value for recipient in recipients) - sent = set(EmailSent.objects.filter( - template=self, hash_value__in=hash_values - ).values_list('hash_value', flat=True)) - for recipient in recipients: - if recipient.hash_value in sent: - continue - self.send(recipient, verbose=verbose, dry_run=dry_run) - - def get_state(self): + state = self.get_state(time=time) + contacts = state.get_contacts() + + contacts = contacts.exclude(emailsent__template=self) + for contact in contacts: + if not contact.is_annoyed: + self.send(contact, verbose=verbose, dry_run=dry_run) + + def get_state(self, time=None, test=False): for s in states: if s.slug == self.state: - return s + return s( + time=time, + min_days_since=self.min_days_since, + max_days_since=self.max_days_since, + test=test + ) raise ValueError('Unknown state', s.state) - def send(self, recipient, verbose=False, dry_run=False): - subject = Template(self.subject).render(Context(recipient.context)) - body = Template(self.body).render(Context(recipient.context)) + def send(self, contact, verbose=False, dry_run=False, test=False): + state = self.get_state(test=test) + ctx = state.get_context(contact) + ctx['contact'] = contact + ctx = Context(ctx) + + subject = Template(self.subject).render(ctx) + if test: + subject = "[test] " + subject + + body_template = '{% extends "messaging/email_body.html" %}{% block body %}' + self.body + '{% endblock %}' + body = Template(body_template).render(ctx) + if verbose: - print(recipient.email, subject) + print(self.pk, subject, contact.email) if not dry_run: try: - send_mail(subject, body, settings.CONTACT_EMAIL, [recipient.email], fail_silently=False) + send_mail(subject, body, settings.CONTACT_EMAIL, [contact.email], fail_silently=False) except: capture_exception() else: - self.emailsent_set.create( - hash_value=recipient.hash_value, - email=recipient.email, - subject=subject, - body=body, - ) - - -class EmailSent(models.Model): - template = models.ForeignKey(EmailTemplate, models.CASCADE) - hash_value = models.CharField(max_length=1024) - timestamp = models.DateTimeField(auto_now_add=True) - email = models.CharField(_('e-mail'), max_length=1024) - subject = models.CharField(_('subject'), max_length=1024) - body = models.TextField(_('body')) - - class Meta: - verbose_name = _('email sent') - verbose_name_plural = _('emails sent') - ordering = ('-timestamp',) + if not test: + self.emailsent_set.create( + contact=contact, + subject=subject, + body=body, + ) - def __str__(self): - return '%s %s' % (self.email, self.timestamp) + def send_test_email(self, email): + contact = Contact( + email=email, + key='test' + ) + self.send(contact, test=True) class Contact(models.Model): - COLD = 10 - TRIED = 20 - SINGLE = 30 - RECURRING = 40 - OPT_OUT = 50 - email = models.EmailField(unique=True) level = models.PositiveSmallIntegerField( choices=[ - (TRIED, _('Would-be donor')), - (SINGLE, _('One-time donor')), - (RECURRING, _('Recurring donor')), - (COLD, _('Cold')), - (OPT_OUT, _('Opt out')), + (Level.COLD, _('Cold')), + (Level.TRIED, _('Would-be donor')), + (Level.SINGLE, _('One-time donor')), + (Level.RECURRING, _('Recurring donor')), + (Level.OPT_OUT, _('Opt out')), ]) since = models.DateTimeField() expires_at = models.DateTimeField(null=True, blank=True) + key = models.CharField(max_length=64, blank=True) + + class Meta: + verbose_name = _('contact') + verbose_name_plural = _('contacts') + + def save(self, *args, **kwargs): + if not self.key: + self.key = get_random_hash(self.email) + super().save(*args, **kwargs) + + @property + def is_annoyed(self): + cutoff = now() - timedelta(settings.MESSAGING_MIN_DAYS) + return self.emailsent_set.filter(timestamp__gte=cutoff).exists() + + def get_optout_url(self): + return reverse('messaging_optout', args=[self.key]) @classmethod def update(cls, email, level, since, expires_at=None): @@ -116,7 +145,9 @@ class Contact(models.Model): if not created: obj.ascend(level, since, expires_at) - def ascend(self, level, since, expires_at=None): + def ascend(self, level, since=None, expires_at=None): + if since is None: + since = now() if level < self.level: return if level == self.level: @@ -132,3 +163,19 @@ class Contact(models.Model): self.expires_at = expires_at self.save() + +class EmailSent(models.Model): + template = models.ForeignKey(EmailTemplate, models.CASCADE) + contact = models.ForeignKey(Contact, models.CASCADE) + timestamp = models.DateTimeField(auto_now_add=True) + subject = models.CharField(_('subject'), max_length=1024) + body = models.TextField(_('body')) + + class Meta: + verbose_name = _('email sent') + verbose_name_plural = _('emails sent') + ordering = ('-timestamp',) + + def __str__(self): + return '%s %s' % (self.contact.email, self.timestamp) +