da8a01edc9da8d4cc525cedbb1864d798e75a315
[wolnelektury.git] / src / messaging / models.py
1 from django.apps import apps
2 from django.conf import settings
3 from django.core.mail import send_mail
4 from django.db import models
5 from django.template import Template, Context
6 from django.urls import reverse
7 from django.utils.translation import ugettext_lazy as _
8 from sentry_sdk import capture_exception
9 from catalogue.utils import get_random_hash
10 from .states import Level, states
11
12
13 class EmailTemplate(models.Model):
14     state = models.CharField(_('state'), max_length=128, choices=[(s.slug, s.name) for s in states], help_text='?')
15     subject = models.CharField(_('subject'), max_length=1024)
16     body = models.TextField(_('body'))
17     min_days_since = models.SmallIntegerField(_('min days since'), null=True, blank=True)
18     max_days_since = models.SmallIntegerField(_('max days since'), null=True, blank=True)
19     min_hour = models.PositiveSmallIntegerField(_('min hour'), null=True, blank=True)
20     max_hour = models.PositiveSmallIntegerField(_('max hour'), null=True, blank=True)
21     min_day_of_month = models.PositiveSmallIntegerField(_('min day of month'), null=True, blank=True)
22     max_day_of_month = models.PositiveSmallIntegerField(_('max day of month'), null=True, blank=True)
23     dow_1 = models.BooleanField(_('Monday'), default=True)
24     dow_2 = models.BooleanField(_('Tuesday'), default=True)
25     dow_3 = models.BooleanField(_('Wednesday'), default=True)
26     dow_4 = models.BooleanField(_('Thursday'), default=True)
27     dow_5 = models.BooleanField(_('Friday'), default=True)
28     dow_6 = models.BooleanField(_('Saturday'), default=True)
29     dow_7 = models.BooleanField(_('Sunday'), default=True)
30     is_active = models.BooleanField(_('active'), default=False)
31
32     class Meta:
33         verbose_name = _('email template')
34         verbose_name_plural = _('email templates')
35
36     def __str__(self):
37         return '%s (%+d)' % (self.get_state_display(), self.min_days_since or 0)
38
39     def run(self, time=None, verbose=False, dry_run=False):
40         state = self.get_state(time=time)
41         contacts = state.get_contacts()
42     
43         contacts = contacts.exclude(emailsent__template=self)
44         for contact in contacts:
45             self.send(contact, verbose=verbose, dry_run=dry_run)
46
47     def get_state(self, time=None, test=False):
48         for s in states:
49             if s.slug == self.state:
50                 return s(
51                     time=time,
52                     min_days_since=self.min_days_since,
53                     max_days_since=self.max_days_since,
54                     test=test
55                 )
56         raise ValueError('Unknown state', s.state)
57
58     def send(self, contact, verbose=False, dry_run=False, test=False):
59         state = self.get_state(test=test)
60         ctx = state.get_context(contact)
61         ctx['contact'] = contact
62         ctx = Context(ctx)
63
64         subject = Template(self.subject).render(ctx)
65         if test:
66             subject = "[test] " + subject
67
68         body_template = '{% extends "messaging/email_body.html" %}{% block body %}' + self.body + '{% endblock %}'
69         body = Template(body_template).render(ctx)
70
71         if verbose:
72             print(contact.email, subject)
73         if not dry_run:
74             try:
75                 send_mail(subject, body, settings.CONTACT_EMAIL, [contact.email], fail_silently=False)
76             except:
77                 capture_exception()
78             else:
79                 if not test:
80                     self.emailsent_set.create(
81                         contact=contact,
82                         subject=subject,
83                         body=body,
84                     )
85
86     def send_test_email(self, email):
87         contact = Contact(
88                 email=email,
89                 key='test'
90             )
91         self.send(contact, test=True)
92
93
94 class Contact(models.Model):
95     email = models.EmailField(unique=True)
96     level = models.PositiveSmallIntegerField(
97         choices=[
98             (Level.COLD, _('Cold')),
99             (Level.TRIED, _('Would-be donor')),
100             (Level.SINGLE, _('One-time donor')),
101             (Level.RECURRING, _('Recurring donor')),
102             (Level.OPT_OUT, _('Opt out')),
103         ])
104     since = models.DateTimeField()
105     expires_at = models.DateTimeField(null=True, blank=True)
106     key = models.CharField(max_length=64, blank=True)
107
108     class Meta:
109         verbose_name = _('contact')
110         verbose_name_plural = _('contacts')
111
112     def save(self, *args, **kwargs):
113         if not self.key:
114             self.key = get_random_hash(self.email)
115         super().save(*args, **kwargs)
116
117     def get_optout_url(self):
118         return reverse('messaging_optout', args=[self.key])
119
120     @classmethod
121     def update(cls, email, level, since, expires_at=None):
122         obj, created = cls.objects.get_or_create(email=email, defaults={
123                 "level": level,
124                 "since": since,
125                 "expires_at": expires_at
126             })
127         if not created:
128             obj.ascend(level, since, expires_at)
129
130     def ascend(self, level, since, expires_at=None):
131         if level < self.level:
132             return
133         if level == self.level:
134             self.since = min(since, self.since)
135
136             if expires_at and self.expires_at:
137                 self.expires_at = max(expires_at, self.expires_at)
138             else:
139                 self.expires_at = expires_at
140         else:
141             self.level = level
142             self.since = since
143             self.expires_at = expires_at
144         self.save()
145
146
147 class EmailSent(models.Model):
148     template = models.ForeignKey(EmailTemplate, models.CASCADE)
149     contact = models.ForeignKey(Contact, models.CASCADE)
150     timestamp = models.DateTimeField(auto_now_add=True)
151     subject = models.CharField(_('subject'), max_length=1024)
152     body = models.TextField(_('body'))
153
154     class Meta:
155         verbose_name = _('email sent')
156         verbose_name_plural = _('emails sent')
157         ordering = ('-timestamp',)
158
159     def __str__(self):
160         return '%s %s' % (self.email, self.timestamp)
161