-
Notifications
You must be signed in to change notification settings - Fork 1
/
models.py
55 lines (45 loc) · 2.07 KB
/
models.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
from django.conf import settings
from django.db import models
from django.utils import translation
from django.utils.translation import ugettext_lazy as _
from notifications.utils import DEFAULT_LANGUAGE
from signups.utils import localize_datetime
class SignupTarget(models.Model):
identifier = models.CharField(max_length=100, verbose_name=_('identifier'), unique=True)
name = models.CharField(max_length=100, verbose_name=_('name'))
users = models.ManyToManyField(
settings.AUTH_USER_MODEL, verbose_name=_('users'), related_name='signup_targets', through='Signup'
)
class Meta:
verbose_name = _('sign-up target')
verbose_name_plural = _('sign-up targets')
ordering = ('id',)
def __str__(self):
return self.name
class Signup(models.Model):
created_at = models.DateTimeField(verbose_name=_('created at'), auto_now_add=True, editable=False)
cancelled_at = models.DateTimeField(verbose_name=_('cancelled at'), null=True, blank=True)
user = models.ForeignKey(
settings.AUTH_USER_MODEL, verbose_name=_('user'), related_name='signups', on_delete=models.CASCADE
)
target = models.ForeignKey(
SignupTarget, verbose_name=_('target'), related_name='signups', on_delete=models.CASCADE
)
class Meta:
verbose_name = _('sign-up')
verbose_name_plural = _('sign-ups')
ordering = ('id',)
def __str__(self):
text = '{} {} {}'.format(self. user, self.target, localize_datetime(self.created_at))
if self.cancelled_at:
text = '{} cancelled {}'.format(text, localize_datetime(self.cancelled_at))
return text
def get_notification_context(self, language=DEFAULT_LANGUAGE):
with translation.override(language):
return {
'created_at': localize_datetime(self.created_at),
'cancelled_at': localize_datetime(self.cancelled_at),
'user__first_name': self.user.first_name,
'user__last_name': self.user.last_name,
'target__name': self.target.name,
}