src/messaging/states.py

   1 from datetime import timedelta
   2 from django.apps import apps
   3 from django.utils.timezone import now
   4 from django.utils.translation import ugettext_lazy as _
   5
   6
   7 class Level:
   8     COLD = 10
   9     TRIED = 20
  10     SINGLE = 30
  11     RECURRING = 40
  12     MANUAL_MEMBER = 45
  13     OPT_OUT = 50
  14
  15
  16 class State:
  17     allow_negative_offset = False
  18     level = None
  19     expired = None
  20
  21     def __init__(self, time=None, min_days_since=None, max_days_since=None, test=False):
  22         self.time = time or now()
  23         self.min_days_since = min_days_since
  24         self.max_days_since = max_days_since
  25         self.test = test
  26
  27     def get_contacts(self):
  28         Contact = apps.get_model('messaging', 'Contact')
  29         contacts = Contact.objects.filter(level=self.level)
  30
  31         if self.min_days_since is not None or self.expired:
  32             cutoff = self.time - timedelta(self.min_days_since or 0)
  33             if self.expired:
  34                 contacts = contacts.filter(expires_at__lt=cutoff)
  35             else:
  36                 contacts = contacts.filter(since__lt=cutoff)
  37
  38         if self.max_days_since is not None:
  39             cutoff = self.time - timedelta(self.max_days_since)
  40             if self.expired:
  41                 contacts = contacts.filter(expires_at__gt=cutoff)
  42             else:
  43                 contacts = contacts.filter(since__gt=cutoff)
  44
  45         if self.expired is False:
  46             contacts = contacts.exclude(expires_at__lt=self.time)
  47
  48         return contacts
  49
  50     def get_context(self, contact):
  51         if self.test:
  52             return self.get_example_context(contact)
  53
  54         Schedule = apps.get_model('club', 'Schedule')
  55         schedules = Schedule.objects.filter(email=contact.email)
  56         return {
  57             "schedule": self.get_schedule(schedules)
  58         }
  59
  60     def get_example_context(self, contact):
  61         Schedule = apps.get_model('club', 'Schedule')
  62         return {
  63             "schedule": Schedule(
  64                 email=contact.email,
  65                 key='xxxxxxxxx',
  66                 amount=100,
  67                 payed_at=self.time - timedelta(2),
  68                 started_at=self.time - timedelta(1),
  69                 expires_at=self.time + timedelta(1),
  70             )
  71         }
  72
  73
  74 class ClubSingle(State):
  75     slug = 'club-single'
  76     name = _('club one-time donors')
  77     level = Level.SINGLE
  78     expired = False
  79
  80     def get_schedule(self, schedules):
  81         # Find first single non-expired schedule.
  82         return schedules.filter(
  83             monthly=False, yearly=False,
  84             expires_at__gt=self.time
  85         ).order_by('started_at').first()
  86
  87
  88 class ClubSingleExpired(State):
  89     slug = 'club-membership-expiring'
  90     allow_negative_offset = True
  91     name = _('club one-time donors with donation expiring')
  92     level = Level.SINGLE
  93     expired = True
  94
  95     def get_schedule(self, schedules):
  96         # Find last single expired schedule.
  97         return schedules.filter(
  98             monthly=False, yearly=False,
  99             expires_at__lt=self.time
 100         ).order_by('-expires_at').first()
 101
 102
 103 class ClubTried(State):
 104     slug = 'club-payment-unfinished'
 105     name = _('club would-be donors')
 106     level = Level.TRIED
 107
 108     def get_schedule(self, schedules):
 109         # Find last unpaid schedule.
 110         return schedules.filter(
 111             payed_at=None
 112         ).order_by('-started_at').first()
 113
 114
 115 class ClubRecurring(State):
 116     slug = 'club-recurring'
 117     name = _('club recurring donors')
 118     level = Level.RECURRING
 119     expired = False
 120
 121     def get_schedule(self, schedules):
 122         # Find first recurring non-expired schedule.
 123         return schedules.exclude(
 124             monthly=False, yearly=False
 125         ).filter(
 126             expires_at__gt=self.time
 127         ).order_by('started_at').first()
 128
 129
 130 class ClubRecurringExpired(State):
 131     slug = 'club-recurring-payment-problem'
 132     name = _('club recurring donors with donation expired')
 133     level = Level.RECURRING
 134     expired = True
 135
 136     def get_schedule(self, schedules):
 137         # Find last recurring expired schedule.
 138         return schedules.exclude(
 139             monthly=False, yearly=False
 140         ).filter(
 141             expires_at__lt=self.time
 142         ).order_by('-expires_at').first()
 143
 144
 145 class Cold(State):
 146     slug = 'cold'
 147     name = _('cold group')
 148     level = Level.COLD
 149
 150     def get_context(self, contact):
 151         return {}
 152
 153
 154 states = [
 155     Cold,
 156     ClubTried,
 157     ClubSingle,
 158     ClubSingleExpired,
 159     ClubRecurring,
 160     ClubRecurringExpired,
 161 ]
 162