Add messaging.

[wolnelektury.git] / src / messaging / models.py

diff --git a/src/messaging/models.py b/src/messaging/models.py
new file mode 100644 (file)
index 0000000..feb5daa
--- /dev/null
+++ b/src/messaging/models.py
@@ -0,0 +1,77 @@
+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.utils.translation import ugettext_lazy as _
+from sentry_sdk import capture_exception
+from .states import states
+
+
+class EmailTemplate(models.Model):
+    state = models.CharField(_('state'), max_length=128, choices=[(s.slug, s.name) for s in states], help_text='?')
+    subject = models.CharField(_('subject'), max_length=1024)
+    body = models.TextField(_('body'))
+    days = models.SmallIntegerField(_('days'), null=True, blank=True)
+    hour = models.IntegerField(_('hour'), null=True, blank=True)
+
+    class Meta:
+        verbose_name = _('email template')
+        verbose_name_plural = _('email templates')
+
+    def __str__(self):
+        return '%s (%+d)' % (self.get_state_display(), self.days)
+        return self.subject
+
+    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):
+        for s in states:
+            if s.slug == self.state:
+                return s
+        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))
+        if verbose:
+            print(recipient.email, subject)
+        if not dry_run:
+            try:
+                send_mail(subject, body, settings.CONTACT_EMAIL, [recipient.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',)
+
+    def __str__(self):
+        return '%s %s' % (self.email, self.timestamp)