add a filter
[wolnelektury.git] / src / messaging / models.py
index da8a01e..b9cbcad 100644 (file)
@@ -1,48 +1,60 @@
+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.translation import ugettext_lazy as _
+from django.utils.timezone import now, get_current_timezone
 from sentry_sdk import capture_exception
 from catalogue.utils import get_random_hash
 from .states import Level, 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'))
-    min_days_since = models.SmallIntegerField(_('min days since'), null=True, blank=True)
-    max_days_since = models.SmallIntegerField(_('max days since'), null=True, blank=True)
-    min_hour = models.PositiveSmallIntegerField(_('min hour'), null=True, blank=True)
-    max_hour = models.PositiveSmallIntegerField(_('max hour'), null=True, blank=True)
-    min_day_of_month = models.PositiveSmallIntegerField(_('min day of month'), null=True, blank=True)
-    max_day_of_month = models.PositiveSmallIntegerField(_('max day of month'), null=True, blank=True)
-    dow_1 = models.BooleanField(_('Monday'), default=True)
-    dow_2 = models.BooleanField(_('Tuesday'), default=True)
-    dow_3 = models.BooleanField(_('Wednesday'), default=True)
-    dow_4 = models.BooleanField(_('Thursday'), default=True)
-    dow_5 = models.BooleanField(_('Friday'), default=True)
-    dow_6 = models.BooleanField(_('Saturday'), default=True)
-    dow_7 = models.BooleanField(_('Sunday'), default=True)
-    is_active = models.BooleanField(_('active'), default=False)
+    state = models.CharField('stan', max_length=128, choices=[(s.slug, s.name) for s in states], help_text='?')
+    subject = models.CharField('temat', max_length=1024)
+    body = models.TextField('treść')
+    min_days_since = models.SmallIntegerField('dni po, od', null=True, blank=True)
+    max_days_since = models.SmallIntegerField('dni po, do', null=True, blank=True)
+    min_hour = models.PositiveSmallIntegerField('od godziny', null=True, blank=True)
+    max_hour = models.PositiveSmallIntegerField('do godziny', null=True, blank=True)
+    min_day_of_month = models.PositiveSmallIntegerField('od dnia miesiąca', null=True, blank=True)
+    max_day_of_month = models.PositiveSmallIntegerField('do dnia miesiąca', null=True, blank=True)
+    dow_1 = models.BooleanField('poniedziałek', default=True)
+    dow_2 = models.BooleanField('wtorek', default=True)
+    dow_3 = models.BooleanField('środa', default=True)
+    dow_4 = models.BooleanField('czwartek', default=True)
+    dow_5 = models.BooleanField('piątek', default=True)
+    dow_6 = models.BooleanField('sobota', default=True)
+    dow_7 = models.BooleanField('niedziela', default=True)
+    is_active = models.BooleanField('aktywny', default=False)
 
     class Meta:
-        verbose_name = _('email template')
-        verbose_name_plural = _('email templates')
+        verbose_name = 'szablon e-maila'
+        verbose_name_plural = 'szablony e-maili'
 
     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(time=time)
         contacts = state.get_contacts()
     
         contacts = contacts.exclude(emailsent__template=self)
         for contact in contacts:
-            self.send(contact, verbose=verbose, dry_run=dry_run)
+            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:
@@ -69,7 +81,7 @@ class EmailTemplate(models.Model):
         body = Template(body_template).render(ctx)
 
         if verbose:
-            print(contact.email, subject)
+            print(self.pk, subject, contact.email)
         if not dry_run:
             try:
                 send_mail(subject, body, settings.CONTACT_EMAIL, [contact.email], fail_silently=False)
@@ -95,25 +107,31 @@ class Contact(models.Model):
     email = models.EmailField(unique=True)
     level = models.PositiveSmallIntegerField(
         choices=[
-            (Level.COLD, _('Cold')),
-            (Level.TRIED, _('Would-be donor')),
-            (Level.SINGLE, _('One-time donor')),
-            (Level.RECURRING, _('Recurring donor')),
-            (Level.OPT_OUT, _('Opt out')),
+            (Level.COLD, 'Lodówka'),
+            (Level.TRIED, 'Niedoszły darczyńca'),
+            (Level.SINGLE, 'Darczyńca z jednorazową wpłatą'),
+            (Level.RECURRING, 'Darczyńca z wpłatą cykliczną'),
+            (Level.MANUAL_MEMBER, 'Członkostwo ustawione ręcznie'),
+            (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')
+        verbose_name = 'kontakt'
+        verbose_name_plural = 'kontakty'
 
     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])
 
@@ -127,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:
@@ -143,19 +163,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'))
+    subject = models.CharField('temat', max_length=1024)
+    body = models.TextField('treść')
 
     class Meta:
-        verbose_name = _('email sent')
-        verbose_name_plural = _('emails sent')
+        verbose_name = 'wysłany e-mail'
+        verbose_name_plural = 'wysłane e-maile'
         ordering = ('-timestamp',)
 
     def __str__(self):
-        return '%s %s' % (self.email, self.timestamp)
+        return '%s %s' % (self.contact.email, self.timestamp)