X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/71d31efcdb70122f705d8136239771747ca3b07d..333f7cc2ab4d123e4ffddd4f09371d06750b3631:/src/messaging/models.py diff --git a/src/messaging/models.py b/src/messaging/models.py index 24cf623f1..7891e0c63 100644 --- a/src/messaging/models.py +++ b/src/messaging/models.py @@ -1,14 +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 catalogue.utils import get_random_hash -from .recipient import Recipient -from .states import states +from .states import Level, states class EmailTemplate(models.Model): @@ -37,107 +38,101 @@ class EmailTemplate(models.Model): def __str__(self): 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, test=False): - ctx = Context(recipient.context) - - if test: - contact = Contact(email=recipient.email, key='test') - else: - # TODO: actually, we should just use Contacts instead of recipients. - contact = Contact.objects.get(email=recipient.email) - + 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: if not test: self.emailsent_set.create( - hash_value=recipient.hash_value, - email=recipient.email, + contact=contact, subject=subject, body=body, ) def send_test_email(self, email): - state = self.get_state()() - recipient = state.get_example_recipient(email) - self.send(recipient, test=True) - - -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',) - - def __str__(self): - return '%s %s' % (self.email, self.timestamp) + 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.MANUAL_MEMBER, _('Manually set as member')), + (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]) @@ -151,7 +146,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: @@ -167,3 +164,32 @@ class Contact(models.Model): self.expires_at = expires_at self.save() + @classmethod + def reset(cls, email): + cls.objects.filter(email=email).delete() + Schedule = apps.get_model('club', 'Schedule') + Membership = apps.get_model('club', 'Membership') + for schedule in Schedule.objects.filter(email=email): + schedule.update_contact() + for membership in Membership.objects.filter(manual=True, user__email=email): + membership.update_contact() + + def wl_optout_url(self): + return 'https://wolnelektury.pl' + self.get_optout_url() + + +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) +