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