X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/ef79260cc592940ed31899db916fb84bd723291c..623676b97cc93293eff63398c7deb45cdec76af9:/src/messaging/models.py

diff --git a/src/messaging/models.py b/src/messaging/models.py
index b7a87b35d..fccd64776 100644
--- a/src/messaging/models.py
+++ b/src/messaging/models.py
@@ -1,10 +1,11 @@
+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
+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
@@ -37,13 +38,24 @@ 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(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:
@@ -70,7 +82,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)
@@ -115,6 +127,11 @@ class Contact(models.Model):
             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])
 
@@ -160,5 +177,5 @@ class EmailSent(models.Model):
         ordering = ('-timestamp',)
 
     def __str__(self):
-        return '%s %s' % (self.email, self.timestamp)
+        return '%s %s' % (self.contact.email, self.timestamp)